非超级用户在PL/PgSQL循环中导出CSV失败求助

首先得明确你遇到的报错根源：COPY TO 'C:\Path.csv'这种写法是服务器端COPY操作，它有两个硬限制：

1. 需要超级用户权限（因为要直接读写数据库服务器的文件系统）；
2. PL/pgSQL的运行环境根本不支持这种客户端方向的COPY——你用RAISE NOTICE能正常输出变量，是因为这是PL/pgSQL原生支持的日志输出，和COPY的执行逻辑完全不搭。

下面给你几个实用的解决方案，都是不需要超级权限就能跑的：

方案1：用psql的\copy结合\gexec自动循环（最推荐） #

psql的\copy是客户端侧的COPY操作，它不需要超级权限，而且能直接把数据导出到你本地的文件系统。结合\gexec命令，我们可以自动生成并执行每个kommunekode的导出命令：

-- 连接到你的数据库后，执行这段SQL
WITH unique_komkodes AS (
    SELECT DISTINCT komkode 
    FROM kommunekoder 
    ORDER BY komkode
)
SELECT format(
    '\copy (SELECT * FROM bbr.co40100t_geo WHERE komkode = %L) TO %L WITH CSV DELIMITER '',''',
    komkode,
    'C:\Path_' || komkode || '.csv'  -- 每个kommunekode生成单独的CSV，避免覆盖
)
FROM unique_komkodes \gexec

这里的关键点：

• format函数用%L自动处理字符串转义，避免SQL注入风险；
• \gexec会把查询结果的每一行当作psql命令执行，自动完成所有kommunekode的导出；
• 给每个CSV加上kommunekode后缀，防止文件被覆盖（你原来的代码每次都写同一个文件，会导致数据被覆盖）。

方案2：用shell/PowerShell循环配合psql（适合自动化脚本） #

如果你需要把导出逻辑做成脚本，可以先提取所有唯一的kommunekode，再循环执行导出命令：

Windows PowerShell示例： #

# 替换成你的数据库名称
$dbName = "your_database_name"

# 获取所有唯一的kommunekode
$komkodes = psql -d $dbName -t -c "SELECT DISTINCT komkode FROM kommunekoder ORDER BY komkode"

# 循环导出每个kommunekode的数据
foreach ($k in $komkodes) {
    $cleanKode = $k.Trim()
    if ($cleanKode) {
        $outputPath = "C:\Path_$cleanKode.csv"
        psql -d $dbName -c "\copy (SELECT * FROM bbr.co40100t_geo WHERE komkode = '$cleanKode') TO '$outputPath' WITH CSV DELIMITER ','"
    }
}

Linux/macOS Bash示例： #

DB_NAME="your_database_name"

# 获取所有唯一的kommunekode，循环导出
psql -d $DB_NAME -t -c "SELECT DISTINCT komkode FROM kommunekoder ORDER BY komkode" | while read komkode; do
    komkode=$(echo $komkode | xargs)  # 去除空格换行
    if [ -n "$komkode" ]; then
        output_path="/path/to/your/Path_$komkode.csv"
        psql -d $DB_NAME -c "\copy (SELECT * FROM bbr.co40100t_geo WHERE komkode = '$komkode') TO '$output_path' WITH CSV DELIMITER ','"
    fi
done

方案3：避免固定循环次数的坑 #

你原来的代码里用FOR loop_counter IN 1..99 LOOP来遍历，但如果kommunekoder表中的唯一komkode数量不是99，要么会漏掉数据，要么会因为找不到对应row_number而报错。上面的方案都是直接遍历所有唯一的komkode，完全规避了这个问题。

内容的提问来源于stack exchange，提问作者asguldbrandsen