EXECUTE的查询字符串参数为null

时间:2016-06-29 22:06:52

标签: postgresql plpgsql dynamic-sql psql

修改

似乎我的问题是当这个select语句返回null时(我试图处理的情况 - 当它返回null时,我希望我的新值为-999)。如果在找到空值时出错,我怎么能这样做呢?

ORIGINAL

我已经阅读了有关此错误的所有其他SO帖子,但其中没有一个似乎解决了我的问题的根源。

错误非常简单 - 我的EXECUTE语句中的一个参数为null。大。但是,我在调用之前打印出构成我的EXECUTE语句的每个值,并且我可以清楚地看到没有值为null。

代码:

CREATE FUNCTION inform_icrm_prob_flow_query(tablename text, location_id int,
                                            product_date_str text, lead_time_start int,
                                            lead_time_end int, first_member_id int,
                                            last_member_id int, dest_file text)
RETURNS void AS $$
DECLARE
  count int;
  product_date TIMESTAMPTZ;
  interval_lead_time_start text;
  interval_lead_time_end text;
  curr_value double precision;
  query text;
BEGIN
  product_date := product_date_str::TIMESTAMPTZ;
  count := first_member_id;
  curr_value := 0;

  interval_lead_time_start :=  ''''|| product_date ||'''::timestamptz +
                               interval '''||lead_time_start||' hours''';
  interval_lead_time_end :=  ''''|| product_date ||'''::timestamptz +
                             interval '''||lead_time_end||' hours'' -
                             interval ''6 hours''';

  --create our temporary table and populate it's date column
  EXECUTE 'CREATE TEMPORARY TABLE temp_table_icrm_prob_flow AS 
             SELECT * FROM generate_series('||interval_lead_time_start || ',' || 
                                            interval_lead_time_end || ', ''6 hours'') 
                           AS date_valid';
  LOOP
    EXIT WHEN count > last_member_id;
    IF NOT EXISTS(
      SELECT 'date_valid'
      FROM information_schema.columns
      WHERE table_name='temp_table_icrm_prob_flow'
        and column_name='value'||count||'') 
    THEN
      EXECUTE 'ALTER TABLE temp_table_icrm_prob_flow ADD COLUMN value' || count 
               || ' double precision DEFAULT -999';
    END IF;

    raise notice 'tablename: %', tablename;
    raise notice 'location_id: %', location_id;
    raise notice 'product_date: %', product_date;
    raise notice 'count: %', count;

    query := 'SELECT value FROM '|| tablename ||' 
              INNER JOIN temp_table_icrm_prob_flow
              ON (temp_table_icrm_prob_flow.date_valid = '|| tablename ||'.date_valid)
              WHERE '|| tablename ||'.id_location = '|| location_id ||'
                AND '|| tablename ||'.date_product = '''|| product_date ||'''
                AND '|| tablename ||'.id_member = '|| count ||'';

    EXECUTE query INTO curr_value;

    EXECUTE 'UPDATE temp_table_icrm_prob_flow 
             SET value'|| count ||' = COALESCE('|| curr_value ||', -999)';

    count := count + 1;
  END LOOP;

  EXECUTE 'ALTER TABLE temp_table_icrm_prob_flow DROP COLUMN date_valid';

  EXECUTE 'COPY temp_table_icrm_prob_flow TO '''||dest_file||''' DELIMITER '','' CSV';

  EXECUTE 'DROP TABLE temp_table_icrm_prob_flow';
END;
$$ LANGUAGE plpgsql;

输出:

NOTICE:  tablename: inform_tseries_data_basin_proc_fcst_prob_flow
NOTICE:  location_id: 38
NOTICE:  product_date: 2015-02-05 12:00:00+00
NOTICE:  count: 1
ERROR:  query string argument of EXECUTE is null
CONTEXT:  PL/pgSQL function inform_icrm_prob_flow_query(text,integer,text,integer,integer,integer,integer,text) line 38 at EXECUTE

如果我传入的变量都不为null,并且引用的唯一其他内容是我知道的临时表,那么可能导致此错误的原因是什么?

注意:将我的查询更改为:

query := 'SELECT value FROM '|| tablename ||' WHERE '|| tablename ||'.id_location = '|| location_id ||' AND '|| tablename ||'.date_product = '''|| product_date ||''' AND '|| tablename ||'.id_member = '|| count ||' AND temp_table_icrm_prob_flow.date_va    lid = '|| tablename ||'.date_valid';

我收到以下错误:

NOTICE:  tablename: inform_tseries_data_basin_proc_fcst_prob_flow
NOTICE:  location_id: 38
NOTICE:  product_date: 2015-02-05 12:00:00+00
NOTICE:  count: 1
ERROR:  missing FROM-clause entry for table "temp_table_icrm_prob_flow"
LINE 1: ..._data_basin_proc_fcst_prob_flow.id_member = 1 AND temp_table...
                                                         ^
QUERY:  SELECT value FROM inform_tseries_data_basin_proc_fcst_prob_flow WHERE inform_tseries_data_basin_proc_fcst_prob_flow.id_location = 38 AND inform_tseries_data_basin_proc_fcst_prob_flow.date_product = '2015-02-05 12:00:00+00' AND inform_tseries_data_basin_proc_fcst_prob_flow.id_member = 1 AND temp_table_icrm_prob_flow.date_valid = inform_tseries_data_basin_proc_fcst_prob_flow.date_valid
CONTEXT:  PL/pgSQL function inform_icrm_prob_flow_query(text,integer,text,integer,integer,integer,integer,text) line 35 at EXECUTE

1 个答案:

答案 0 :(得分:6)

很抱歉小的offtopic。您的代码非常难以理解(并且SQL注入易受攻击)。您可以使用一些技巧:

  1. 使用EXECUTE语句的USING条作为常规参数。

    DO $$
    DECLARE
      tablename text := 'mytab';
      from_date date := CURRENT_DATE;
    BEGIN
       EXECUTE 'INSERT INTO ' || quote_ident(tablename) || ' VALUES($1)'
          USING from_date;
    END
    $$;
    

    此代码将是安全的(由于使用quote_ident函数),速度稍快一些(由于使用了 from_date 变量的二进制值 - 删除了多个字符串< - >日期转换和一点点更具可读性(因为字符串表达式更短)。

  2. 使用函数format。构建查询字符串将更短且更易读(表别名也有帮助):

    query := format('
    SELECT value
       FROM %I _dtn
            INNER JOIN temp_table_icrm_prob_flow t ON t.date_valid = _dtn.date_valid
      WHERE _dtn.id_location = $1
        AND _dtn.date_product = $2
        AND _dtd.id_member = $3'
                     , tablename);
     EXECUTE query INTO curr_value USING location_id, product_date, count;
    
  3. 使用名为重要SQL关键字和标识符的变量是错误的想法 - 名称​​计数是错误的。

    错误消息是干净的 - 您使用的是标识符temp_table_icrm_prob_flow.date_valid,但查询中未提及表 temp_table_icrm_prob_flow 。查询缺少JOIN部分。