首页 > 数据库 > mysql教程 > 如何正确定义返回记录或复合类型的 PostgreSQL 函数?

如何正确定义返回记录或复合类型的 PostgreSQL 函数?

Barbara Streisand
发布: 2025-01-05 10:57:40
原创
680 人浏览过

How to Correctly Define PostgreSQL Functions Returning Record or Composite Types?

PostgreSQL:错误:42601:返回“记录”的函数需要列定义列表

错误“错误:42601:需要列定义列表返回“record”的函数表示声明返回记录类型的函数在其返回值中没有指定列定义列表

返回选定的列

CREATE OR REPLACE FUNCTION get_user_by_username(_username text
                                              , _online bool DEFAULT false)
  RETURNS TABLE (
    user_id int
  , user_name varchar
  , last_activity timestamptz
  )
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF _online THEN
      RETURN QUERY
      UPDATE users u 
      SET    last_activity = current_timestamp  -- ts with time zone
      WHERE  u.user_name = _username
      RETURNING u.user_id
              , u.user_name
              , u.last_activity;
   ELSE
      RETURN QUERY
      SELECT u.user_id
           , u.user_name
           , u.last_activity
      FROM   users u
      WHERE  u.user_name = _username;
   END IF;
END
$func$;
登录后复制

调用:

SELECT * FROM get_user_by_username('myuser', true);
登录后复制

返回(一组)整行

返回所有现有表用户的列,有一个更简单的方法。 Postgres 自动为每个表定义一个同名的复合类型。只需使用 RETURNS SETOF 用户即可大大简化查询:

CREATE OR REPLACE FUNCTION get_user_by_username(_username text
                                              , _online bool DEFAULT false)
  RETURNS SETOF users
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF _online THEN
      RETURN QUERY
      UPDATE users u 
      SET    last_activity = current_timestamp
      WHERE  u.user_name = _username
      RETURNING u.*;
   ELSE
      RETURN QUERY
      SELECT *
      FROM   users u
      WHERE  u.user_name = _username;
   END IF;
END
$func$;
登录后复制

返回整行加上自定义添加

CREATE OR REPLACE FUNCTION get_user_by_username3(_username text
                                               , _online bool DEFAULT false)
  RETURNS TABLE (
    users_row users
  , custom_addition text
  )
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF _online THEN
      RETURN QUERY
      UPDATE users u 
      SET    last_activity = current_timestamp  -- ts with time zone
      WHERE  u.user_name = _username
      RETURNING u  -- whole row
              , u.user_name || u.user_id;
   ELSE
      RETURN QUERY
      SELECT u, u.user_name || u.user_id
      FROM   users u
      WHERE  u.user_name = _username;
   END IF;
END
$func$;
登录后复制

“神奇”在于函数调用中,我们(可选)分解行类型:

SELECT (users_row).*, custom_addition FROM get_user_by_username('foo', true);
登录后复制

以上是如何正确定义返回记录或复合类型的 PostgreSQL 函数?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板