Go语言中将MySQL数据高效转换为JSON的实用指南

import (
    "database/sql"
    "encoding/json"
    "fmt"
    _ "github.com/go-sql-driver/mysql" // MySQL驱动
)

// 假设db是一个已初始化并连接到MySQL数据库的*sql.DB实例
var db *sql.DB 

// getJSONFromSQL 将SQL查询结果转换为JSON字符串
func getJSONFromSQL(sqlString string) (string, error) {
    // 执行SQL查询
    rows, err := db.Query(sqlString)
    if err != nil {
        return "", fmt.Errorf("执行SQL查询失败: %w", err)
    }
    defer rows.Close() // 确保行结果集关闭

    // 获取列名
    columns, err := rows.Columns()
    if err != nil {
        return "", fmt.Errorf("获取列名失败: %w", err)
    }
    count := len(columns) // 列的数量

    // 准备存储所有行数据的切片
    tableData := make([]map[string]interface{}, 0)

    // 准备用于rows.Scan()的切片
    // values 用于存储实际扫描到的数据
    values := make([]interface{}, count)
    // valuePtrs 用于存储values中每个元素的指针，供rows.Scan()使用
    valuePtrs := make([]interface{}, count)
    for i := 0; i < count; i++ {
        valuePtrs[i] = &values[i] // 将每个元素的地址赋值给valuePtrs
    }

    // 遍历每一行数据
    for rows.Next() {
        // 将当前行的数据扫描到values切片中
        err := rows.Scan(valuePtrs...)
        if err != nil {
            return "", fmt.Errorf("扫描行数据失败: %w", err)
        }

        // 为当前行创建一个map来存储键值对
        entry := make(map[string]interface{})
        for i, colName := range columns {
            var v interface{}
            val := values[i] // 获取扫描到的原始值

            // 检查值是否为[]byte类型
            b, ok := val.([]byte)
            if ok {
                // 如果是[]byte，则转换为string
                v = string(b)
            } else {
                // 否则，保持其原始类型（如int64, float64, bool等）
                v = val
            }
            entry[colName] = v // 将处理后的值存入map
        }
        tableData = append(tableData, entry) // 将当前行添加到总数据切片中
    }

    // 将整个数据切片序列化为JSON格式
    jsonData, err := json.Marshal(tableData)
    if err != nil {
        return "", fmt.Errorf("JSON序列化失败: %w", err)
    }

    return string(jsonData), nil
}

CREATE TABLE users (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(255) NOT NULL
);

INSERT INTO users (name) VALUES ('Alice'), ('Bob'), ('Charlie');

func main() {
    // 假设db已经通过sql.Open和db.Ping成功连接到MySQL
    // db, err = sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database_name")
    // if err != nil { /* handle error */ }
    // defer db.Close()

    jsonOutput, err := getJSONFromSQL("SELECT id, name FROM users")
    if err != nil {
        fmt.Printf("获取JSON数据失败: %v\n", err)
        return
    }
    fmt.Println(jsonOutput)
}

[{"id":1,"name":"Alice"},{"id":2,"name":"Bob"},{"id":3,"name":"Charlie"}]