PostgreSQL中用存储过程导入CSV可行吗？批量处理同结构文件

问题说明 #

现有10个列结构完全一致的CSV文件需导入PostgreSQL的Sales表，目前已可通过COPY语句实现单个文件导入，但导师要求使用存储过程完成导入操作。需明确两个问题：

1. 是否可通过存储过程实现CSV文件导入？
2. 仅使用COPY语句是否满足需求？

建表语句 #

CREATE TABLE Sales(
    OrderID INTEGER,
    Product VARCHAR(100),
    Quantity INTEGER,
    PriceEach NUMERIC(10,2),
    OrderDate TIMESTAMP,
    PurchaseAddress VARCHAR(100)
);

单个CSV导入的COPY语句 #

-- 从CSV文件复制记录
COPY Sales(OrderID,Product,Quantity,PriceEach,OrderDate,PurchaseAddress)
FROM 'C:\sampledb\Sales.csv'
DELIMITER ','
CSV HEADER;

问题解答 #

1. 能否通过存储过程实现CSV导入？ #

完全可以。PostgreSQL的存储过程支持动态执行SQL，你可以编写存储过程批量处理多个CSV文件，比如循环遍历指定目录下的目标文件，或接收文件路径列表作为参数，逐一执行COPY操作。

以下是批量导入的示例存储过程：

CREATE OR REPLACE PROCEDURE import_sales_csvs(file_paths TEXT[])
LANGUAGE plpgsql
AS $$
DECLARE
    file_path TEXT;
    copy_sql TEXT;
BEGIN
    FOREACH file_path IN ARRAY file_paths LOOP
        -- 构造动态COPY语句
        copy_sql := format(
            'COPY Sales(OrderID,Product,Quantity,PriceEach,OrderDate,PurchaseAddress) 
             FROM %L DELIMITER '','' CSV HEADER;',
            file_path
        );
        -- 执行动态SQL
        EXECUTE copy_sql;
        -- 可选：输出导入日志
        RAISE NOTICE '已成功导入文件: %', file_path;
    END LOOP;
END;
$$;

调用方式：

-- 传入10个CSV文件的路径数组
CALL import_sales_csvs(
    ARRAY[
        'C:\sampledb\Sales_1.csv',
        'C:\sampledb\Sales_2.csv',
        -- 依次添加剩余8个文件路径
        'C:\sampledb\Sales_10.csv'
    ]
);

2. 仅使用COPY语句是否满足需求？ #

从功能上看，仅用COPY语句可以完成导入需求——你可以手动逐个执行10次COPY语句，将每个CSV文件导入表中。但这种方式存在明显不足：

• 需要手动重复执行相同逻辑的代码，效率低且易出错；
• 无法实现自动化批量处理，不符合导师要求用存储过程实现的核心目标（通常是为了规范操作、自动化批量任务）。

因此，虽然COPY本身能完成单个文件的导入，但面对批量文件场景，用存储过程封装逻辑更符合工程化要求，也满足导师的要求。

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