gorm 的model怎么复用

问答 / 0 / 18 / 创建于 2年前 / 更新于 2年前

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

gorm

quanjinxiao

课程读者 18 声望

暂无个人描述~

0 人点赞

renxiaotu

377 声望

最佳答案

方式二：


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,
    })

}

2年前评论

renxiaotu （作者）

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

renxiaotu （作者）

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

讨论数量: 18

renxiaotu

377 声望

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

2年前评论

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

见习助教 762 声望

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

2年前评论

quanjinxiao （楼主）

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

deatil （作者）

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

renxiaotu

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

quanjinxiao （楼主）

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

renxiaotu

377 声望

按：链式调用《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,
    })

}

2年前评论

quanjinxiao （楼主）

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

renxiaotu （作者）

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

renxiaotu

377 声望

方式二：


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,
    })

}

2年前评论

renxiaotu （作者）

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

renxiaotu （作者）

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

soonio

17 声望

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

2年前评论

quanjinxiao （楼主）

有的

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

帮助

gorm 的model怎么复用

社区赞助商

关于 LearnKu

资源推荐

服务提供商

其他信息

gorm 的model怎么复用

社区赞助商

关于 LearnKu

资源推荐

服务提供商

其他信息

请登录