声明变量set = select

时间:2013-12-11 11:55:51

标签: postgresql variables set plpgsql declare

这可能听起来像是一个非常愚蠢的问题,但是我如何声明一个用于PostgreSQL 9.3查询的变量?

CREATE or replace FUNCTION public.test()
  returns int4
AS
$BODY$
DECLARE    
    cod_process bigint :=30001;
    cod_instance bigint ;
    utc_log timestamp without time zone := localtimestamp;
    cod_log_type varchar(100) :='information ';
    txt_log_text varchar(100):= 'start process';
    txt_log varchar(100):= txt_log_text||'_'||cod_process;
    set cod_instance= select max(cod_instance) as cod_instance from public.instance where public.instance.cod_process=cod_process;    
BEGIN  
    INSERT INTO public.log (cod_process, cod_instance, utc_log,cod_log_type,txt_log)
    VALUES (cod_process, cod_instance, utc_log,cod_log_type,txt_log );
    RETURN 11;
END;
$BODY$ LANGUAGE 'plpgsql';
ERROR: type "cod_instance" does not exist
SQL state: 42704
Character: 383

2 个答案:

答案 0 :(得分:19)

您需要使用实际代码块内的into子句运行select,而不是declare块:

begin
   select max(cod_instance) 
      into cod_instance
   from public.instance 
   where public.instance.cod_process=cod_process;

   ....
end;

给变量(或参数)提供与表中列相同的名称通常不是一个好主意。在某些情况下,这可能会使解析器混淆。为避免任何潜在问题,请尝试为变量使用不同的名称,例如通过为它们添加前缀(例如l_cod_process代替cod_processl_cod_instance代替cod_instance

有关变量赋值的更多详细信息,请参见手册:http://www.postgresql.org/docs/current/static/plpgsql-statements.html

答案 1 :(得分:19)

你的演示功能会像这样工作:

CREATE or replace FUNCTION public.test()
  RETURNS int4 AS
$func$
DECLARE
   _cod_process  bigint := 30001;
   _cod_instance bigint := (SELECT max(cod_instance)
                            FROM   public.instance
                            WHERE  cod_process = _cod_process);
   _utc_log      timestamp := localtimestamp;
   _cod_log_type varchar(100) := 'information';
   _txt_log_text varchar(100) := 'start process';
   _txt_log      varchar(100) := txt_log_text || '_' || cod_process;
BEGIN
   INSERT INTO public.log
          ( cod_process,  cod_instance,  utc_log,  cod_log_type,  txt_log)
   VALUES (_cod_process, _cod_instance, _utc_log, _cod_log_type, _txt_log);

   RETURN 11;
END
$func$ LANGUAGE plpgsql;

重点

  • 无法使用SET分配变量。这被设置为用于设置运行时参数的SQL command SET

  • 可以在声明时分配变量,甚至可以使用子查询。

  • 使用LANGUAGE plpgsql,而不是 LANGUAGE 'plpgsql' 。这是一个标识符。

  • @a_horse_with_no_name已经写过关于命名冲突的文章。

  • 调试代码时,使用干净的格式有很长的路要走......


但你可以简化为:

CREATE OR REPLACE FUNCTION public.test(_cod_process bigint = 30001)
  RETURNS integer AS
$func$
   INSERT INTO public.log
   (cod_process, cod_instance     , utc_log, cod_log_type , txt_log)
   SELECT    $1, max(cod_instance), now()  , 'information', 'start process_' || $1
   FROM   public.instance
   WHERE  cod_process = $1
   GROUP  BY cod_process
   RETURNING 11
$func$ LANGUAGE sql;

呼叫:

SELECT public.test();     -- for default 30001
SELECT public.test(1234);

根据utc_log的实际数据类型,您可能需要now() AT TIME ZONE 'UTC'