PostgreSQL：如何在函数的 CASE 表达式中使用 SELECT 语句？

Question

我们可以在函数的 CASE 条件表达式中使用 SELECT 语句吗？
我尝试了以下功能来完成上述任务。

示例：我有一个函数用于显示包含一些行的表格。

create or replace function test(n integer) 
returns table (name text,city text) as
$body$
begin
     case n when 1 then
     select * from table1

     when 2 then
     select * from table2

     when 3 then
     select * from view1

     end;
end;
$body$
language plpgsql;

--调用函数

select * from test(1);  
/*have to show details of table1*/

select * from test(2);
/*have to show details of table2*/

select * from test(3);
/*have to display details of view1*/

Answer 1

其实有一个plpgsql CASE statement，不要和SQL CASE expression混淆：

CREATE OR REPLACE function test(n integer) 
  RETURNS TABLE (name text, city text) AS
$func$
BEGIN

CASE n
WHEN 1 THEN
    RETURN QUERY SELECT t.name, t.city FROM table1 t;
WHEN 2 THEN
    RETURN QUERY SELECT t.foo, t.bar   FROM table2 t;
WHEN 3 THEN
    RETURN QUERY SELECT t.bar, t.bamm  FROM view1 t;
END CASE;

END
$func$ LANGUAGE plpgsql;

• 如果您将函数声明为 RETURNS TABLE (name text, city text)，那么您的 SELECT 语句应该有一个匹配类型的列列表。
  另一方面，如果您想SELECT *，请相应地将函数声明为RETURNS SETOF table1。

• 在返回类型中命名列时，这些变量在函数体中可见。确保对可能与相同名称冲突的列名进行表限定。所以t.name 而不仅仅是name。

• 函数内部查询的列名在外部不可见。只有声明的返回类型。所以名称不必匹配，只需匹配数据类型即可。

无论哪种方式，我建议简化一些事情：

CREATE OR REPLACE function test(n integer) 
  RETURNS SETOF table1 AS
$func$

SELECT * FROM table1 t WHERE n = 1
UNION ALL
SELECT * FROM table2 t WHERE n = 2
UNION ALL
SELECT * FROM view1  t WHERE n = 3;

$func$ LANGUAGE sql;

同样的结果。一样快。 SQL 或 PL/pgSQL 函数是个人喜好和其他一些细节问题。 PL/pgSQL 对于独立调用可能更快。 SQL 更容易嵌套。

Answer 2

为此，您需要一个IF 声明和RETURN QUERY，例如

create or replace function test(n integer) 
returns table (name text,city text) as
$body$
begin
     IF n = 1 THEN
         RETURN QUERY select * from table1;
     ELIF n = 2 THEN
         RETURN QUERY select * from table2;
     ELIF n = 3 THEN
         RETURN QUERY select * from view1;
     END IF;
end;
$body$
language plpgsql;

不过，这是一件非常奇怪的事情。