本文介绍在 Go（配合 pq 驱动）中安全、灵活地调用支持默认参数和命名参数的 PostgreSQL 函数（如 read_user），通过固定占位符 + nil 传参与结构体封装实现类型安全、可维护的多参数查询。
本文介绍在 go（配合 pq 驱动）中安全、灵活地调用支持默认参数和命名参数的 postgresql 函数（如 `read_user`），通过固定占位符 + `nil` 传参与结构体封装实现类型安全、可维护的多参数查询。
PostgreSQL 的 := 命名参数语法（如 _id := 2）仅在 SQL 解析层生效，Go 的数据库驱动（包括 pq）不支持运行时动态拼接命名参数名——它只识别位置占位符（$1, $2, …）。因此，无法像 Python 的 psycopg2 那样直接传入 {"_first": "John"} 字典。但我们可以巧妙利用 PostgreSQL 函数的默认参数机制，结合 Go 的 nil 传递能力，实现等效的“按需传参”。
✅ 正确做法：固定占位符 + 显式 nil
由于函数 read_user 中所有参数均已声明 DEFAULT NULL，且 WHERE 条件使用 OR _param IS NULL 模式，我们只需为每个参数预留一个位置占位符，并在 Go 中对未使用的参数传入 nil：
// 预编译一次，复用语句（推荐用于高频调用）
stmt, err := db.Prepare("SELECT * FROM read_user(_id := $1, _phone := $2, _type := $3, _last := $4, _first := $5)")
if err != nil {
log.Fatal("failed to prepare statement:", err)
}
defer stmt.Close()
// 查询 id=2 的用户（其余参数为 nil，触发默认 NULL 行为）
rows, err := stmt.Query(2, nil, nil, nil, nil)
// 等价于: SELECT * FROM read_user(_id := 2)
// 查询姓 Doe、名 John 的用户
rows, err = stmt.Query(nil, nil, nil, "Doe", "John")
// 等价于: SELECT * FROM read_user(_last := 'Doe', _first := 'John')
⚠️ 注意：必须使用
SELECT * FROM read_user(...)而非SELECT read_user(...)，因为该函数返回SETOF T_USERS（行集），直接SELECT read_user()会将整个结果集作为单个复合类型字段返回，导致扫描失败。
? 进阶封装：结构体 + 类型安全调用
为提升可读性与可维护性，建议定义参数结构体并封装调用逻辑：
type ReadUserParams struct {
ID interface{} `json:"id,omitempty"`
Phone interface{} `json:"phone,omitempty"`
Type interface{} `json:"type,omitempty"`
Last interface{} `json:"last,omitempty"`
First interface{} `json:"first,omitempty"`
}
// ReadUser 执行带默认参数的用户查询，未设置字段自动传 nil
func ReadUser(db *sql.DB, p *ReadUserParams) ([]TUser, error) {
stmt, err := db.Prepare("SELECT * FROM read_user(_id := $1, _phone := $2, _type := $3, _last := $4, _first := $5)")
if err != nil {
return nil, fmt.Errorf("prepare failed: %w", err)
}
defer stmt.Close()
rows, err := stmt.Query(p.ID, p.Phone, p.Type, p.Last, p.First)
if err != nil {
return nil, fmt.Errorf("query failed: %w", err)
}
defer rows.Close()
var users []TUser
for rows.Next() {
var u TUser
if err := rows.Scan(&u.ID, &u.Phone, &u.Type, &u.Last, &u.First); err != nil {
return nil, fmt.Errorf("scan failed: %w", err)
}
users = append(users, u)
}
return users, rows.Err()
}
// 使用示例：
users, _ := ReadUser(db, &ReadUserParams{ID: int64(2)})
users, _ := ReadUser(db, &ReadUserParams{First: "John", Last: "Doe"})
? 关键注意事项
-
nil的含义：pq驱动会将 Go 的nil正确映射为 PostgreSQL 的NULL，前提是列类型兼容（如*int64、*string或直接nil）。若使用强类型字段（如int64），需改用指针或sql.Null*类型。 -
性能优化：预编译语句（
db.Prepare）应复用，避免重复解析；若调用频率低，也可直接用db.Query（内部会自动 prepare + close）。 -
错误处理：PostgreSQL 函数中的
EXCEPTION块仅影响函数内事务，不会阻止 Go 层获取结果；但若 SQL 语法错误或类型不匹配，仍会触发pq的sql.ErrNoRows或其他error，务必检查。 -
替代方案提醒：若未来需更复杂动态条件（如组合
AND/OR、模糊搜索），建议改用原生WHERE查询 +sqlx.NamedQuery（需配合sqlx库及pq的命名参数扩展），而非依赖函数封装。
通过以上方式，你既能享受 PostgreSQL 函数的抽象便利，又能在 Go 中保持代码清晰、类型可控、性能可靠。
