从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文件对应的列创建一个临时表
2. COPY FROM CSV文件COPY FROM到该临时表中
3. 根据从临时表LEFT JOIN到最终表中现有行的SELECT插入到最终表中，因此可以确定primary行是否已存在
4. 删除临时表

（顺便说一句，这意味着您可以使用SECURITY DEFINER函数，该函数以root身份创建，但由特权较低的用户以CSV文件名硬编码运行，而不是以root身份运行整个导入。）