使用PL / pgSQL将查询结果存储在变量中

如何将查询结果分配给PL / pgSQL（PostgreSQL的过程语言）中的变量？

我有一个功能：

CREATE OR REPLACE FUNCTION test(x numeric)
RETURNS character varying AS
$BODY$
DECLARE
name   character varying(255);
begin
 name ='SELECT name FROM test_table where id='||x;

 if(name='test')then
  --do somthing
 else
  --do the else part
 end if;
end;
return -- return my process result here
$BODY$
LANGUAGE plpgsql VOLATILE

在上面的函数中，我需要存储此查询的结果：

'SELECT name FROM test_table where id='||x;

到变量name。

如何处理呢？

我认为您正在寻找SELECT INTO：

select test_table.name into name from test_table where id = x;

这将拉动name从test_table哪里id是你的函数的参数，让它在name变。不要遗漏表名前缀，test_table.name否则您会抱怨引用不明确。

只要分配单个变量，就可以在plpgsql函数中使用普通分配：

name := (SELECT t.name from test_table t where t.id = x);

或使用SELECT INTO像@mu已经提供。

这也起作用：

name := t.name from test_table t where t.id = x;

但是最好使用前两种更清晰的方法之一，如@Pavel所评论。

我另外使用表别名缩短了语法。
更新：我删除了我的代码示例，并建议使用IF EXISTS()，而不是像由@Pavel提供。

通常的模式是EXISTS(subselect)：

BEGIN
  IF EXISTS(SELECT name
              FROM test_table t
             WHERE t.id = x
               AND t.name = 'test')
  THEN
     ---
  ELSE
     ---
  END IF;

此模式用于PL / SQL，PL / pgSQL，SQL / PSM，...

创建学习表：

CREATE TABLE "public"."learning" (
    "api_id" int4 DEFAULT nextval('share_api_api_id_seq'::regclass) NOT NULL,
    "title" varchar(255) COLLATE "default"
);

插入数据学习表：

INSERT INTO "public"."learning" VALUES ('1', 'Google AI-01');
INSERT INTO "public"."learning" VALUES ('2', 'Google AI-02');
INSERT INTO "public"."learning" VALUES ('3', 'Google AI-01');

步骤：01

CREATE OR REPLACE FUNCTION get_all (pattern VARCHAR) RETURNS TABLE (
        learn_id INT,
        learn_title VARCHAR
) AS $$
BEGIN
    RETURN QUERY SELECT
        api_id,
        title
    FROM
        learning
    WHERE
        title = pattern ;
END ; $$ LANGUAGE 'plpgsql';

步骤：02

SELECT * FROM get_all('Google AI-01');

步骤：03

DROP FUNCTION get_all();

演示：

您可以使用以下示例使用PL / pgSQL将查询结果存储在变量中：

 select * into demo from maintenanceactivitytrack ; 
    raise notice'p_maintenanceid:%',demo;