使用 Go database/sql 动态获取查询结果列类型

package main

import (
    "database/sql"
    "fmt"
    "log"
    "reflect" // 用于获取实际Go类型
    _ "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()

    // 创建一个表并插入示例数据
    _, err = db.Exec(`
        CREATE TABLE users (
            id INTEGER PRIMARY KEY,
            name TEXT NOT NULL,
            age INTEGER,
            email TEXT UNIQUE,
            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
        );
        INSERT INTO users (name, age, email) VALUES ('Alice', 30, 'alice@example.com');
        INSERT INTO users (name, age, email) VALUES ('Bob', 25, 'bob@example.com');
        INSERT INTO users (name, age, email) VALUES ('Charlie', NULL, 'charlie@example.com'); -- 演示NULL值
    `)
    if err != nil {
        log.Fatalf("创建表或插入数据失败: %v", err)
    }

    // 2. 执行查询
    query := "SELECT id, name, age, email, created_at FROM users WHERE id > ?"
    rows, err := db.Query(query, 0) // 查询所有用户
    if err != nil {
        log.Fatalf("执行查询失败: %v", err)
    }
    defer rows.Close()

    // 3. 获取列的元数据
    columnTypes, err := rows.ColumnTypes()
    if err != nil {
        log.Fatalf("获取列类型失败: %v", err)
    }

    // 准备一个 []interface{} 来存放扫描到的值
    // 和一个 []interface{} 的指针切片供 rows.Scan() 使用
    values := make([]interface{}, len(columnTypes))
    scanArgs := make([]interface{}, len(columnTypes))
    for i := range values {
        scanArgs[i] = &values[i] // Scan() 需要指针
    }

    fmt.Println("--- 查询结果 ---")
    rowCounter := 0
    // 4. 遍历查询结果集
    for rows.Next() {
        rowCounter++
        fmt.Printf("\n--- 第 %d 行 ---\n", rowCounter)

        // 将当前行的数据扫描到 scanArgs 中
        err = rows.Scan(scanArgs...)
        if err != nil {
            log.Printf("扫描第 %d 行失败: %v", rowCounter, err)
            continue
        }

        // 5. 处理当前行的每一列数据
        for i, colType := range columnTypes {
            colName := colType.Name()
            dbTypeName := colType.DatabaseTypeName()
            scanGoType := colType.ScanType() // database/sql 建议的 Go 类型
            actualValue := values[i]         // 实际扫描到的值

            fmt.Printf("  列名: %s\n", colName)
            fmt.Printf("    数据库类型名: %s\n", dbTypeName)
            fmt.Printf("    建议的 Go 扫描类型: %s\n", scanGoType)

            // 确定扫描到的值的实际 Go 类型
            // 注意：NULL 值在 Go 中会扫描为 nil
            if actualValue == nil {
                fmt.Printf("    实际值: NULL\n")
                fmt.Printf("    实际 Go 类型: <nil>\n")
            } else {
                fmt.Printf("    实际值: %v\n", actualValue)
                fmt.Printf("    实际 Go 类型: %s\n", reflect.TypeOf(actualValue))
            }
        }
    }

    // 检查遍历过程中是否有错误
    if err = rows.Err(); err != nil {
        log.Fatalf("遍历行时发生错误: %v", err)
    }
}

--- 查询结果 ---

--- 第 1 行 ---
  列名: id
    数据库类型名: INTEGER
    建议的 Go 扫描类型: int64
    实际值: 1
    实际 Go 类型: int64
  列名: name
    数据库类型名: TEXT
    建议的 Go 扫描类型: string
    实际值: Alice
    实际 Go 类型: string
  列名: age
    数据库类型名: INTEGER
    建议的 Go 扫描类型: int64
    实际值: 30
    实际 Go 类型: int64
  列名: email
    数据库类型名: TEXT
    建议的 Go 扫描类型: string
    实际值: alice@example.com
    实际 Go 类型: string
  列名: created_at
    数据库类型名: DATETIME
    建议的 Go 扫描类型: time.Time
    实际值: 2023-10-27 10:00:00 +0000 UTC
    实际 Go 类型: time.Time

--- 第 2 行 ---
...

--- 第 3 行 ---
  列名: id
    数据库类型名: INTEGER
    建议的 Go 扫描类型: int64
    实际值: 3
    实际 Go 类型: int64
  列名: name
    数据库类型名: TEXT
    建议的 Go 扫描类型: string
    实际值: Charlie
    实际 Go 类型: string
  列名: age
    数据库类型名: INTEGER
    建议的 Go 扫描类型: int64
    实际值: NULL
    实际 Go 类型: <nil>
  列名: email
    数据库类型名: TEXT
    建议的 Go 扫描类型: string
    实际值: charlie@example.com
    实际 Go 类型: string
  列名: created_at
    数据库类型名: DATETIME
    建议的 Go 扫描类型: time.Time
    实际值: 2023-10-27 10:00:00 +0000 UTC
    实际 Go 类型: time.Time