gorm 的model怎么复用

问答 / 211 / 18 / 创建于 1年前 / 更新于 1年前

go r m 的model怎么复用
类似于这种代码，我model怎么给他优化，这样写也太难看了，我查几次就要建几个model

gorm

quanjinxiao

课程读者 18 声望

暂无个人描述~

0 人点赞

推荐文章：

更多推荐...

博客

第三代微服务架构：基于 Go 的博客微服务实战案例，支持分布式事务 21 / 6 |

博客

重大翻车现场，记一次线上事故 18 / 5 |

博客

go-admin 基于 Gin + Vue + Element UI 的前后端分离权限管理系统 58 / 15 |

翻译

教程：使用 go 的 gin 和 gorm 框架来构建 RESTful API 微服务 41 / 6 |

公告

GORM 中文文档开始翻译 10 / 2 |

博客

gin+gorm+router 快速搭建 crud restful API 接口 22 / 12 |

renxiaotu

359 声望

最佳答案

方式二：


func (glc GuestLossController) countBase(c *gin.Context) *gorm.DB {
    model := database.DB.Model(guest_loss.GuestLoss{})
    //列表显示权限判断
    curuser := auth.CurrentUser(c)
    if curuser.CheckRole() == 1 {
        model.Joins("inner join user on user.id = guest_loss.option_id").Where("user.branch_id = ?", curuser.BranchId)
    } else if curuser.CheckRole() == 3 {
        model.Where("guest_loss.option_id=?", curuser.ID)
    }
    if c.Query("begin_at") != "" {
        model.Where("guest_loss.created_at > ?", c.Query("begin_at"))
    }
    if c.Query("end_at") != "" {
        model.Where("guest_loss.created_at <= ?", c.Query("end_at"))
    }
    if c.Query("phone") != "" {
        model.Joins("inner join guest on guest.id = guest_loss.guest_id").Where("guest.phone like ?", "%"+c.Query("phone")+"%")
    }
    return model
}

func (glc GuestLossController) Count(c *gin.Context) {
    var (
        pendding int64
        success  int64
        faild    int64
    )
    glc.countBase().Scopes(guest_loss.Status(0)).Count(&pendding)
    glc.countBase().Scopes(guest_loss.Status(1)).Count(&success)
    glc.countBase().Scopes(guest_loss.Status(2)).Count(&faild)
    response.JSON(c, gin.H{
        "pendding": pendding,
        "success":  success,
        "faild":    faild,
    })

}

1年前评论

renxiaotu （作者）

因为我电脑跑不了，大概是这样，如果报错，你自己调一下

renxiaotu （作者）

如@deatil所说的封装最简单了，我之前都没发现，你这三个查询竟然只有一个status不一样

讨论数量: 18

renxiaotu

359 声望

参见文档：链式调用《GORM 中文文档》

1年前评论

renxiaotu （作者）

也可以用查询范围：查询范围 Scope《GORM 中文文档》

quanjinxiao （楼主）

这个看了，其实也是一样呀，也是得用3个model分别去获取，意义上是一样的呀，只是少了点代码，能不能把我上面那些查询简化成一条，因为除了model model1 model2 不一样其他的其实都是一样的

renxiaotu （作者）

@quanjinxiao 你贴代码出来，别截图

quanjinxiao （楼主）

func (glc GuestLossController) Count(c *gin.Context) {
    var (
        pendding int64
        success  int64
        faild    int64
    )
    model := database.DB.Model(guest_loss.GuestLoss{})
    model1 := database.DB.Model(guest_loss.GuestLoss{})
    model2 := database.DB.Model(guest_loss.GuestLoss{})
    //列表显示权限判断
    curuser := auth.CurrentUser(c)
    if curuser.CheckRole() == 1 {
        model.Joins("inner join user on user.id = guest_loss.option_id").Where("user.branch_id = ?", curuser.BranchId)
        model1.Joins("inner join user on user.id = guest_loss.option_id").Where("user.branch_id = ?", curuser.BranchId)
        model2.Joins("inner join user on user.id = guest_loss.option_id").Where("user.branch_id = ?", curuser.BranchId)
    } else if curuser.CheckRole() == 3 {
        model.Where("guest_loss.option_id=?", curuser.ID)
        model1.Where("guest_loss.option_id=?", curuser.ID)
        model2.Where("guest_loss.option_id=?", curuser.ID)
    }
    if c.Query("begin_at") != "" {
        model.Where("guest_loss.created_at > ?", c.Query("begin_at"))
        model1.Where("guest_loss.created_at > ?", c.Query("begin_at"))
        model2.Where("guest_loss.created_at > ?", c.Query("begin_at"))
    }
    if c.Query("end_at") != "" {
        model.Where("guest_loss.created_at <= ?", c.Query("end_at"))
        model1.Where("guest_loss.created_at <= ?", c.Query("end_at"))
        model2.Where("guest_loss.created_at <= ?", c.Query("end_at"))
    }
    if c.Query("phone") != "" {
        model.Joins("inner join guest on guest.id = guest_loss.guest_id").Where("guest.phone like ?", "%"+c.Query("phone")+"%")
        model1.Joins("inner join guest on guest.id = guest_loss.guest_id").Where("guest.phone like ?", "%"+c.Query("phone")+"%")
        model2.Joins("inner join guest on guest.id = guest_loss.guest_id").Where("guest.phone like ?", "%"+c.Query("phone")+"%")
    }
    model.Scopes(guest_loss.Status(0)).Count(&pendding)
    model1.Scopes(guest_loss.Status(1)).Count(&success)
    model2.Scopes(guest_loss.Status(2)).Count(&faild)
    response.JSON(c, gin.H{
        "pendding": pendding,
        "success":  success,
        "faild":    faild,
    })

}

