使用 pgx 将 QueryRow 结果映射到结构体的方法

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《使用 pgx 将 QueryRow 结果映射到结构体的方法》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

我们使用的用户结构包含很多字段，如下所示：

type user struct {
    id       int    `json:"id,omitempty"`
    uuid     string `json:"uuid,omitempty"`
    role     int    `json:"role,omitempty"`
    name     string `json:"name,omitempty"`
    surname  string `json:"surname,omitempty"`
    phone    string `json:"phone,omitempty"`
    email    string `json:"email,omitempty"`
    street   string `json:"street,omitempty"`
    city     string `json:"city,omitempty"`
    password string `json:"password,omitempty"`
}

以及通过电子邮件获取用户的功能：

func getuserbyemail(email string) (u user, err error) {
    row := db.psql.queryrow(
        context.background(),
        "select * from users where email=$1",
        email)
    err = row.scan(&u.id, &u.uuid, &u.role, &u.name, &u.surname, &u.phone, &u.email, &u.street, &u.city, &u.password)
    if err != nil {
        log.fatal(err)
    }
    return
}

有没有办法直接扫描结构体而不是其所有属性？理想情况下：

row.Scan(&u)

解决方案

从5.1.0版本起pgx支持结构体扫描。

type product struct {
    id    int32
    name  string
    price int32
}

rows, _ := conn.query(ctx, "select * from products")
products, err := pgx.collectrows(rows, pgx.rowtostructbyname[product])
if err != nil {
    fmt.printf("collectrows error: %v", err)
    return
}

for _, p := range products {
    fmt.printf("%s: $%d\n", p.name, p.price)
}

您可以使用 db-tags 命名字段：

type name struct {
    last  string `db:"last_name"`
    first string `db:"first_name"`
}

文档：https://pkg.go.dev/github.com/jackc/pgx/v5#RowToStructByName

还有另一个库 scany。 它可以与 pgx 本机接口以及 database/sql 一起使用：

package main

import (
    "context"

    "github.com/jackc/pgx/v4/pgxpool"

    "github.com/georgysavva/scany/pgxscan"
)

type User struct {
    ID    string
    Name  string
    Email string
    Age   int
}

func main() {
    ctx := context.Background()
    db, _ := pgxpool.Connect(ctx, "example-connection-url")

    var users []*User
    pgxscan.Select(ctx, db, &users, `SELECT id, name, email, age FROM users`)
    // users variable now contains data from all rows.
}

它经过了充分的测试和记录，并且概念比 sqlx 少得多。

免责声明，我是这个库的作者。

好了，本文到此结束，带大家了解了《使用 pgx 将 QueryRow 结果映射到结构体的方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！