Go语言中实现数据库查询结果的按列名映射：提升代码健壮性与可维护性

package main

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

    _ "github.com/go-sql-driver/mysql" // 假设使用MySQL驱动
)

type User struct {
    ID   int
    Name string
    Age  int
}

func main() {
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/testdb")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    rows, err := db.Query("SELECT name, age FROM users WHERE age >= ?", 50)
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    for rows.Next() {
        var name string
        var age int
        // 这里的&name和&age必须与SELECT语句中的列顺序（name是第一列，age是第二列）严格匹配
        if err := rows.Scan(&name, &age); err != nil {
            log.Fatal(err)
        }
        fmt.Printf("%s is %d years old\n", name, age)
    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }
}

package main

import (
    "database/sql"
    "fmt"
    "log"
    "reflect"
    "strings" // 用于将列名转换为结构体字段名（通常首字母大写）

    _ "github.com/go-sql-driver/mysql"
)

// Person 结构体用于映射数据库中的person表
type Person struct {
    ID   int    // 对应数据库的id列
    Name string // 对应数据库的name列
    // Age int // 如果数据库有age列但我们不关心，可以不在这里定义
}

func main() {
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/testdb")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 假设数据库中有一个person表，包含id和name列
    // 这里我们查询id和name
    rows, err := db.Query("SELECT id, name FROM person;")
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    // 获取查询结果的列名
    columnNames, err := rows.Columns() // 例如：[]string{"id", "name"}
    if err != nil {
        log.Fatal(err)
    }

    // 用于存储所有Person对象的切片
    people := make([]Person, 0)

    for rows.Next() {
        person := Person{} // 为每一行创建一个新的Person实例

        // 创建一个切片来存储指向结构体字段的指针，供rows.Scan使用
        // 其长度与列名数量一致
        pointers := make([]interface{}, len(columnNames))

        // 获取person结构体的反射值
        structVal := reflect.ValueOf(&person).Elem() // 注意这里需要获取可设置的元素值

        // 遍历所有列名，并使用反射找到结构体中对应的字段
        for i, colName := range columnNames {
            // 数据库列名通常是小写或蛇形命名，而Go结构体字段名通常是首字母大写
            // 这里简单地将列名首字母大写以匹配结构体字段名
            fieldName := strings.Title(colName) 
            fieldVal := structVal.FieldByName(fieldName)

            if !fieldVal.IsValid() {
                // 如果结构体中没有对应的字段，可以选择跳过或记录错误
                // 对于不关心的列，这里可以将其指针设为nil或一个临时变量的地址
                // 为了简化示例，这里直接Fatal，实际应用中应更灵活处理
                log.Printf("Warning: struct field for column '%s' not found. Skipping.", colName)
                // 或者将其映射到一个临时的interface{}变量，避免Scan报错
                var temp interface{}
                pointers[i] = &temp
                continue 
            }

            // 检查字段是否可导出（首字母大写）且可设置
            if !fieldVal.CanSet() {
                log.Fatalf("Error: field '%s' is not settable (e.g., unexported).", fieldName)
            }

            // 获取字段的地址，并转换为interface{}类型
            pointers[i] = fieldVal.Addr().Interface()
        }

        // 将指针切片传递给rows.Scan()
        err := rows.Scan(pointers...)
        if err != nil {
            log.Fatal(err)
        }

        // 扫描成功后，person实例已经填充了数据
        people = append(people, person)
    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }

    // 打印结果
    for _, p := range people {
        fmt.Printf("Person ID: %d, Name: %s\n", p.ID, p.Name)
    }
}