如何在PostgreSQL COPY命令中为文件缺失的非空列设置自定义值（多文件不同值场景）

当然可以搞定！你碰到的这个场景其实挺常见的——要从不同文件导入数据，每个文件对应不同的固定relevance值，还得满足非空约束。下面给你几个实用的解决方案：

方法1：临时表中转（最直观易维护） #

这是最稳妥的方案，尤其适合新手或者需要处理复杂逻辑的场景：

1. 先创建一个临时表，只存要导入的id列：

CREATE TEMP TABLE temp_user (id varchar(36));

2. 用COPY把文件数据导入临时表：

COPY temp_user(id) FROM '/home/users_ids.txt';

3. 把临时表的数据插入正式user表，同时指定当前文件对应的relevance值：

INSERT INTO user(id, relevance)
SELECT id, 'ONE' FROM temp_user; -- 把'ONE'换成当前文件对应的实际值就行

临时表会在你当前数据库会话结束后自动清理，不用手动删，非常省心。

方法2：直接在COPY里拼接常量值（更高效） #

如果不想用临时表，可以借助PostgreSQL的COPY ... FROM PROGRAM特性，用命令行工具给每行数据追加relevance值。比如用sed给每行末尾加个制表符和目标值：

COPY user(id, relevance) 
FROM PROGRAM 'cat /home/users_ids.txt | sed ''s/$/\tONE/'''
DELIMITER E'\t'; -- 指定分隔符为制表符，和sed添加的格式对应

简单说就是sed帮你给原文件的每一行末尾加了个制表符和ONE，这样COPY就能同时读到id和relevance两列的数据了。如果你的原文件用的是其他分隔符，记得调整DELIMITER参数。

方法3：psql变量替换（适合批量脚本） #

如果你是用psql写批量导入脚本，可以用变量动态指定relevance值，结合临时表的方法会很方便：
比如在命令行运行psql时传变量：

psql -v rel_val='ONE' -f import_script.sql

然后在import_script.sql里写：

CREATE TEMP TABLE temp_user (id varchar(36));
COPY temp_user(id) FROM '/home/users_ids.txt';
INSERT INTO user(id, relevance)
SELECT id, :rel_val FROM temp_user;

这样每次导入不同文件时，只需要改rel_val的值就行，批量处理超方便。

为什么不用表级默认值？ #

你已经说到点子上了——表级默认值是全局的，没法给不同文件设置不同的relevance值，所以上面这些动态指定的方法才是更合适的选择。

内容的提问来源于stack exchange，提问作者Wall-E