主页 / dba / 问题 / 134082
Accepted
dw8547
Asked: 2016-04-02 05:01:20 +0800 CST

循环字符串文字作为函数 PostgreSQL 9.4 的输入

  • 772

我有一个update_total看起来像这样的函数：

CREATE OR REPLACE FUNCTION update_total (table_name CHARACTER VARYING, year CHARACTER(4), donor_type CHARACTER VARYING, donor_code CHARACTER(5)) RETURNS VOID AS $$

DECLARE

table_name ALIAS FOR $1;
year ALIAS FOR $2;
donor_type ALIAS FOR $3;
donor_code ALIAS FOR $4;
query_statement TEXT;

BEGIN

    query_statement :=
    '
    UPDATE gha.' || table_name  || '
    SET "' || year || '" = "' || year ||'_total"
    FROM gha.' || table_name || ' "' || year || '",
    (
        SELECT
        donor_type
        , COALESCE(SUM("' || year || '"), 0) AS "' || year ||'_total"
        FROM gha.' || table_name || '
        WHERE donor_type = ''' || donor_type || '''
        GROUP BY donor_type
    ) "new_data"
    WHERE
    gha.' || table_name || '.donor_code = ' || CAST(donor_code AS INT) || ';'
    ;

    EXECUTE query_statement;

END;
$$ LANGUAGE plpgsql;

它更新表中包含聚合值的行，我称它为，例如，如下所示：

SELECT update_total ('bilateral_oda_dac_1', '1990', 'Multilateral', '20002');

它更新的表设置为每年都有一个范围内的列，即：

   Column   |       Type        
------------+-------------------
 donor_code | smallint           
 donor_name | character varying 
 donor_type | character varying  
 1990       | numeric            
 1991       | numeric      
 1992       | numeric  
...

使用此功能，我一次只能更新一列中的聚合值，但我希望能够一次更新一系列列，大致如下：

CREATE FUNCTION update_all_total ()
    RETURNS void LANGUAGE plpgsql AS

$BODY$
BEGIN

FOR i IN 1990 .. 1995
LOOP
   PERFORM update_total ('bilateral_oda_dac_1', i, 'DAC', '20001');
END LOOP;

END;
$BODY$;

（我用这篇文章让我开始：https ://stackoverflow.com/questions/11164409/sql-call-function-multiple-times-in-a-loop-postgres-8-3 ）。上面截断的代码不起作用，因为我正在使用的函数将字符串作为输入，并且我向它传递了一个整数。我如何设置循环以便将字符串传递给我在其中调用的函数？

postgresql-9.4 functions

1 个回答

  1. Best Answer

    你可以使用to_char()，

    创建函数 update_all_total () RETURNS void LANGUAGE plpgsql AS
$身体$
开始
对于我在 1990 .. 1995
环形
   PERFORM update_total ('bilateral_oda_dac_1', to_char(i, '9999'), 'DAC', '20001');
结束循环；
结尾;
$身体$；

    或者可以使用 cast ::char(4)。

    创建函数 update_all_total () RETURNS void LANGUAGE plpgsql AS
$身体$
开始
对于我在 1990 .. 1995
环形
   PERFORM update_total ('bilateral_oda_dac_1', i::char(4), 'DAC', '20001');
结束循环；
结尾;
$身体$；
    • 3

相关问题