news 2026/3/27 18:42:24

postgresql存储过程如何写

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
postgresql存储过程如何写

在 PostgreSQL 里,其实 99% 场景你写的是 函数(FUNCTION);

真正的 存储过程(PROCEDURE) 是 11 版才引入,语法几乎一样,只是调用方式不同。

下面把“函数”和“过程”都给你一次讲透,并附完整可直接 `psql -f` 运行的脚本。

---

一、先分清:FUNCTION vs PROCEDURE

特性 FUNCTION PROCEDURE
最早版本 一直就有 ≥ 11
能否回传结果集 ✅ `RETURNS TABLE / SETOF` ❌(只能 OUT 参数)
调用方式 `SELECT func(args);` `CALL proc(args);`
能否无返回值 ❌ 必须有 `RETURNS` ✅ 可以啥都不返回
事务控制 由外层决定 内部可 `COMMIT/ROLLBACK`

结论:
- 只要你想“查”数据 → 写 FUNCTION;
- 只要你想“干”事且内部自己提交 → 写 PROCEDURE。

---

二、FUNCTION 模板(最常用)

```sql
-- 返回单行
CREATE OR REPLACE FUNCTION get_user_name(uid bigint)
RETURNS text
LANGUAGE plpgsql
AS $$
DECLARE
v_name text;
BEGIN
SELECT name INTO v_name
FROM users
WHERE id = uid;
RETURN v_name; -- 找不到会返回 NULL
END;
$$;

-- 返回结果集
CREATE OR REPLACE FUNCTION list_users(min_age int)
RETURNS TABLE(id bigint, name text, age int)
LANGUAGE plpgsql
AS $$
BEGIN
RETURN QUERY
SELECT u.id, u.name, u.age
FROM users u
WHERE u.age >= min_age
ORDER BY u.age;
END;
$$;

-- 测试
SELECT get_user_name(1);
SELECT * FROM list_users(18);
```

---

三、PROCEDURE 模板(≥ pg11)

```sql
-- 内部自己提交,外部自动回滚不了
CREATE OR REPLACE PROCEDURE transfer_money(
from_id bigint,
to_id bigint,
amount numeric
)
LANGUAGE plpgsql
AS $$
BEGIN
-- 扣钱
UPDATE accounts SET balance = balance - amount WHERE id = from_id;
-- 加钱
UPDATE accounts SET balance = balance + amount WHERE id = to_id;
-- 记日志
INSERT INTO log(from_acc, to_acc, amt, ts)
VALUES (from_id, to_id, amount, now());
-- 手动提交(函数里不允许)
COMMIT;
END;
$$;

-- 调用
CALL transfer_money(3, 5, 100.00);
```

---

四、参数 5 种写法速查
1. `IN` 默认,只进不出
2. `OUT` 只出不进
3. `INOUT` 能进能出
4. `VARIADIC` 可变长参数
5. 缺省值 `age int DEFAULT 18`

示例(INOUT):

```sql
CREATE OR REPLACE FUNCTION swap(INOUT a int, INOUT b int)
LANGUAGE plpgsql AS $$
DECLARE tmp int;
BEGIN
tmp := a; a := b; b := tmp;
END;
$$;

-- 调用
SELECT swap(3,7); -- 返回 7,3
```

---

五、异常处理模板

```sql
CREATE OR REPLACE FUNCTION safe_divide(a numeric, b numeric)
RETURNS numeric
LANGUAGE plpgsql
AS $$
BEGIN
RETURN a / b;
EXCEPTION
WHEN division_by_zero THEN
RAISE NOTICE '除零,返回 NULL';
RETURN NULL;
WHEN OTHERS THEN
RAISE WARNING '未知错误: %', SQLERRM;
RETURN NULL;
END;
$$;
```

---

六、快速验证脚本
把下面内容存成 `proc.sql`,然后

`psql -U postgres -d yourdb -f proc.sql` 即可一键验证。

