Go語言數(shù)據(jù)庫操作：巧妙地將db.QueryRow.Scan結(jié)果映射到map

在Go語言數(shù)據(jù)庫操作中，將查詢結(jié)果映射到自定義結(jié)構(gòu)體是常見做法。然而，有時需要將結(jié)果映射到map中。本文將詳細(xì)講解如何將db.QueryRow.Scan的結(jié)果掃描到map[string]interface{}中，并解決常見錯誤。

直接使用map[string]interface{}作為Scan的參數(shù)是錯誤的，因為Scan函數(shù)需要的是指針，以便寫入數(shù)據(jù)。 以下代碼片段展示了常見的錯誤：

res := map[string]interface{}{"id": nil, "name": nil, "password": nil, "add_time": nil}
// ... Scan(res["id"], res["name"], ...)  // 錯誤！

res["id"]等返回的是interface{}類型的值，而不是指針。 Scan函數(shù)無法將數(shù)據(jù)寫入到這些值中。

正確的做法是為每個map值分配內(nèi)存空間，并使用指針：

res := map[string]interface{}{"id": new(int), "name": new(string), "password": new(string), "add_time": new(int64)}

這里使用new()函數(shù)為int、string和int64類型分別分配內(nèi)存，并得到它們的指針。 Scan函數(shù)可以將數(shù)據(jù)寫入這些指針指向的內(nèi)存地址。

改進后的selectOne函數(shù)如下：

func selectOne(id int) {
    res := map[string]interface{}{"id": new(int), "name": new(string), "password": new(string), "add_time": new(int64)}
    fmt.Println("Initial map:", res) // 添加打印語句，方便調(diào)試

    sql := "select id, name, password, add_time from test where id = ?"
    err := db.QueryRow(sql, id).Scan(res["id"], res["name"], res["password"], res["add_time"])

    if err != nil {
        fmt.Println("獲取數(shù)據(jù)失敗：", err.Error())
    } else {
        fmt.Println("Result map:", res) // 添加打印語句，方便調(diào)試
        // 訪問map中的數(shù)據(jù)
        idVal := *res["id"].(*int)
        nameVal := *res["name"].(*string)
        // ...
    }
}

請注意，訪問map中的數(shù)據(jù)需要進行類型斷言，例如*res["id"].(*int)。 這確保了正確地將interface{}轉(zhuǎn)換為int類型。 我們還添加了打印語句，方便調(diào)試和理解數(shù)據(jù)流向。 SQL語句也進行了調(diào)整，明確指定了要查詢的列名，以避免潛在的列名不匹配問題。 記住，map的鍵名必須與數(shù)據(jù)庫列名一致。

通過這種方法，可以有效地將db.QueryRow.Scan的結(jié)果映射到map中，并避免常見的指針錯誤。 記住始終為map中的值分配內(nèi)存并使用指針。

以上是Go語言中如何將db.QueryRow.Scan結(jié)果映射到map中？的詳細(xì)內(nèi)容。更多信息請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！