如何從同一SELECT查詢插入到兩個表

Question

我在PostgreSQL有以下代碼：

    for row in select .... from .... where ....
    loop
        insert into A(id1,id2,id3,quantity)
               select aa,bb,1,quantity
               from func1(row.idfirst);
   end loop;

我想用相同的查詢做另一個INSERT ，所以我可以這樣做：

    for row in select .... from .... where ....
    loop
        insert into A(id1,id2,id3,quantity)
               select aa,bb,1,quantity
               from func1(row.idfirst);

        insert into B(first,second,third,forth)
               select aa,bb,1,quantity
               from func1(row.idfirst);
   end loop;

事實是，相同代碼是多余的...而且func1是一個巨大的功能。 它需要很長的時間才能工作，而且我不瘋狂地運行兩次。

有什么解決方案可以對同一select查詢執行兩次插入嗎？

Answer 1

您可以在PostgreSQL 9.1+中使用CTE，如下所示：

WITH CTE AS (
    INSERT INTO A(id1,id2,id3,quantity)
               SELECT aa,bb,1,quantity
               FROM func1(row.idfirst);
    RETURNING id1,id2,id3,quantity
    )
INSERT INTO B(first,second,third,forth)
SELECT id1,id2,id3,quantity
FROM CTE;