检查触发函数中是否存在id并采取相应行动

Question

在触发器函数中，我想从表中选择一个id。如果这个id存在，我想用它做点什么，如果不存在，我想用它做点别的事情。

我的代码：

CREATE OR REPLACE FUNCTION public.handle_new_user()
    RETURNS TRIGGER
    LANGUAGE plpgsql
    SECURITY DEFINER
AS
$function$
DECLARE
    _organisation_id INT;
BEGIN
    SELECT id FROM public.organisation WHERE $CONDITION
    RETURNING id
        INTO _organisation_id;

    // check if _organisation_id is set
    RETURN new;
END;
$function$

我可以将 id 存储到

_organisation_id

中吗？如果没有 id 会怎样？声明为

INT

是否正确？或者我是否需要明确声明它可以是

INT

或 Null 或其他什么？

如何检查id是否设置？如何在 PL/pgSQL 中执行此 if/else 结构？

当新用户注册我的supabase项目时，触发函数被调用。这是触发器：

-- trigger the function every time a user is created
create trigger on_auth_user_created
  after insert on auth.users
  for each row execute procedure public.handle_new_user();

我的

organisation.id

是

bigint

。

Answer 1

你想要类似的东西：

CREATE OR REPLACE FUNCTION public.handle_new_user()
  RETURNS trigger
  LANGUAGE plpgsql AS   -- why SECURITY DEFINER ?
$func$
DECLARE
   _org_id bigint;  -- matching type!
BEGIN
   SELECT o.id
   FROM   public.organisation o
   WHERE  -- $CONDITION  -- condition matters!
   INTO   _org_id;

   IF FOUND THEN
      -- do something
   ELSE
      -- do something else
   END IF;

   RETURN null;  -- irrelevant for AFTER trigger
END
$func$;