如何从数据库行创建Golang地图?

  • 回答 (3)
  • 关注 (0)
  • 查看 (38)

基本上,在执行查询之后,我希望获取结果行并生成一个[]map[string]interface{},但是我不知道如何使用API来实现这一点,因为Rows.Scan()函数需要与请求的列数(可能还包括类型)匹配的特定数量的参数才能正确获取数据。

再次,我想泛化这个调用,并接受任何查询并将其转换为[]map[string]interface{},其中映射包含映射到该行值的列名。

这可能非常低效,我计划稍后更改结构,以便interface{}是单个数据点的结构。

我如何使用数据库/SQL包,或者如果必要的话使用数据库/SQL/驱动程序包?

袁弋雯袁弋雯提问于
Bilibili科技莫道君行早,更有早行人回答于

可以创建一个结构来维护映射键到[]interface{} slice。通过这样做,不需要创建预定义的结构。例如:

IDOrder: 0
IsClose: 1
IsConfirm: 2
IDUser: 3

然后,可以这样使用它:

  // create a fieldbinding object.
  var fArr []string
  fb := fieldbinding.NewFieldBinding()

  if fArr, err = rs.Columns(); err != nil {
    return nil, err
  }

  fb.PutFields(fArr)

  //
  outArr := []interface{}{}

  for rs.Next() {
    if err := rs.Scan(fb.GetFieldPtrArr()...); err != nil {
      return nil, err
    }

    fmt.Printf("Row: %v, %v, %v, %s\n", fb.Get("IDOrder"), fb.Get("IsConfirm"), fb.Get("IDUser"), fb.Get("Created"))
    outArr = append(outArr, fb.GetFieldArr())
  }

样本输出:

Row: 1, 1, 1, 2016-07-15 10:39:37 +0000 UTC
Row: 2, 1, 11, 2016-07-15 10:42:04 +0000 UTC
Row: 3, 1, 10, 2016-07-15 10:46:20 +0000 UTC
SampleQuery: [{"Created":"2016-07-15T10:39:37Z","IDOrder":1,"IDUser":1,"IsClose":0,"IsConfirm":1},{"Created":"2016-07-15T10:42:04Z","IDOrder":2,"IDUser":11,"IsClose":0,"IsConfirm":1},{"Created":"2016-07-15T10:46:20Z","IDOrder":3,"IDUser":10,"IsClose":0,"IsConfirm":1}]

请参阅下面的完整示例或字段绑定:

main.go

package main

import (
    "bytes"
    "database/sql"
    "encoding/json"
    "fmt"
)

import (
    _ "github.com/go-sql-driver/mysql"
    "github.com/junhsieh/goexamples/fieldbinding/fieldbinding"
)

var (
    db *sql.DB
)

// Table definition
// CREATE TABLE `salorder` (
//   `IDOrder` int(10) unsigned NOT NULL AUTO_INCREMENT,
//   `IsClose` tinyint(4) NOT NULL,
//   `IsConfirm` tinyint(4) NOT NULL,
//   `IDUser` int(11) NOT NULL,
//   `Created` datetime NOT NULL,
//   `Changed` datetime NOT NULL,
//   PRIMARY KEY (`IDOrder`),
//   KEY `IsClose` (`IsClose`)
// ) ENGINE=InnoDB DEFAULT CHARSET=utf8;

func main() {
    var err error

    // starting database server
    db, err = sql.Open("mysql", "Username:Password@tcp(Host:Port)/DBName?parseTime=true")

    if err != nil {
        panic(err.Error()) // Just for example purpose. You should use proper error handling instead of panic
    }

    defer db.Close()

    // SampleQuery
    if v, err := SampleQuery(); err != nil {
        fmt.Printf("%s\n", err.Error())
    } else {
        var b bytes.Buffer

        if err := json.NewEncoder(&b).Encode(v); err != nil {
            fmt.Printf("SampleQuery: %v\n", err.Error())
        }

        fmt.Printf("SampleQuery: %v\n", b.String())
    }
}

func SampleQuery() ([]interface{}, error) {
    param := []interface{}{}

    param = append(param, 1)

    sql := "SELECT "
    sql += "  SalOrder.IDOrder "
    sql += ", SalOrder.IsClose "
    sql += ", SalOrder.IsConfirm "
    sql += ", SalOrder.IDUser "
    sql += ", SalOrder.Created "
    sql += "FROM SalOrder "
    sql += "WHERE "
    sql += "IsConfirm = ? "
    sql += "ORDER BY SalOrder.IDOrder ASC "

    rs, err := db.Query(sql, param...)

    if err != nil {
        return nil, err
    }

    defer rs.Close()

    // create a fieldbinding object.
    var fArr []string
    fb := fieldbinding.NewFieldBinding()

    if fArr, err = rs.Columns(); err != nil {
        return nil, err
    }

    fb.PutFields(fArr)

    //
    outArr := []interface{}{}

    for rs.Next() {
        if err := rs.Scan(fb.GetFieldPtrArr()...); err != nil {
            return nil, err
        }

        fmt.Printf("Row: %v, %v, %v, %s\n", fb.Get("IDOrder"), fb.Get("IsConfirm"), fb.Get("IDUser"), fb.Get("Created"))
        outArr = append(outArr, fb.GetFieldArr())
    }

    if err := rs.Err(); err != nil {
        return nil, err
    }

    return outArr, nil
}

