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

Postgres-Funktion zum Einfügen mehrerer Datensätze in zwei Tabellen

Sie können ein Array von Tupeln verwenden, um mehrere Zeilen an die Funktion zu übergeben. Sie benötigen einen benutzerdefinierten Typ:

create type order_input as (
    item text,
    quantity integer);

Verwenden Sie ein Array dieses Typs für ein Argument der Funktion:

create or replace function insert_into_orders(order_input[])
returns void language plpgsql as $$
declare 
    inserted_id integer;
begin
    insert into public.orders(orderdate) 
    values (now()) 
    returning orderid into inserted_id;

    insert into public.orderdetails(orderid, item, quantity)
    select inserted_id, item, quantity
    from unnest($1);
end $$;

Verwendung:

select insert_into_orders(
    array[
        ('Red Widget', 10), 
        ('Blue Widget', 5)
    ]::order_input[]
);

select * from orderdetails;

 orderdetailid | orderid |    item     | quantity 
---------------+---------+-------------+----------
             1 |       1 | Red Widget  |       10
             2 |       1 | Blue Widget |        5
(2 rows)