deatil

见习助教 680 声望

代码差不多的提出来做成方法啊

1年前评论

quanjinxiao （楼主）

对，是可以，但是提出来做方法，也是一样的每个要调用3次，就是这个问题，有没有好的处理办法

deatil （作者）

@quanjinxiao 看你的是几个状态的数量查询吧，那就把前面的相同的合并成一个，条件不同那放最后来获取不同的数量。go的话得注意条件的覆盖情况

renxiaotu

确实是，把status作为参数，就是现成的方法了

quanjinxiao （楼主）

@deatil 嗯嗯，谢谢，懂啦，没考虑到这个，之前一直考虑怎么把同一个model复用出来

renxiaotu

359 声望

按：链式调用《GORM 中文文档》的‘方法链和协程安全’

func (glc GuestLossController) Count(c *gin.Context) {
    var (
        pendding int64
        success  int64
        faild    int64
    )
    model := database.DB.Model(guest_loss.GuestLoss{})
    //列表显示权限判断
    curuser := auth.CurrentUser(c)
    if curuser.CheckRole() == 1 {
        model.Joins("inner join user on user.id = guest_loss.option_id").Where("user.branch_id = ?", curuser.BranchId)
    } else if curuser.CheckRole() == 3 {
        model.Where("guest_loss.option_id=?", curuser.ID)
    }
    if c.Query("begin_at") != "" {
        model.Where("guest_loss.created_at > ?", c.Query("begin_at"))
    }
    if c.Query("end_at") != "" {
        model.Where("guest_loss.created_at <= ?", c.Query("end_at"))
    }
    if c.Query("phone") != "" {
        model.Joins("inner join guest on guest.id = guest_loss.guest_id").Where("guest.phone like ?", "%"+c.Query("phone")+"%")
    }
    ctx1, _ := context.WithTimeout(context.Background(), time.Second)
    ctx2, _ := context.WithTimeout(context.Background(), time.Second)
    model.WithContext(ctx1).Scopes(guest_loss.Status(0)).Count(&pendding)
    model.WithContext(ctx2).Scopes(guest_loss.Status(1)).Count(&success)
    model.Scopes(guest_loss.Status(2)).Count(&faild)
    response.JSON(c, gin.H{
        "pendding": pendding,
        "success":  success,
        "faild":    faild,
    })

}

1年前评论

quanjinxiao （楼主）

牛逼，谢谢啦这个ctx1，ctx2 就是相当于复制一个地址出来吗

renxiaotu （作者）

@quanjinxiao 虽然我知道你要的就是这样，但事实上这个是为携程而生的功能，内部实际是创建了新的会话，你直接封装个方法不更简单吗？

renxiaotu

359 声望

方式二：


func (glc GuestLossController) countBase(c *gin.Context) *gorm.DB {
    model := database.DB.Model(guest_loss.GuestLoss{})
    //列表显示权限判断
    curuser := auth.CurrentUser(c)
    if curuser.CheckRole() == 1 {
        model.Joins("inner join user on user.id = guest_loss.option_id").Where("user.branch_id = ?", curuser.BranchId)
    } else if curuser.CheckRole() == 3 {
        model.Where("guest_loss.option_id=?", curuser.ID)
    }
    if c.Query("begin_at") != "" {
        model.Where("guest_loss.created_at > ?", c.Query("begin_at"))
    }
    if c.Query("end_at") != "" {
        model.Where("guest_loss.created_at <= ?", c.Query("end_at"))
    }
    if c.Query("phone") != "" {
        model.Joins("inner join guest on guest.id = guest_loss.guest_id").Where("guest.phone like ?", "%"+c.Query("phone")+"%")
    }
    return model
}

func (glc GuestLossController) Count(c *gin.Context) {
    var (
        pendding int64
        success  int64
        faild    int64
    )
    glc.countBase().Scopes(guest_loss.Status(0)).Count(&pendding)
    glc.countBase().Scopes(guest_loss.Status(1)).Count(&success)
    glc.countBase().Scopes(guest_loss.Status(2)).Count(&faild)
    response.JSON(c, gin.H{
        "pendding": pendding,
        "success":  success,
        "faild":    faild,
    })

}

1年前评论

renxiaotu （作者）

因为我电脑跑不了，大概是这样，如果报错，你自己调一下

renxiaotu （作者）

如@deatil所说的封装最简单了，我之前都没发现，你这三个查询竟然只有一个status不一样

soonio

17 声望

有没有一种sql，叫group by status，可以直接查出来三种结果在一个列表里？

1年前评论

quanjinxiao （楼主）

有的

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容，与人为善，比聪明更重要！

帮助

gorm 的model怎么复用

推荐文章：

社区赞助商

关于 LearnKu

资源推荐

服务提供商

其他信息

gorm 的model怎么复用

推荐文章：

社区赞助商

关于 LearnKu

资源推荐

服务提供商

其他信息

请登录