PostgreSQL
 sql >> Datenbank >  >> RDS >> PostgreSQL

Implementieren eines UPDATE auf SELECT in Postgres

Beachten Sie den Standardwert für die Spalte „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;

Fragen Sie nun die Tabelle mit obiger Funktion ab:

select * from select_and_update('a');

Update laut Kommentar:

Sie können es dynamisch erstellen und anstelle einer Funktion einfach den SQL-Code, was auch immer es ist, in eine Transaktion einschließen. Keine Notwendigkeit für Cursor.

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;