卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章35444本站已运行396

GORM 返回结果列表或按 id 分组的结果图

gorm 返回结果列表或按 id 分组的结果图

php小编新一今天为大家介绍GORM的一个重要功能,即返回结果列表或按id进行分组的结果图。GORM是一个强大的数据库访问库,它提供了丰富的查询和操作数据库的方法。通过使用GORM,我们可以轻松地从数据库中获取结果列表,也可以按照id进行分组并返回结果图。这个功能在实际开发中非常实用,能够帮助我们更方便地处理数据库查询结果。接下来,让我们一起来了解一下如何使用GORM来实现这个功能吧!

问题内容

本质上,使用 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)。实际上,根据我的理解,您需要拆分记录,而不是使用聚合函数将它们分组(例如 countsum 等)。
我添加的其他代码只是为了清楚起见而添加的。
如果这解决了您的问题或者您需要其他东西,请告诉我,谢谢!

卓越飞翔博客
上一篇: 编辑时在 YAML 文件中保留单引号
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