当前位置：首页 > news >正文

gorm如何调存储过程postgresql

news 2026/6/14 5:16:04

GORM 本身不封装“存储过程”概念，但可以直接拿 `*gorm.DB` 当原生 SQL 执行器用，把 PostgreSQL 的函数（或 v11+ 的 PROCEDURE）当成普通 SQL 去调即可。核心就两步：

1. 用 `db.Raw(...).Scan(...)` 调函数；
2. 有 OUT 参数或返回结果集时，按 PostgreSQL 语法把结果 `SELECT` 出来，再扫到 Go 结构体里。

下面给出完整可运行示例（GORM v2 + pgx 驱动）。

---

一、准备 PostgreSQL 端

```sql
-- 返回结果集的函数（最常用）
CREATE OR REPLACE FUNCTION get_user_info(uid int)
RETURNS TABLE(id bigint, name varchar, age int) AS $$
BEGIN
RETURN QUERY SELECT u.id, u.name, u.age FROM users u WHERE u.id = uid;
END;
$$ LANGUAGE plpgsql;

-- 带 IN/OUT 参数的函数
CREATE OR REPLACE FUNCTION add_user(IN p_name varchar, IN p_age int,
OUT new_id bigint, OUT msg text)
AS $$
BEGIN
INSERT INTO users(name, age) VALUES (p_name, p_age) RETURNING id INTO new_id;
msg := 'ok';
END;
$$ LANGUAGE plpgsql;
```

---

二、Go 端代码

```go
package main

import (
"fmt"
"log"

"gorm.io/driver/postgres"
"gorm.io/gorm"
)

type User struct {
ID int64 `gorm:"column:id"`
Name string `gorm:"column:name"`
Age int `gorm:"column:age"`
}

func main() {
dsn := "host=127.0.0.1 user=postgres password=123 db=gorm port=5432 sslmode=disable TimeZone=Asia/Shanghai"
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
log.Fatalln(err)
}

// 1. 调函数返回结果集
var rows []User
if err := db.Raw("SELECT * FROM get_user_info(?)", 1).Scan(&rows).Error; err != nil {
log.Println(err)
}
fmt.Printf("结果集: %+v\n", rows)

// 2. 调函数取 OUT 参数
var newID int64
var msg string
db.Raw("SELECT * FROM add_user(?,?)", "tom", 18).Row().Scan(&newID, &msg)
fmt.Printf("新ID=%d msg=%s\n", newID, msg)
}
```

---

三、几点注意
1. PostgreSQL 11 以后才有真正的 `PROCEDURE`，老版本只有函数；GORM 统一用 `SELECT/VALUES` 语法即可。
2. 如果函数返回复合类型或 refcursor，需要再包一层 `SELECT * FROM func(...)` 才能被 `Scan` 识别。
3. 事务、参数化、连接池都由 GORM 托管，放心使用 `db.Raw` 即可，无需额外取 `*sql.DB`。

查看全文

http://www.rkmt.cn/news/176647.html