PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Ajoutez dynamiquement une colonne avec plusieurs valeurs à n'importe quelle table à l'aide d'une fonction PL/pgSQL

Je l'ai fait avec refcursor si vous voulez que la liste des colonnes de sortie soit complètement dynamique :

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;

Ensuite, il faut appeler FETCH ALL IN pour obtenir des résultats. Syntaxe détaillée ici ou là :https://stackoverflow.com/a/12483222/630169 . Apparemment c'est le seul moyen pour l'instant. J'espère que quelque chose sera changé dans PostgreSQL 11 avec PROCEDURES.