现场绑定包:

package fieldbinding

import (
    "sync"
)

// NewFieldBinding ...
func NewFieldBinding() *FieldBinding {
    return &FieldBinding{}
}

// FieldBinding is deisgned for SQL rows.Scan() query.
type FieldBinding struct {
    sync.RWMutex // embedded.  see http://golang.org/ref/spec#Struct_types
    FieldArr     []interface{}
    FieldPtrArr  []interface{}
    FieldCount   int64
    MapFieldToID map[string]int64
}

func (fb *FieldBinding) put(k string, v int64) {
    fb.Lock()
    defer fb.Unlock()
    fb.MapFieldToID[k] = v
}

// Get ...
func (fb *FieldBinding) Get(k string) interface{} {
    fb.RLock()
    defer fb.RUnlock()
    // TODO: check map key exist and fb.FieldArr boundary.
    return fb.FieldArr[fb.MapFieldToID[k]]
}

// PutFields ...
func (fb *FieldBinding) PutFields(fArr []string) {
    fCount := len(fArr)
    fb.FieldArr = make([]interface{}, fCount)
    fb.FieldPtrArr = make([]interface{}, fCount)
    fb.MapFieldToID = make(map[string]int64, fCount)

    for k, v := range fArr {
        fb.FieldPtrArr[k] = &fb.FieldArr[k]
        fb.put(v, int64(k))
    }
}

// GetFieldPtrArr ...
func (fb *FieldBinding) GetFieldPtrArr() []interface{} {
    return fb.FieldPtrArr
}

// GetFieldArr ...
func (fb *FieldBinding) GetFieldArr() map[string]interface{} {
    m := make(map[string]interface{}, fb.FieldCount)

    for k, v := range fb.MapFieldToID {
        m[k] = fb.FieldArr[v]
    }

    return m
}

回答过的其他问题

如何给数据加密技术选择并使用密钥为防止数据库数据外泄?

Bilibili科技莫道君行早,更有早行人
在日常设计及开发中,为确保数据传输和数据存储的安全,可通过特定的算法,将数据明文加密成复杂的密文。目前主流加密手段大致可分为单向加密和双向加密。 单向加密:通过对数据进行摘要计算生成密文,密文不可逆推还原。算法代表:Base64,MD5,SHA; 双向加密:与单向加密相反,可以把...... 展开详请

如何看待小程序云开发开放之后带来的影响?

Bilibili科技莫道君行早,更有早行人
简单来说,小程序开发者借助「云开发」可完成小程序开发,无需自行搭建服务器,即可实现小程序快速开发和迭代。 云开发平台对开发者有哪些好处? 1.开发成本将降低 在以前,小程序开发者开发一款小程序,需要接触到诸如:域名备案、数据库运维、服务器运维、后端开发、打包发布等等工作,但在「云...... 展开详请

如何寻找最适合自身的云应用模式?

Bilibili科技莫道君行早,更有早行人
公有云:有云开放给所有公众成员。 这种云计算模式由拥有极其庞大数据中心的服务商运营,其中的计算和存储资源在客户之间进行共享。 私有云:私有云是一个公司使用的特定云环境。不同于公有云模式中共享的设施使用,私有云模式中每个公司使用的服务器或存储应用都是单独的。 托管私有云:一些公司会...... 展开详请

实时音视频产品优势是什么?

Bilibili科技莫道君行早,更有早行人

好像可以通过web接入与报警市民进行双向音视频通话,方便市民在一些不方便打字或者电话的场景下报警。

区块链电子发票,对比传统发票有何优势?

Bilibili科技莫道君行早,更有早行人
此前发票的流转场景特征是:参与方多,流转周期长,各参与方之间信息不互通,存在篡改等痛点。区块链是一种不可篡改防抵赖防删除的分步式账本技术,如今作为区块链电子发票的底层核心技术,可以发挥两个核心作用: 1. 确保发票唯一,从领票、开票到流转、入账、报销、全环节流转状态完整可追溯; ...... 展开详请

Python模块和Python包有什么区别?

Bilibili科技莫道君行早,更有早行人
模块是在一个导入下导入并使用的单个文件(或多个文件)。例如 import my_module 包是一个包含层次结构的目录中的模块集合。 from my_package.timing.danger.internets import function_of_love Docume...... 展开详请

关于作者

Bilibili科技

莫道君行早,更有早行人

所属标签

扫码关注云+社区