```sql
-- 建测试表
DROP TABLE IF EXISTS users;
CREATE TABLE users(
id bigserial PRIMARY KEY,
name text NOT NULL,
age int CHECK (age>0)
);

-- 灌点数据
INSERT INTO users(name,age) VALUES
('Alice',19),('Bob',25),('Carol',30);

-- 函数:返回结果集
CREATE OR REPLACE FUNCTION list_users(min_age int)
RETURNS TABLE(id bigint, name text, age int)
LANGUAGE plpgsql AS $$
BEGIN
RETURN QUERY SELECT u.id, u.name, u.age FROM users u WHERE u.age >= min_age;
END;
$$;

-- 过程:无结果,内部可 COMMIT
CREATE OR REPLACE PROCEDURE add_user(p_name text, p_age int)
LANGUAGE plpgsql AS $$
BEGIN
INSERT INTO users(name,age) VALUES (p_name, p_age);
COMMIT;
END;
$$;

-- 测试
SELECT * FROM list_users(20); -- 函数
CALL add_user('David', 40); -- 过程
SELECT * FROM list_users(20);
```

---

七、小结口诀
- 要查询 → `FUNCTION` + `RETURNS`/`TABLE`
- 要干活 → `PROCEDURE` + `CALL`(≥11)
- 想回传单个值 → `OUT` 参数
- 想回传结果集 → `RETURN QUERY`

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/3/12 11:04:57

PyTorch-CUDA-v2.7镜像中打包自定义Python模块的方法

PyTorch-CUDA-v2.7镜像中打包自定义Python模块的方法 在深度学习项目开发过程中,一个常见的痛点是:明明在本地跑得通的模型代码,换到服务器或同事机器上就报错——不是CUDA版本不匹配,就是某个自定义模块导入失败。这种“在我机器…

作者头像 李华
网站建设 2026/3/20 8:48:14

PyTorch-CUDA-v2.7镜像中实现敏感词检测与替换机制

PyTorch-CUDA-v2.7 镜像中实现敏感词检测与替换机制 在当今内容爆炸的数字时代,社交平台、在线教育、客服系统等场景每天面临海量文本输入。如何高效、准确地识别并处理其中的违规表达,已成为企业合规运营的关键挑战。传统的敏感词过滤多依赖正则匹配或调…

作者头像 李华
网站建设 2026/3/25 10:22:21

PyTorch-CUDA-v2.7镜像中设置webhook触发自动化流程

PyTorch-CUDA-v2.7镜像中设置webhook触发自动化流程 在AI研发日益工程化的今天,一个常见的痛点是:开发者提交代码后,还得手动登录远程训练服务器,拉取最新代码、激活环境、启动脚本——这一连串操作不仅耗时,还容易因…

作者头像 李华
网站建设 2026/3/27 14:56:04

快慢双指针算法笔记

文章目录场景解决方案为什么要以值作为下标?双指针严格来说不是一种算法,而是一种思路。场景 数组长度为n1,值在1~n之间,有且仅有一个重复数。 1. 数组值在合法下标范围内(如长度为n1,值在1~n之间) 2. 可…

作者头像 李华
网站建设 2026/3/23 16:43:48

springboot人口老龄化社区服务与管理平台(11613)

有需要的同学,源代码和配套文档领取,加文章最下方的名片哦 一、项目演示 项目演示视频 二、资料介绍 完整源代码(前后端源代码SQL脚本)配套文档(LWPPT开题报告)远程调试控屏包运行 三、技术介绍 Java…

作者头像 李华
网站建设 2026/3/23 20:18:20

PyTorch-CUDA-v2.7镜像中实现按token计费的计量系统原型

PyTorch-CUDA-v2.7镜像中实现按token计费的计量系统原型 在AI推理服务日益普及的今天,如何对模型调用进行精细化资源管理,已经成为云平台和企业级AI系统的共同挑战。传统的“按请求次数”或“按时长计费”模式,难以准确反映实际计算消耗——一…

作者头像 李华