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 = $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:<a href="https://stackoverflow.com/cdn-cgi/l/email-protection" class="__cf_email__" data-cfemail="6d1d0c1e1e1a021f092d5c5f5a435d435d435c" rel="noreferrer noopener nofollow">[email protected]</a>: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不同并且可能会引起困惑，我认为这可能是一个错误(我猜这可能是减少分配的尝试)。我快速查看了这些问题，但找不到任何报告(稍后将进行更详细的查看)。