postgresql - 如何将 pgx.Rows 从 Query() 转换为 json 数组？

我正在使用 github.com/jackc/pgx 来处理 postgreSQL。
Noq 我想将 pgx.Rows 从 Query() 转换为 json 数组。
我尝试 func 用于 *sql.Rows ，但它不适用于 *pgx.Rows

func PgSqlRowsToJson(rows *pgx.Rows) []byte {
   fieldDescriptions := rows.FieldDescriptions()
   var columns []string
   for _, col := range fieldDescriptions {
    columns = append(columns, col.Name)
   }

   count := len(columns)
   tableData := make([]map[string]interface{}, 0)
   values := make([]interface{}, count)
   valuePtrs := make([]interface{}, count)
   for rows.Next() {
    for i := 0; i < count; i++ {
        valuePtrs[i] = &values[i]
    }
    rows.Scan(valuePtrs...)
    entry := make(map[string]interface{})
    for i, col := range columns {
        var v interface{}
        val := values[i]
        b, ok := val.([]byte)
        if ok {
            v = string(b)
        } else {
            v = val
        }
        entry[col] = v
    }
    tableData = append(tableData, entry)
   }
   jsonData, _ := json.Marshal(tableData)

   return jsonData
}

问题是 Scan() 不适用于 interface{}，它仅适用于显式定义的类型。
你能帮我解决吗？

最佳答案

您可以使用 pgx.FieldDescription 的 Type 方法来检索列的预期类型。将它传递给 reflect.New ，然后您可以分配一个指向该类型值的指针，然后使用这些新分配的值，您可以制作一个非 nil interface{}s 的 slice ，其基础值具有预期的类型。

例如:

func PgSqlRowsToJson(rows *pgx.Rows) []byte {
    fieldDescriptions := rows.FieldDescriptions()
    var columns []string
    for _, col := range fieldDescriptions {
        columns = append(columns, col.Name)
    }

    count := len(columns)
    tableData := make([]map[string]interface{}, 0)

    valuePtrs := make([]interface{}, count)
    for rows.Next() {
        for i := 0; i < count; i++ {
            valuePtrs[i] = reflect.New(fieldDescriptions[i].Type()).Interface() // allocate pointer to type
        }
        rows.Scan(valuePtrs...)

        entry := make(map[string]interface{})
        for i, col := range columns {
            var v interface{}
            val := reflect.ValueOf(valuePtrs[i]).Elem().Interface() // dereference pointer
            b, ok := val.([]byte)
            if ok {
                v = string(b)
            } else {
                v = val
            }
            entry[col] = v
        }
        tableData = append(tableData, entry)
    }
    jsonData, _ := json.Marshal(tableData)

    return jsonData
}

关于postgresql - 如何将 pgx.Rows 从 Query() 转换为 json 数组？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/50238439/