从CSV到postgres DB的条件插入

Question

我想要将CSV导入到Postgres表中，但我希望能够根据表中的现有行动态决定一列应该是0还是1。

例如，我正在向表中添加联系人，如果不存在主要联系人，则将其标记为主要联系人，否则将其添加为次要联系人：

现有行：

contact_id | branch_id | primary
-------------+-----------+--------
1          | 100       | 1
2          | 101       | 1
3          | 101       | 0

CSV数据，

contact_id | branch_id
-----------+-----------
4          | 100
5          | 101
6          | 102
7          | 103

期望的结果，

contact_id | branch_id | primary
-----------+-----------+--------
1          | 100       | 1
2          | 101       | 1
3          | 101       | 0
4          | 100       | 0
5          | 101       | 0
6          | 102       | 1
7          | 103       | 1

请注意，联系人4和5被添加为次要联系人，因为这些分支机构已存在主要联系人，而6和7添加为主要联系人，因为这些分支机构不存在主要联系人。

postgres 9.2可以做到这一点吗？

Answer 1

我将使用PL/pgSQL函数和临时表分两个阶段来实现。基本上：

1. 创建一个临时表，其中包含与CSV file
2. COPY FROM对应的列，并将CSV文件插入到该临时表中，并根据临时表中的SELECT将其插入到最终表中，这样您就可以确定primary行是否已经存在
3. 删除临时表

(顺便说一下，这意味着您可以使用创建为root的SECURITY DEFINER函数，但由较低权限的用户使用硬编码的CSV文件名运行，而不是以root运行整个导入。)