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

Question

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

函数顶部声明的 var 正确覆盖来自编组 JSON 的输出。

[{"person_id":3,"first_name":"马克","last_name":"布朗"},{"person_id":3,"first_name":"山姆","last_name":"史密斯" },{"person_id":3,"first_name":"鲍勃","last_name":"琼斯"}]

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

[{"person_id":1,"first_name":"马克","last_name":"布朗"},{"person_id":2,"first_name":"山姆","last_name":"史密斯" },{"person_id":3,"first_name":"鲍勃","last_name":"琼斯"}]

我有这个结构

// 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 = $1`

    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）不为零，那么它指向的任何内容都会被重用。为了证明这一点，我复制了该问题并确认您也可以通过以下方式修复它：

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

我可以使用此简化代码复制该问题：

conn, err := pgx.Connect(context.Background(), "postgresql://user:[email protected]: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())

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

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