如何将最后一个 sql 行追加到列表中而不替换 Golang 中的先前行

2024-04-24

此代码提供了据我所知正确的 JSON 输出 [{},{}],但每一行都会被附加并替换所有先前的行,因此结果仅显示最后一行的副本。

var rows *sql.Rows
rows, err = db.Query(query)
cols, _ := rows.Columns()
colnames, _ := rows.Columns()
vals := make([]interface{}, len(cols))

for i, _ := range cols {
   vals[i] = &cols[i]
}

m := make(map[string]interface{})
 
for i, val := range vals {
  m[colnames[i]] = val
}
    
list := make([]map[string]interface{}, 0)
for rows.Next() {
err = rows.Scan(vals...)
   list = append(list, m)
}
json, _ := json.Marshal(list)
fmt.Fprintf(w,"%s\n", json)

这是循环行的幕后发生的事情:

循环1:{“ID”:“1”,“NAME”:“约翰}

循环2:{“ID”:“2”,“NAME”:“Jane Doe”}{“ID”:“2”,“NAME”:“Jane Doe”}

循环3:{“ID”:“3”,“NAME”:“唐老鸭”}{“ID”:“3”,“NAME”:“唐老鸭”}{“ID”:“3”,“NAME ”:“唐老鸭”}

rows.Scan 获取正确的值,但它会追加并替换所有以前的值。

最终的输出是这样的

[{“ID”:“3”,“NAME”:“唐老鸭”},{“ID”:“3”,“NAME”:“唐老鸭”},{“ID”:“3”,“NAME ”:“唐老鸭”}]

但应该是这样的:

[{“ID”:“1”,“姓名”:“约翰·多伊”},{“ID”:“2”,“姓名”:“简·多伊”},{“ID”:“3”,“姓名” ”:“唐老鸭”}]

我究竟做错了什么?

您可以对此投反对票,但请解释原因。我还是 Golang 的新手,想学习。


我修复了它并用评论解释了你做错了什么:

// 1. Query
var rows *sql.Rows
rows, err = db.Query(query)
cols, _ := rows.Columns()

// 2. Iterate
list := make([]map[string]interface{}, 0)
for rows.Next() {
    vals := make([]interface{}, len(cols))
    for i, _ := range cols {
        // Previously you assigned vals[i] a pointer to a column name cols[i].
        // This meant that everytime you did rows.Scan(vals),
        // rows.Scan would see pointers to cols and modify them
        // Since cols are the same for all rows, they shouldn't be modified.

        // Here we assign a pointer to an empty string to vals[i],
        // so rows.Scan can fill it.
        var s string
        vals[i] = &s

        // This is effectively like saying:
        // var string1, string2 string
        // rows.Scan(&string1, &string2)
        // Except the above only scans two string columns
        // and we allow as many string columns as the query returned us — len(cols).
    }

    err = rows.Scan(vals...)

    // Don't forget to check errors.
    if err != nil {
        log.Fatal(err)
    }

    // Make a new map before appending it.
    // Remember maps aren't copied by value, so if we declared
    // the map m outside of the rows.Next() loop, we would be appending
    // and modifying the same map for each row, so all rows in list would look the same.
    m := make(map[string]interface{})
    for i, val := range vals {
        m[cols[i]] = val
    }
    list = append(list, m)
}

// 3. Print.
b, _ := json.MarshalIndent(list, "", "\t")
fmt.Printf("%s\n", b)

别担心,当我还是个初学者时,这对我来说也很难理解。

现在,有一些有趣的事情:

var list []map[string]interface{}
rows, err := db.Queryx(query)
for rows.Next() {
    row := make(map[string]interface{})
    err = rows.MapScan(row)
    if err != nil {
      log.Fatal(err)
    }
    list = append(list, row)
}

b, _ := json.MarshalIndent(list, "", "\t")
fmt.Printf("%s\n", b)

这与上面的代码的作用相同,但是sqlx http://jmoiron.github.io/sqlx/。稍微简单一点,不是吗?

sqlx 是之上的扩展database/sql使用直接将行扫描到映射和结构的方法,因此您不必手动执行此操作。

我认为你的模型作为一个结构看起来更好:

type Person struct {
    ID int
    Name string
}

var people []Person
rows, err := db.Queryx(query)
for rows.Next() {
    var p Person
    err = rows.StructScan(&p)
    if err != nil {
        log.Fatal(err)
    }
    people = append(people, p)
}

你不觉得吗?

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何将最后一个 sql 行追加到列表中而不替换 Golang 中的先前行 的相关文章

随机推荐