在INSERT语句的RETURNING中包含表名

Question

我有一个INSERT语句，以便返回插入的ID：

INSERT INTO encyclopedias (page_id)
SELECT page.id FROM pages
RETURNING id;

INSERT INTO cookbooks (page_id)
SELECT page.id FROM pages
RETURNING id;

返回如下内容：

 id 
----
1
2

 id 
----
3
4

解析输出时，我想告诉您ID来自哪个表，例如：

encyclopedia id
----
1
2

cookbook id
----
3
4

要么：

table, id
----
encyclopedias, 1
encyclopedias, 2

table, id
----
cookbooks, 3
cookbooks, 4

我该怎么做？

Answer 1

有一个简单的解决方案，可以使用系统列tableoid来自动执行此操作。

那就是源表的内部OID（对象ID），可以将其转换为regclass并将其转换为实际的表名。

INSERT INTO encyclopedias (page_id)
SELECT id FROM pages
RETURNING tableoid::regclass::text AS table, page_id AS id;

精确返回所需的输出：

table         | id
--------------+----
encyclopedias | 1
encyclopedias | 2

如果您更改FROM子句中的表（例如您的示例），则无需改编RETURNING子句。

有关：

• 查询行继承的父表时，获取行的源表的名称

Answer 2

在returning子句中使用列别名和常量：

insert into t(x) values(1) returning x as xx, 'the table' as table_name;
╔════╤════════════╗
║ xx │ table_name ║
╠════╪════════════╣
║  1 │ the table  ║
╚════╧════════════╝

UPD：

另外，您可以为psql指定几种输出格式设置，例如：

$ echo "
> \pset format unaligned
> \pset tuples_only on
> \echo --==## foo ##==--
> select 1,2,3;" | psql
Output format is unaligned.
Tuples only is on.
--==## foo ##==--
1|2|3

在文档中找到更多信息 。

Answer 3

一种方法是使用WITH查询：

WITH inserts AS (
INSERT INTO encyclopedias (page_id)
SELECT page.id FROM pages
RETURNING id
)
SELECT inserts.id AS encyclopedia_id;

要么

WITH inserts AS (
INSERT INTO encyclopedias (page_id)
SELECT page.id FROM pages
RETURNING id
)
SELECT 'encyclopedia' AS "table", inserts.id AS "id";

PostgreSQL文档页面上有关INSERT的更多信息