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

implémenter une mise à jour sur SELECT dans Postgres

Notez la valeur par défaut de la colonne qry_count :

CREATE TABLE t (
    a INTEGER PRIMARY KEY, 
    b TEXT, 
    entered_by INTEGER, 
    qry_count INTEGER default 0
);

create function select_and_update(parameter text)
returns setof t as $$
    update t
    set qry_count = qry_count + 1
    from (
        select a
        from t
        where b = $1
        ) s
    where t.a = s.a
    ;
    select *
    from t
    where b = $1
    ;
$$ language sql;

Interrogez maintenant la table à l'aide de la fonction ci-dessus :

select * from select_and_update('a');

Mettre à jour selon le commentaire :

Vous pouvez le construire dynamiquement et au lieu d'une fonction, encapsulez simplement le code sql, quel qu'il soit, dans une transaction. Pas besoin de curseurs.

begin;
    update t
    set qry_count = qry_count + 1
    from (
        select a
        from t
        where b = 'a'
        ) s
    where t.a = s.a
    ;
    select *
    from t
    where b = 'a'
    ;
commit;