将数据库行读入 map[string]string

Question

我想从一个简单的 SQL 表中读取结果，如下所示

顾客	钥匙
一种	12345
乙	6789

现在，我想构建一个map[string]string ，它的键值对等于行值，如下所示：

map[A:12345, B:6789]

但是，我无法从查询结果中取出值并动态创建键值对。 我将提供代码的粗略概述（SQL 连接不是问题，我已经弄清楚了）

import "database/sql"

func main() {

    // Some code to connect to MSSQL...

    db, _ := sql.Open("mssql", connString)

    stmt := "SELECT customer, key from tbl"
    rows, _ := db.Query(stmt)
    defer rows.Close()

    data := make(map[string]string)
    for rows.Next() {
        // Need help here grabbing each row pair and adding them to my map...
    }
}

我也愿意尝试使用 Empty 结构来执行此操作，其中字段成为我的结果集（动态）的第一列，而值成为我的结果集的第二列。

Answer 1

您可以暂时将值存储在两个变量中，然后将它们存储在 map 中：

func main() {
    stmt := "SELECT customer, key from tbl"
    rows, _ := db.Query(stmt)
    defer rows.Close()

    data := make(map[string]string)
    var (
        consumer string
        key string
    )
    for rows.Next() {
        if err := rows.Scan(&consumer, &key); err != nil {
            // Handle err
        }

        data[consumer] = key 
    }

    for k, v := range data {
        fmt.Println(k, v)
    }
    // "A" "12345"
    // "B" "6789"

}

请注意， consumer和key变量是在循环之外分配的，因此我们可以重新使用它们。 即使值是空字符串，变量也会在每次迭代时被覆盖。