Golang開發實踐:把數據庫數據保存到map[string]interface{}中

Golang讀取數據庫很簡單，可以使用ORM也可以直接使用SQL語句。然而要處理的數據庫表有幾十個字段，光是定義一個四十多個字段的數據模型就讓人感覺痛苦，況且並不是所有字段都要參與運算。於是就決定使用SQL語句來處理。保存到map[string]interface{},這樣也方便實現導出字段可配置的需求。

保存到map[string]interface{}

看了一下Golang標准庫database/sql中Rows的API文檔，並沒有提供獲取所有字段的API接口，文檔提供的實例代碼調用了Scan函數並傳入相關指針獲取數據.

rows, err := db.Query("SELECT ...")
...
defer rows.Close()
for rows.Next() {
    var id int
    var name string
    err = rows.Scan(&id, &name)
    ...
}
err = rows.Err() 

這樣的處理方式明顯無法完成需求，於是就想到了Golang強大的interface{}，可以將相應長度的[]interface{}作為不定長參數傳入Scan函數：

type Record interface{}

func (tf *DataIterator) parseRows(rows *sql.Rows, cSize int) (error) {
    columns, err := rows.Columns()
    if err != nil {
        return err
    }
    size := len(columns)
    pts := make([]interface{}, size)
    container := make([]interface{}, size)
    tf.cache = make([]Record, cSize)
    cursor := 0
    for i := range pts {
        pts[i] = &container[i]
    }
    for rows.Next() {
        err = rows.Scan(pts...)
        if err != nil {
            return err
        }
        var r Record = make(map[string]interface{}, size)
        for i, column := range columns {
            r[column] = container[i]
        }
        tf.cache[cursor] = r
        cursor++
    }
    return nil
}

　　

首先通過Rows.Columns獲取數據庫返回的字段
聲明相同長度的[]interface{}用以保存一行數據
聲明相同長度的[]interface{}作為容器
將pts中元素的指針存入容器
調用Rows.Scan()並將container作為不定長參數傳入
遍歷Rows中的每一行數據即可保存數據庫數據到map[string]interface{}

關於字符串

在實際操作中發現用這種操作方法數據庫中的字符串(CHAR\VARCHAR等)類型會被作為[]uint8。遍歷結果轉換為string即可：

func formatRecord(r Record) {
    for k, v := range r {
        switch v.(type) {
        case []uint8:
            arr := v.([]uint8)
            r[k] = uiToS(arr)
        case nil:
            r[k] = ""
        }
    }
}

　　

---------------------
作者：粗鄙之語
來源：CSDN
原文：https://blog.csdn.net/iyeahme/article/details/74932619