go語言將表資料動態轉成切片(欄位任意擴充)

xxfaxy發表於2019-02-16
package main

import (
    "log"
    "fmt"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

//先把欄位的值都當成字串
func Rows2SliceMap(rows *sql.Rows) (list []map[string]string) {
    //欄位名稱
    columns, _ := rows.Columns()
    //多少個欄位
    length := len(columns)
    //每一行欄位的值
    values := make([]sql.RawBytes, length)
    //儲存的是values的記憶體地址
    pointer := make([]interface{}, length)
    //
    for i := 0; i < length; i++ {
        pointer[i] = &values[i]
    }
    //
    for rows.Next() {
        //把引數展開,把每一行的值存到指定的記憶體地址去,迴圈覆蓋,values也就跟著被賦值了
        rows.Scan(pointer...)
        //每一行
        row := make(map[string]string)
        for i := 0; i < length; i++ {
            row[columns[i]] = string(values[i])
        }
        list = append(list, row)
    }
    //
    return
}

func main() {
    db, err := sql.Open("mysql", "root:1234@tcp(192.168.99.165:3306)/test?charset=utf8")
    if err != nil {
        log.Println(err.Error())
    }
    err = db.Ping()
    if err != nil {
        log.Println(err.Error())
    }
    rows, _ := db.Query("select * from cms")
    defer rows.Close()
    list := Rows2SliceMap(rows)
    for k, v := range list {
        fmt.Println(k)
        fmt.Println(v["id"], v["name"], v["sex"], v["age"])
    }
}

相關文章