GORM复杂查询：多列数据映射至嵌套结构体与Map

首先要明确：Go语言中只有首字母大写的导出字段才能被反射或GORM访问，所以你需要先修正结构体定义：

type FinalResult struct {
    A        A              // 嵌套A结构体
    BMap     map[string]int64 // 用列名（如"B1"）作为键存储B类列的值
    BTotal   int64          // B类列值的总和
    C        interface{}    // 根据C1的实际类型替换，比如string/int64等
}

type A struct {
    A1 string
    A2 string
    // ... 补充An字段
}

接下来是核心的行数据处理逻辑：由于rows.Scan只能一次性扫描整行数据，无法拆分多次扫描，所以我们需要先获取所有列的数值，再按需分配到各个字段：

import (
    "reflect"
    "strings"
)

var results []FinalResult

for rows.Next() {
    res := FinalResult{
        BMap: make(map[string]int64), // 初始化map
    }

    // 获取当前结果集的所有列名
    cols, err := rows.Columns()
    if err != nil {
        // 按需处理错误，比如打印日志、continue等
        continue
    }

    // 创建切片存储每行的所有值，每个元素是对应类型的指针
    values := make([]interface{}, len(cols))
    for i := range values {
        var v interface{}
        values[i] = &v
    }

    // 扫描整行数据到values切片
    if err := rows.Scan(values...); err != nil {
        // 处理扫描错误
        continue
    }

    // 遍历列名和对应值，分配到目标字段
    for i, col := range cols {
        val := *(values[i].(*interface{})) // 取出实际值

        switch {
        // 处理A类列：赋值给嵌套的A结构体
        case strings.HasPrefix(col, "A"):
            // 用反射找到A结构体中对应的字段
            field, ok := reflect.ValueOf(&res.A).Elem().FieldByName(col)
            if ok && field.CanSet() {
                // 根据实际类型转换，这里假设A的字段都是string
                if strVal, ok := val.(string); ok {
                    field.SetString(strVal)
                }
                // 如果有其他类型（如int），需补充类型转换逻辑
            }

        // 处理B类列：存入map并累加总和
        case strings.HasPrefix(col, "B"):
            // 假设B类列的值都是int64类型，按需调整
            if intVal, ok := val.(int64); ok {
                res.BMap[col] = intVal
                res.BTotal += intVal
            }

        // 处理C1列
        case col == "C1":
            // 直接赋值，或根据实际类型转换
            res.C = val
            // 示例：如果C1是string类型 -> res.C, _ = val.(string)
        }
    }

    // 将当前行结果加入结果列表
    results = append(results, res)
}

// 最后检查遍历过程中的错误
if err := rows.Err(); err != nil {
    // 处理错误
}

关键注意事项 #

1. 列名与字段名匹配：如果数据库列名和结构体字段名不一致（比如数据库是a1，结构体是A1），需要给A结构体的字段加上GORM标签，例如：

   type A struct {
       A1 string `gorm:"column:a1"`
       A2 string `gorm:"column:a2"`
   }
   

2. 类型转换准确性：根据你实际查询返回的列类型调整类型转换逻辑，避免类型断言失败导致数据丢失。
3. 性能优化：如果A的字段固定，也可以提前定义列名到A字段的映射表，避免反射带来的开销，例如：

   aColMap := map[string]string{
       "A1": "A1",
       "A2": "A2",
       // ... 其他A类列
   }
   

   处理A类列时直接通过映射表找到字段名，再用反射赋值。

内容的提问来源于stack exchange，提问作者kohrhea