PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

Testare le funzioni di PostgreSQL che consumano e restituiscono riflessor

Q1

Il tuo "piccolo test" può essere un semplice SQL:

BEGIN;
SELECT get_function_that_returns_cursor('ret', 4100, 'foo', 123); -- note: 'ret'
FETCH ALL IN ret; -- works for any rowtype

COMMIT;  -- or ROLLBACK;

Esegui COMMIT / ROLLBACK dopo hai controllato i risultati. La maggior parte dei client visualizza solo il risultato del comando lat.

Maggiori informazioni nel capitolo Ritorno dei cursori del manuale.

2Q

E se non conosciamo il tipo di riga che viene restituito, come potremmo farlo?

Dato che vuoi solo esaminare i risultati, puoi trasmettere l'intero record su text .In questo modo eviti del tutto il problema con i tipi di ritorno dinamici per la funzione.

Considera questa demo:

CREATE TABLE a (a_id int PRIMARY KEY, a text);
INSERT INTO a VALUES (1, 'foo'), (2, 'bar');

CREATE OR REPLACE FUNCTION reffunc(INOUT ret refcursor) AS  -- INOUT param :)
$func$
BEGIN
    OPEN ret FOR SELECT * FROM a;
END
$func$ LANGUAGE plpgsql;


CREATE OR REPLACE FUNCTION ctest()
  RETURNS SETOF text AS
$func$
DECLARE
    curs1 refcursor;
    rec   record;
BEGIN
  curs1 := reffunc('ret');   -- simple assignment
  
  LOOP
    FETCH curs1 INTO rec;
    EXIT WHEN NOT FOUND;     -- note the placement!
    RETURN NEXT rec::text;
  END LOOP;
END
$func$ LANGUAGE plpgsql;

-> SQLfiddle