使用 plpgsql 控制 FOR LOOP 中的 SELECT 子句

Question

我尝试制作一个脚本来输出仅由一个用户使用的所有 foo，如果一个 foo 被多个用户使用，则不应输出。

这是我的桌子

foos  (id, value)
users (id, name)
used  (foo_id, user_id)

我的脚本不工作

FUNCTION output_unshared_foos ()
RETURNS foos AS
$a$
DECLARE
  foocounts RECORD;
BEGIN
  SELECT u.foo_id, count(*)
  INTO foocounts -- store in the local variable
  FROM used u
  GROUP BY u.foo_id;

  FOR f IN SELECT * FROM foos
  LOOP
    IF (SELECT fc.count < 2 FROM foocounts fc WHERE fc.foo_id = f.id) THEN
      RETURN NEXT f;
    END IF;
  END LOOP;
END
$a$ language plpgsql;

似乎不起作用，返回每一行，条件控制似乎总是为真。

Answer 1

您的第一个问题是您无法将返回多行的查询结果存储到单个变量中（SELECT u.foo_id, count(*) INTO ... 部分）。我很惊讶您在调用函数时没有收到运行时错误。

您的函数也无法编译，因为记录 f 未声明且定义为 returns foos 的函数无法使用 return next

但是你的方法是错误的（即使它有效）。在 SQL 中进行逐行处理几乎总是错误的选择。 SQL 和关系数据库旨在处理集合，而不是单行。

您的问题可以通过一个查询来解决：

select foo_id
from used
group by foo_id
having count(distinct user_id) = 1

将返回仅由一个用户使用的所有 foo id。

如果您需要foos 表中的其他信息，可以将上述查询加入到 foos 表中：

select f.*
from foos f
  join (
     select foo_id
     from used
     group by foo_id
     having count(distinct user_id) = 1
  ) u on f.id = u.foo_id