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

Aggiungi dinamicamente una colonna con più valori a qualsiasi tabella utilizzando una funzione PL/pgSQL

L'ho fatto con refcursor se vuoi che l'elenco delle colonne di output sia completamente dinamico:

CREATE OR REPLACE FUNCTION is_record_exists(tablename character varying, columns character varying[], keepcolumns character varying[] DEFAULT NULL::character varying[])
    RETURNS SETOF refcursor AS
$BODY$

DECLARE 
    ref refcursor;
    keepColumnsList text;
    columnsList text; 
    valuesList text;
    existQuery text;
    keepQuery text;
BEGIN
    IF keepcolumns IS NOT NULL AND array_length(keepColumns, 1) > 0 THEN
        keepColumnsList :=  array_to_string(keepColumns, ', ');
    ELSE
        keepColumnsList :=  'COUNT(*)';
    END IF;

    columnsList := (SELECT array_to_string(array_agg(name || ' = ' || value), ' OR ') FROM
        (SELECT unnest(columns[1:1]) AS name, unnest(columns[2:2]) AS value) pair);

    existQuery := 'SELECT ' || keepColumnsList || ' FROM ' || tableName || ' WHERE ' || columnsList;
    RAISE NOTICE 'Exist query: %', existQuery;

    OPEN ref FOR EXECUTE
        existQuery;
    RETURN next ref;
END;$BODY$
  LANGUAGE plpgsql;

Quindi devi chiamare FETCH ALL IN per ottenere risultati. Sintassi dettagliata qui oppure lì:https://stackoverflow.com/a/12483222/630169 . Sembra che sia l'unico modo per ora. Spero che qualcosa cambierà in PostgreSQL 11 con PROCEDURE.