GORM 结果的列表或按 id 分组的输出图

本篇文章给大家分享《GORM 结果的列表或按 id 分组的输出图》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

本质上，使用 gormdb，我当前的代码看起来像这样：

res = []*modelExample

DB.Model(&modelExample{}).
        Order("task_id ").
        Find(res)

我对 res 所做的事情是，我将手动循环并将具有相同 task_id 的模型附加到一个列表中，然后附加要处理的此列表。我需要这样做的原因是因为我需要对特定列执行一些特定操作，我需要提取这些操作，而这些操作在 gorm 中无法执行。

但是，有没有一种方法可以更有效地执行此操作，让我像列表的列表一样返回，然后我可以 for 循环并对每个列表元素执行操作？

正确答案

您应该能够使用以下代码片段来实现您的需求：

package main

import (
    "fmt"

    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

type modelExample struct {
    TaskId int
    Name   string
}

func main() {
    dsn := "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable"
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(err)
    }

    db.AutoMigrate(&modelExample{})

    // here you should populate the database with some data

    // querying
    res := make(map[int][]modelExample, 0)
    rows, err := db.Table("model_examples").Select("task_id, name").Rows()
    if err != nil {
        panic(err)
    }
    defer rows.Close()

    // scanning
    for rows.Next() {
        var taskId int
        var name string
        rows.Scan(&taskId, &name)
        if _, isFound := res[taskId]; !isFound {
            res[taskId] = []modelExample{{taskId, name}}
            continue
        }
        res[taskId] = append(res[taskId], modelExample{taskId, name})
    }

    // always good idea to check for errors when scanning
    if err = rows.Err(); err != nil {
        panic(err)
    }

    for _, v := range res {
        fmt.Println(v)
    }
}

完成初始设置后，让我们仔细看看查询部分。
首先，您将从表中获取所有记录。您获取的记录存储在 rows 变量中。
在 for 循环中，您扫描所有记录。每条记录都将添加为新的地图条目或附加到现有条目（如果 taskid 已存在于地图中）。
这是根据特定列创建不同列表的最简单方法（例如 taskid）。实际上，根据我的理解，您需要拆分记录，而不是使用聚合函数将它们分组（例如 count、sum 等）。
我添加的其他代码只是为了清楚起见而添加的。
如果这解决了您的问题或者您需要其他东西，请告诉我，谢谢！

到这里，我们也就讲完了《GORM 结果的列表或按 id 分组的输出图》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！