如何在PL/pgSQL中使用COPY命令前检测源文件是否为空？

当然可以解决这个问题！在PL/pgSQL中检测磁盘文件是否为空，结合你Windows Server 2012 R2 + PostgreSQL 10.3的环境，这里有两种可靠的方案帮你避免空文件导致的导入冻结：

方案1：使用PostgreSQL自带的pg_stat_file函数（推荐） #

PostgreSQL内置的pg_stat_file函数可以获取文件的元数据（包括大小、修改时间等），完全跨平台，不需要依赖外部工具，是最简洁的方法。

步骤1：创建检测空文件的函数 #

这个函数会返回文件是否为空，同时处理文件不存在或无权限访问的异常情况：

CREATE OR REPLACE FUNCTION is_file_empty(p_file_path text)
RETURNS boolean AS $$
BEGIN
    -- 查询文件大小，若大小为0则返回true（空文件）
    RETURN (SELECT size = 0 FROM pg_stat_file(p_file_path));
EXCEPTION
    WHEN others THEN
        -- 处理文件不存在、无权限等异常，可根据需求调整逻辑
        RAISE NOTICE '无法访问文件 "%"：%', p_file_path, SQLERRM;
        RETURN true; -- 视为无法处理的文件，后续跳过导入
END;
$$ LANGUAGE plpgsql;

步骤2：在导入循环中集成检测 #

修改你现有的PL/pgSQL导入循环，在执行COPY命令前先检查文件是否为空，为空则跳过：

-- 假设你通过数组或查询获取所有待导入的文件名
FOR file_name IN SELECT unnest(ARRAY['file_01.txt', 'file_02.txt', ..., 'file_44.txt']) LOOP
    DECLARE
        v_full_file_path text := 'C:\\your_data_directory\\' || file_name; -- Windows路径需用双反斜杠转义
    BEGIN
        -- 检查文件是否为空或无法访问
        IF is_file_empty(v_full_file_path) THEN
            RAISE NOTICE '跳过空文件或无法访问的文件：%', v_full_file_path;
            CONTINUE; -- 跳过当前文件，继续下一个循环
        END IF;

        -- 执行正常的COPY导入逻辑
        EXECUTE format(
            'COPY your_target_table FROM %L WITH (FORMAT text, DELIMITER ''\t'', HEADER false)',
            v_full_file_path
        );
        RAISE NOTICE '成功导入文件：%', v_full_file_path;
    END;
END LOOP;

方案2：调用Windows命令行工具（备选） #

如果需要更复杂的文件检查逻辑，可以通过调用Windows的PowerShell或CMD命令来获取文件大小，再在PL/pgSQL中处理结果。需要注意的是，此方法需要PostgreSQL超级用户权限，且服务账号需有执行命令的权限。

示例代码（通过PowerShell获取文件大小）：

CREATE OR REPLACE FUNCTION get_file_size(p_file_path text)
RETURNS bigint AS $$
DECLARE
    v_size bigint;
BEGIN
    -- 调用PowerShell获取文件大小（字节）
    COPY (
        SELECT * FROM program('powershell -Command "(Get-Item ''' || p_file_path || ''').Length"')
    ) TO temp TABLE temp_size;
    
    SELECT CAST(column1 AS bigint) INTO v_size FROM temp_size;
    DROP TABLE temp_size;
    RETURN v_size;
EXCEPTION
    WHEN others THEN
        RAISE NOTICE '获取文件大小失败：%', SQLERRM;
        RETURN 0;
END;
$$ LANGUAGE plpgsql;

之后你可以用get_file_size(v_full_file_path) = 0来判断是否为空文件，逻辑和方案1类似。

关键注意事项 #

• 权限问题：确保PostgreSQL服务运行的账号（默认是NT SERVICE\PostgreSQL）拥有待导入文件所在目录的读取权限，否则pg_stat_file或命令行工具会无法访问文件。
• 路径转义：Windows系统中的文件路径需要用双反斜杠（\\）转义，避免被PL/pgSQL解析为转义字符。
• 异常处理：根据你的业务需求调整异常处理逻辑，比如是否抛出错误终止循环，还是仅跳过异常文件。

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