从 DO 状态中的查询中获取结果

Question

如何在 plpgsql 中的 IF 条件中运行 SQL 语句？我不想创建或替换函数。这是我尝试过的：

DO LANGUAGE plpgsql $$
BEGIN
IF 'Khosla' = 'Khosla' THEN
SELECT * FROM test_log limit 10;
ELSE
RAISE NOTICE 'not worked';
END IF;
END;
$$;
ERROR:  query has no destination for result data
HINT:  If you want to discard the results of a SELECT, use PERFORM instead.
CONTEXT:  PL/pgSQL function "inline_code_block" line 3 at SQL statement

我也试过这个，但无法获取数据：

DO LANGUAGE plpgsql $$
BEGIN
if 'a' = 'a' THEN
PERFORM * FROM test_log limit 10;
else
raise notice 'not worked';
end if;
end;
$$;
Output: Query returned successfully with no result in 13 ms.

正确阅读文档后，我开始知道 EXECUTE 执行 SQL 并丢弃结果集。

你能帮助运行 SQL 语句并获取结果集吗？

Answer 1

不能从 DO 语句返回值。改为创建一个 plpgsql 函数。

有几种方法可以使用 RETURNING 子句或OUT参数定义返回类型。阅读有关CREATE FUNCTION的手册。

有几种方法可以从函数返回值。阅读手册中的从函数返回一章。

特别是，由于您尝试从表中返回整行，因此可以使用表的注册类型进行函数声明：

CREATE FUNCTION foo ()
  RETURNING SETOF test_log 
$func$
BEGIN
IF 'a' = 'a' THEN
  RETURN QUERY
  SELECT * FROM test_log LIMIT 10;
ELSE
   RAISE WARNING $$Didn't work!$$;
END IF;
END
$func$ LANGUAGE plpgsql;

叫：

SELECT * FROM foo ();

或者尝试在SO上搜索。我发布了许多相关的代码示例。

DO语句的解决方法

如果不能使用函数，则使用 DO 语句的唯一明智的解决方法是使用临时表：

CREATE TEMP TABLE tbl_tmp AS
SELECT * FROM test_log LIMIT 0;
$do$
BEGIN
IF 'a' = 'a' THEN
  INSERT INTO tbl_tmp
  SELECT * FROM test_log LIMIT 10;
ELSE
   RAISE WARNING $$Didn't work!$$;
END IF;
END
$do$ LANGUAGE plpgsql;
SELECT * FROM tbl_tmp;

临时表将在会话结束时自动删除。

Answer 2

最后我做了 IT - 在 plpgsql 函数中传递参数。

CREATE OR REPLACE FUNCTION param_test_1(text)
RETURNS TABLE (comp_id int, comp_name text, comp_sort text) AS $$
BEGIN
IF 'company' = $1 THEN
RETURN QUERY SELECT id::int as comp_id, ltrim(company.compname)::text as compname, ltrim(company.compname)::text as sort FROM company where deleteflag = '0' and active = '1' and id in (270,394,376,396,403);
ELSEIF 'jobs' = $1 THEN
RAISE NOTICE 'Not Working';
END IF;
RETURN;
END;
$$ LANGUAGE plpgsql;
SELECT * FROM param_test_1('company');

这工作得很好。任何关于我的工作方法的建议将不胜感激。谢谢