如何在Golang中将* sql.Rows转换为键入的JSON

发布时间：2020-12-16 09:22:38 所属栏目：大数据来源：网络整理

导读：本质上,我试图在 MySQL数据库上运行查询,将数据转换为JSON并发送回客户端.我尝试了几种方法,所有“简单”方法都会导致将所有JSON作为字符串发回.我需要将其作为带有[] float64值的键(字符串)发回.这样我就有一个与密钥相关的数据数组.此外,这需要一个类型.到

本质上,我试图在 MySQL数据库上运行查询,将数据转换为JSON并发送回客户端.我尝试了几种方法,所有“简单”方法都会导致将所有JSON作为字符串发回.我需要将其作为带有[] float64值的键(字符串)发回.这样我就有一个与密钥相关的数据数组.此外,这需要一个类型.到目前为止我发现的最好的方法是将所有数据构建到一个结构中,对其进行编码并将其发送回ResponseWriter.

我已经看到了几个关于从数据库制作JSON的问题,但我还没有找到任何利用struct方法的东西.我将下面的代码写入一个函数来说明我的问题.这是非常有限的,因为它只处理两个字段,它必须是一个float64.

因此,我的问题是：如何在将回复发送回客户端之前从具有正确类型的查询响应创建此JSON,并且有一种方法可以动态地执行此操作(即,可以接受可变数量的列和未知类型)？：

{“值”：[12.54,76.98,34.90],“日期”：[“2017-02-03”,“2017-02-04：,”2017-02-05“]}

type DbDao struct{
      db *sql.DB
    }

    type JSONData struct {
      Values []float64
      Dates []string
    }


    func (d *DbDao) SendJSON(sqlString string,w http.ResponseWriter) (error) {

      stmt,err := d.db.Prepare(sqlString)
      if err != nil {
        return err
      }
      defer stmt.Close()

      rows,err := stmt.Query()
      if err != nil {
        return err
      }
      defer rows.Close()

      values := make([]interface{},2)
      scanArgs := make([]interface{},2)
      for i := range values {
        scanArgs[i] = &values[i]
      }

      for rows.Next() {
        err := rows.Scan(scanArgs...)
        if err != nil {
          return err
        }

        var tempDate string 
        var tempValue float64
        var myjson JSONData

        d,dok := values[0].([]byte)
        v,vok := values[1].(float64)

        if dok {
          tempDate = string(d)
          if err != nil {
            return err
          }
          myjson.Dates = append(myjson.Dates,tempDate)
        }

        if vok {      
          tempValue = v 
          myjson.Values = append(myjson.Values,tempValue)
          fmt.Println(v)
          fmt.Println(tempValue)

        }    

        err = json.NewEncoder(w).Encode(&myjson)
        if err != nil {
          return err
        }
      }

      return nil 
    }

解决方法

这是我能够提出的最佳实现,它将使其变得动态.它也比我原来短得多.正如我已经看到这种类型的问题,我希望这有助于其他人.我对其他有更好实现的答案持开放态度：

func (d *DbDao) makeStructJSON(queryText string,w http.ResponseWriter) error {

    // returns rows *sql.Rows
    rows,err := d.db.Query(queryText)
    if err != nil {
        return err
    }
    columns,err := rows.Columns()
    if err != nil {
        return err
    }

    count := len(columns)
    values := make([]interface{},count)
    scanArgs := make([]interface{},count)
    for i := range values {
        scanArgs[i] = &values[i]
    }

    masterData := make(map[string][]interface{})

    for rows.Next() {
        err := rows.Scan(scanArgs...)
        if err != nil {
            return err
        }
        for i,v := range values {

            x := v.([]byte)

            //NOTE: FROM THE GO BLOG: JSON and GO - 25 Jan 2011:
            // The json package uses map[string]interface{} and []interface{} values to store arbitrary JSON objects and arrays; it will happily unmarshal any valid JSON blob into a plain interface{} value. The default concrete Go types are:
            //
            // bool for JSON booleans,// float64 for JSON numbers,// string for JSON strings,and
            // nil for JSON null.

            if nx,ok := strconv.ParseFloat(string(x),64); ok == nil {
                masterData[columns[i]] = append(masterData[columns[i]],nx)
            } else if b,ok := strconv.ParseBool(string(x)); ok == nil {
                masterData[columns[i]] = append(masterData[columns[i]],b)
            } else if "string" == fmt.Sprintf("%T",string(x)) {
                masterData[columns[i]] = append(masterData[columns[i]],string(x))
            } else {
                fmt.Printf("Failed on if for type %T of %vn",x,x)
            }

        }
    }

    w.Header().Set("Content-Type","application/json")

    err = json.NewEncoder(w).Encode(masterData)

    if err != nil {
        return err
    }
    return err
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!