Go database/sql 包：动态获取查询结果的列类型及其应用

func (rows *Rows) ColumnTypes() ([]*ColumnType, error)

package main

import (
    "database/sql"
    "fmt"
    "log"
    "reflect"

    _ "github.com/mattn/go-sqlite3" // 导入 SQLite 驱动
)

func main() {
    // 1. 打开一个内存中的 SQLite 数据库
    db, err := sql.Open("sqlite3", ":memory:")
    if err != nil {
        log.Fatalf("无法打开数据库: %v", err)
    }
    defer db.Close()

    // 2. 创建一个示例表并插入数据
    sqlStmt := `
    CREATE TABLE users (
        id INTEGER NOT NULL PRIMARY KEY,
        name TEXT,
        age INTEGER,
        balance REAL,
        is_active BOOLEAN
    );
    INSERT INTO users(id, name, age, balance, is_active) values(1, 'Alice', 30, 100.50, TRUE);
    INSERT INTO users(id, name, age, balance, is_active) values(2, 'Bob', 25, 75.25, FALSE);
    `
    _, err = db.Exec(sqlStmt)
    if err != nil {
        log.Fatalf("创建表或插入数据失败: %q: %s", err, sqlStmt)
    }

    // 3. 执行查询
    rows, err := db.Query("SELECT id, name, age, balance, is_active FROM users")
    if err != nil {
        log.Fatalf("执行查询失败: %v", err)
    }
    defer rows.Close()

    // 4. 获取列类型信息
    columnTypes, err := rows.ColumnTypes()
    if err != nil {
        log.Fatalf("获取列类型失败: %v", err)
    }

    fmt.Println("--- 列类型信息 ---")
    for _, ct := range columnTypes {
        fmt.Printf("列名: %s\n", ct.Name())
        fmt.Printf("  数据库类型名称: %s\n", ct.DatabaseTypeName())
        fmt.Printf("  Go 语言推荐扫描类型: %v\n", ct.ScanType())

        // 获取其他可选信息
        if nullable, ok := ct.Nullable(); ok {
            fmt.Printf("  可为空: %t\n", nullable)
        }
        if length, ok := ct.Length(); ok {
            fmt.Printf("  长度: %d\n", length)
        }
        if precision, scale, ok := ct.PrecisionScale(); ok {
            fmt.Printf("  精度: %d, 刻度: %d\n", precision, scale)
        }
        fmt.Println("-------------------------------")
    }

    fmt.Println("\n--- 动态行扫描 ---")
    // 5. 准备用于动态扫描的变量切片
    var scanArgs []interface{} // 存储指向 Go 变量的指针
    var columnNames []string   // 存储列名，用于结果映射

    for _, ct := range columnTypes {
        columnNames = append(columnNames, ct.Name())
        // 根据 ScanType 创建一个新的 Go 变量，并获取其地址
        // reflect.New(ct.ScanType()) 创建一个指向零值的指针
        scanArgs = append(scanArgs, reflect.New(ct.ScanType()).Interface())
    }

    // 6. 遍历查询结果并动态扫描数据
    for rows.Next() {
        err = rows.Scan(scanArgs...) // 将行数据扫描到 scanArgs 中指向的变量
        if err != nil {
            log.Fatalf("扫描行数据失败: %v", err)
        }

        // 7. 处理扫描到的数据
        rowData := make(map[string]interface{})
        for i, colName := range columnNames {
            // 通过反射解引用指针，获取实际的值
            val := reflect.ValueOf(scanArgs[i]).Elem().Interface()
            rowData[colName] = val
        }
        fmt.Printf("行数据: %v\n", rowData)
    }

    if err = rows.Err(); err != nil {
        log.Fatalf("遍历行时发生错误: %v", err)
    }
}

--- 列类型信息 ---
列名: id
  数据库类型名称: INTEGER
  Go 语言推荐扫描类型: int64
  可为空: false
-------------------------------
列名: name
  数据库类型名称: TEXT
  Go 语言推荐扫描类型: string
  可为空: true
-------------------------------
列名: age
  数据库类型名称: INTEGER
  Go 语言推荐扫描类型: int64
  可为空: true
-------------------------------
列名: balance
  数据库类型名称: REAL
  Go 语言推荐扫描类型: float64
  可为空: true
-------------------------------
列名: is_active
  数据库类型名称: BOOLEAN
  Go 语言推荐扫描类型: bool
  可为空: true
-------------------------------

--- 动态行扫描 ---
行数据: map[age:30 balance:100.5 is_active:true id:1 name:Alice]
行数据: map[age:25 balance:75.25 is_active:false id:2 name:Bob]