Go语言查询扫描未正确将结果存储到结构中

从现在开始，我们要努力学习啦！今天我给大家带来《Go语言查询扫描未正确将结果存储到结构中》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

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

来自编组 json 的输出，并在函数顶部声明了 var。

[{"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
}

正确答案

我相信您在 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 不同，并且可能会引起混乱，我认为这可能是一个错误（我猜这可能是减少分配的尝试）。我快速查看了这些问题，但找不到任何报告（稍后将进行更详细的查看）。

到这里，我们也就讲完了《Go语言查询扫描未正确将结果存储到结构中》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！