How to call postgres stored procedure from golang using gorm package

在Go中使用 gorm包来调用PostgreSQL的存储过程是一个相对直接的过程。首先，确保你已经在Go项目中正确安装并导入了 gorm包。以下是从Go调用PostgreSQL存储过程的步骤：

步骤 1: 安装与导入gorm包

确保你的Go项目中已经安装了 gorm和PostgreSQL的驱动，通常是 pgx或 pq。可以通过以下命令安装：

bash
go get -u gorm.io/gorm
go get -u gorm.io/driver/postgres

导入到你的Go文件：

go
import (
    "gorm.io/gorm"
    "gorm.io/driver/postgres"
)

步骤 2: 连接到PostgreSQL数据库

首先，你需要设置数据库连接。这里是一个连接字符串的例子：

go
dsn := "host=localhost user=gorm password=gorm dbname=gorm port=9920 sslmode=disable"
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
    panic("failed to connect database")
}

步骤 3: 定义存储过程

假设你已经在PostgreSQL数据库中定义了一个存储过程。例如，一个简单的存储过程，用于查询某个特定ID的用户信息：

sql
CREATE OR REPLACE FUNCTION get_user_info(userId INT)
RETURNS TABLE(name VARCHAR, age INT) AS $$
BEGIN
    RETURN QUERY SELECT name, age FROM users WHERE id = userId;
END;
$$ LANGUAGE plpgsql;

步骤 4: 从Go调用存储过程

现在，你可以使用 gorm的原生SQL执行功能来调用这个存储过程。使用 Raw方法执行存储过程，并扫描结果：

go
var result []struct {
    Name string
    Age  int
}

db.Raw("SELECT * FROM get_user_info(?)", 1).Scan(&result)

for _, user := range result {
    fmt.Printf("Name: %s, Age: %d\n", user.Name, user.Age)
}

这里，get_user_info(?)是存储过程，1是传递给存储过程的参数，表示用户ID。通过 Scan方法，我们将结果映射到一个结构体切片中，每个结构体代表一个用户的信息。

总结

使用 gorm调用PostgreSQL的存储过程是通过组合使用 Raw和 Scan方法执行原生SQL并处理返回结果。这种方式不仅适用于调用存储过程，也适用于执行任何自定义SQL查询。建议在实际应用中根据存储过程的具体逻辑和返回类型调整相应的代码。