如何在PL/pgSQL中使用\copy并捕获异常？求替代方案

首先明确：\copy是psql客户端专属命令，不能在PL/pgSQL（服务器端存储过程/匿名块）中直接调用——PL/pgSQL运行在PostgreSQL服务器进程内，而\copy是由psql客户端执行的本地文件读写操作，两者运行上下文完全独立。

针对你的需求（无需超级权限、可捕获异常、实现CSV数据导入），提供以下可行替代方案：

方案1：拆分\copy与PL/pgSQL，在客户端处理异常 #

把\copy和后续插入操作拆分为两步，在psql脚本中用客户端逻辑处理异常：

-- 先执行\copy，客户端直接处理文件读写，失败则终止后续操作
\copy tab1 FROM 'C:\\Documents\test.csv' DELIMITER E'\t' CSV HEADER;

-- 再调用PL/pgSQL块处理插入，捕获插入阶段的异常
DO $$
BEGIN
  INSERT INTO tab2 SELECT * FROM tab1;
EXCEPTION
  WHEN OTHERS THEN
    RAISE EXCEPTION '插入操作失败：%', SQLERRM;
END $$;

若\copy执行失败，psql会直接返回错误，PL/pgSQL块不会启动；若\copy成功，则执行插入并捕获异常。

方案2：用pg_read_file+COPY FROM STDIN（需特定权限，无需超级用户） #

如果CSV文件能被PostgreSQL服务器访问（注意是服务器端路径，非客户端本地），可通过pg_read_file读取文件内容，结合COPY FROM STDIN实现导入，同时在PL/pgSQL中捕获异常：

DO $$
DECLARE
  file_content text;
BEGIN
  -- 读取服务器端文件，需当前用户拥有pg_read_server_files角色权限
  file_content := pg_read_file('/path/to/server/test.csv', 0, 1000000);
  
  -- 使用COPY FROM STDIN，无需超级用户权限
  EXECUTE format('COPY tab1 FROM STDIN DELIMITER E''\t'' CSV HEADER') USING file_content;
  
  INSERT INTO tab2 SELECT * FROM tab1;
EXCEPTION
  WHEN OTHERS THEN
    RAISE EXCEPTION '导入或插入失败：%', SQLERRM;
END $$;

注：管理员可通过GRANT pg_read_server_files TO your_user;为用户授予所需权限。

方案3：外部程序+参数化查询（适配客户端本地文件场景） #

若CSV在客户端本地且无法让服务器访问，可通过Python/Shell等外部程序读取本地文件，批量插入数据库，同时在程序中处理异常，再调用PL/pgSQL完成后续逻辑。示例用Python的psycopg2库：

import psycopg2

try:
    conn = psycopg2.connect("dbname=your_db user=your_user password=your_pass")
    cur = conn.cursor()

    # 读取本地CSV并批量插入tab1
    with open('C:\\Documents\\test.csv', 'r') as f:
        next(f)  # 跳过表头
        cur.copy_from(f, 'tab1', sep='\t')
    
    # 调用PL/pgSQL块处理tab2插入
    cur.execute("""
        DO $$
        BEGIN
          INSERT INTO tab2 SELECT * FROM tab1;
        EXCEPTION
          WHEN OTHERS THEN
            RAISE EXCEPTION '插入失败：%', SQLERRM;
        END $$;
    """)
    
    conn.commit()
except Exception as e:
    conn.rollback()
    print(f"错误：{e}")
finally:
    cur.close()
    conn.close()

这种方式彻底绕开服务器端文件权限限制，异常处理在客户端完成，还能灵活控制数据导入逻辑。

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