postgresql - 多行子选择作为“执行使用”的参数

Question

多行子选择将用于子句中in运算符的右侧：where

create table t (a integer);
insert into t (a) values (1), (9);

drop function if exists f();

create function f()
returns void as $$
begin
execute '
    select a
    from t
    where a in $1
' using (select 1 union select 2);
end;$$
language plpgsql;

select f();

ERROR:  more than one row returned by a subquery used as an expression
CONTEXT:  SQL statement "SELECT (select 1 union select 2)"
PL/pgSQL function "f" line 3 at EXECUTE statement

如果有效，如何实现上述功能？

Answer 1

我在您的问题中看不到任何无法更容易解决的问题：

SELECT a
FROM   t
JOIN  (VALUES (1), (2)) x(a) USING (a); -- any query returning multiple int

你能澄清你的例子中的必要性吗？

---

作为概念证明，这将更简单/更快：

CREATE OR REPLACE FUNCTION x.f1()
  RETURNS SETOF integer AS
$BODY$
BEGIN
RETURN QUERY EXECUTE '
    SELECT a
    FROM t
    WHERE a = ANY($1)'
USING ARRAY(VALUES (1), (2)); -- any query here
END;
$BODY$
LANGUAGE plpgsql;

---

IN () 和 = ANY() 的性能

你的观察是正确的。这是有原因的。尝试：

EXPLAIN ANALYZE SELECT * FROM tbl WHERE id IN (1,2,3);

查询计划将显示：

索引条件：(id = ANY ('{1,2,3}'::integer[]))

PostgreSQL 将id IN (..)结构转换为id = ANY(..)内部。这两个执行相同 - 除了可以忽略不计的开销损失。

我的代码只是在构建语句时更快 - 正如您所诊断的那样。

Answer 2

create function f()
returns setof integer as $$
begin
return query execute format('
    select a
    from t
    where a in %s
', replace(replace(array(select 1 union select 2)::text, '{', '('), '}', ')'));
end$$
language plpgsql;