For循环中的plpgsql Select语句创建多个CSV文件

Question

我想重复以下查询 8760 次，将“2”替换为 1 到 8760 一年中的每个小时。我们的想法是为每个小时创建一个单独的 CSV 文件以供进一步处理。

COPY 
(SELECT *
FROM 
  public.completedsolarirad2012
WHERE 
  completedsolarirad2012."UniquetmstmpID" = 2)
TO 'C:\temp\2012hour2.csv' WITH DELIMITER ','
CSV HEADER

我整理了以下功能（仅用了几个小时进行了测试）：

CREATE OR REPLACE FUNCTION everyhour()
  RETURNS void AS
$BODY$BEGIN
FOR i IN 0..5 LOOP
  EXECUTE $x$ 
  COPY (
    SELECT *
    FROM 
    public.completedsolarirad2012
    WHERE 
    completedsolarirad2012."UniquetmstmpID" = i
    )
   TO $concat$ 'C:\temp.' || i::text 
         || '.out.csv' WITH DELIMITER ',' CSV HEADER $concat$
    $x$;
END LOOP;
END;
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;
ALTER FUNCTION everyhour()
  OWNER TO postgres;

我似乎有两个不同的问题：

首先，我得到：

错误：“i”列不存在。

其次，当我仅通过将“i”替换为 e.g. 来测试连接语句时“2”，我得到：

错误：不允许 COPY 到文件的相对路径

我是 postgres 超级用户，所以我不明白我为什么会遇到这个问题。

注意：删除连接语句周围的 $concat$ 双引号会导致以下错误：

ERROR:  syntax error at or near "||"
LINE 9:    TO 'C:\temp.' || i::text

如果有任何帮助，我将不胜感激。

Answer 1

假设您的服务器操作系统是 Windows。

CREATE OR REPLACE FUNCTION everyhour()
  RETURNS void AS
$func$
BEGIN
FOR i IN 0..5 LOOP
   EXECUTE '
   COPY (
      SELECT *
      FROM   public.completedsolarirad2012 c
      WHERE  c."UniquetmstmpID" = ' || i || $x$)
   TO 'C:/temp.'$x$ || i || $x$'.out.csv' WITH DELIMITER ',' CSV HEADER$x$;
END LOOP;
END
$func$  LANGUAGE plpgsql;

• 你的一层美元报价太多了。
• 您还意外连接了字母“i”而不是其值。
• 使用正斜杠，即使在窗口中也是如此。或者您可能需要将反斜杠加倍，具体取决于您的设置。更多相关答案：
  PostgreSQL: export resulting data from SQL query to Excel/CSV

format() 更简单

从 Postgres 9.1 开始，您可以使用 format() 来简化复杂的连接：

CREATE OR REPLACE FUNCTION everyhour()
  RETURNS void AS
$func$
BEGIN
FOR i IN 0..5 LOOP
   EXECUTE format($x$COPY (
   SELECT *
   FROM   public.completedsolarirad2012 c
   WHERE  c."UniquetmstmpID" = %1$s)
   TO 'C:/temp.%1$s.out.csv' WITH DELIMITER ',' CSV HEADER$x$, i);
END LOOP;
END
$func$  LANGUAGE plpgsql;