Golang 查询扫描未将查询正确扫描到结构中

Question

我在从 Golang 中的 pgx 查询进行扫描时遇到问题。 id 字段始终是最后一条记录的字段。如果我取消注释函数顶部的 var person Person 声明，则每个 id 都是 3。我的数据库中有 3 条 id 从 1 到 3 的记录。当我评论该声明并在 rows.Next() 循环中声明变量时，我得到了正确的 ID。我不明白为什么 personId 没有被正确覆盖

来自编组的输出 JSON 使用在函数顶部声明的 var。

[{"person_id":3,"first_name":"Mark","last_name":"Brown"},{"person_id":3,"first_name":"Sam","last_name":"Smith"},{"person_id":3,"first_name":"Bob","last_name":"Jones"}]

在扫描循环的每次迭代中声明 person 后的输出

[{"person_id":1,"first_name":"Mark","last_name":"Brown"},{"person_id":2,"first_name":"Sam","last_name":"Smith"},{"person_id":3,"first_name":"Bob","last_name":"Jones"}]

我有这个结构

// Person model
type Person struct {
    PersonId       *int64   `json:"person_id"`
    FirstName      *string  `json:"first_name"`
    LastName       *string  `json:"last_name"`
}

这是我的查询函数

func getPersons(rs *appResource, companyId int64) ([]Person, error) {

    //  var person Person

    var persons []Person

    queryString := `SELECT 
      user_id, 
      first_name, 
      last_name,
      FROM users 
      WHERE company_id = `

    rows, err := rs.db.Query(context.Background(), queryString, companyId)

    if err != nil {
        return persons, err
    }

    for rows.Next() {

        var person Person

        err = rows.Scan(
            &person.PersonId,
            &person.FirstName,
            &person.LastName)

        if err != nil {
            return persons, err
        }

        log.Println(*person.PersonId) // 1, 2, 3 for both var patterns

        persons = append(persons, person)
    }

    if rows.Err() != nil {
        return persons, rows.Err()
    }

    return persons, err
}

Answer 1

我相信您在 github.com/jackc/pgx/v4 中发现了一个错误（或者至少是意外行为）。当运行 Scan 时，如果指针（因此 person.PersonId）不为 nil，那么它指向的任何内容都会被重用。为了证明这一点，我复制了这个问题并确认您也可以通过以下方式修复它：

persons = append(persons, person)
person.PersonId = nil

我可以用这个简化的代码重现这个问题：

conn, err := pgx.Connect(context.Background(), "postgresql://user:password@127.0.0.1:5432/schema?sslmode=disable")
if err != nil {
    panic(err)
}
defer conn.Close(context.Background())

queryString := `SELECT num::int FROM generate_series(1, 3) num`

var scanDst *int64
var slc []*int64

rows, err := conn.Query(context.Background(), queryString)

if err != nil {
    panic(err)
}

for rows.Next() {
    err = rows.Scan(&scanDst)

    if err != nil {
        panic(err)
    }

    slc = append(slc, scanDst)
    // scanDst = nil
}

if rows.Err() != nil {
    panic(err)
}

for _, i := range slc {
    fmt.Printf("%v %d\n", i, *i)
}

这个输出是：

0xc00009f168 3
0xc00009f168 3
0xc00009f168 3

您会注意到在每种情况下指针都是相同的。我做了一些进一步的测试：

在上面取消注释 scanDst = nil 解决了这个问题。
使用 database/sql（使用 "github.com/jackc/pgx/stdlib" 驱动程序）时，代码按预期工作。
如果 PersonId 是 *string（并且查询使用 num::text）它按预期工作。

问题似乎归结为 convert.go 中的以下内容：

if v := reflect.ValueOf(dst); v.Kind() == reflect.Ptr {
    el := v.Elem()
    switch el.Kind() {
    // if dst is a pointer to pointer, strip the pointer and try again
    case reflect.Ptr:
        if el.IsNil() {
            // allocate destination
            el.Set(reflect.New(el.Type().Elem()))
        }
        return int64AssignTo(srcVal, srcStatus, el.Interface())

所以这处理了目标是指针的情况（对于某些数据类型）。该代码检查它是否为 nil，如果是，则创建一个相关类型的新实例作为目标。如果它不是 nil 它只是重用指针。注意：我有一段时间没有使用 reflect，所以我的解释可能有问题。

由于行为不同于 database/sql 并且可能会引起混淆，我认为这可能是一个错误（我猜想这可能是为了减少分配）。我快速查看了这些问题，但没有找到任何报告（稍后会进行更详细的查看）。

Golang 查询扫描未将查询正确扫描到结构中

Golang query scan not scanning query correctly into struct

go

pgx