本篇将介绍我们集成的 gorm v2 操作非常流畅的增删改查功能

1.gormv2 功能非常强大,本篇将介绍 gorm_v2 在 GinSkeleton 中非常简洁、简单的操作流程,以 增删改查 操作为例介绍.
2.阅读完本篇,您可以继续阅读官方文档,学习更多功能:https://gorm.io/zh_CN/docs/

前言

1.一个简单的 CURD 操作,我们的起始点为表单参数验证器,终点为数据写入数据库,接下来流程我们将沿着这个主线展开编写.

用户表单参数验证器

  1. // 给表单参数验证器设置 form 标签,gin框架会获取用户提交的表单参数绑定在此结构体上
  2. // 设置 json 标签 GinSkeleton 会将json对应的字段绑定在上下文(gin.Context)
  3. type UserStore struct {
  4. Base // Base 表示你可以继续组合其他结构体
  5. Pass string `form:"pass" json:"pass" binding:"required,min=6"`
  6. RealName string `form:"real_name" json:"real_name" binding:"required,min=2"`
  7. Phone string `form:"phone" json:"phone" binding:"required,len=11"`
  8. Remark string `form:"remark" json:"remark" `
  9. }
  10. // 验证器语法,更详细用法参见常用开发模块列表专项介绍
  11. func (u UserStore) CheckParams(context *gin.Context) {
  12. // 省略代码...
  13. }

验证器完成进入控制器

控制器可以直接将 gin.Context 继续传递给 UsersModel

  1. // 定义一个控制器结构体
  2. type UsersController struct{}
  3. //2.新增
  4. func (u *UsersController) Create(context *gin.Context) {
  5. if model.CreateUserFactory("").InsertData(context) {
  6. response.Success(context, consts.CurdStatusOkMsg, "")
  7. } else {
  8. response.Fail(context, consts.CurdCreatFailCode, consts.CurdCreatFailMsg+",新增错误", "")
  9. }
  10. }
  11. //5.更新
  12. func (u *UsersController) Edit(context *gin.Context) {
  13. //注意:这里没有实现权限控制逻辑,例如:超级管理管理员可以更新全部用户数据,普通用户只能修改自己的数据。目前只是验证了token有效、合法之后就可以进行后续操作
  14. // 实际使用请根据真是业务实现权限控制逻辑、再进行数据库操作
  15. if model.CreateUserFactory("").UpdateData(context) {
  16. response.Success(context, consts.CurdStatusOkMsg, "")
  17. } else {
  18. response.Fail(context, consts.CurdUpdateFailCode, consts.CurdUpdateFailMsg, "")
  19. }
  20. }

1.以下代码将以 model目录 > users 模型展开

  1. // 创建 userFactory
  2. // 参数说明: 传递空值,默认使用 配置文件选项:UseDbType(mysql)
  3. // 以下函数为固定写法,复制即可,不需要深度研究
  4. func CreateUserFactory(sqlType string) *UsersModel {
  5. return &UsersModel{BaseModel: model.BaseModel{DB: model.UseDbConn(sqlType)}}
  6. }
  7. type UsersModel struct {
  8. model.BaseModel // BaseModel 主要有Id 、 CreatedAt 、UpdatedAt 字段,这里主要是演示UsersModel支持结构体的组合
  9. UserName string `gorm:"column:user_name" json:"user_name"`
  10. Pass string `json:"pass"`
  11. Phone string `json:"phone"`
  12. RealName string `gorm:"column:real_name" json:"real_name"`
  13. Status int `json:"status"`
  14. Remark string `json:"remark"`
  15. LastLoginIp string `gorm:"column:last_login_ip" json:"last_login_ip"`
  16. }
  17. // 设置表名
  18. func (u *UsersModel) TableName() string {
  19. return "tb_users"
  20. }
  21. // UsersModel 结构体组合了 *gorm.DB 的所有功能,您可以通过 u.xxx 直接调用 gorm.DB 的所有功能

1.新增数据

以下代码引用了 data_bind.ShouldBindFormDataToModel(c, &tmp) 函数,这个函数是我们对 gin.ShouldBind 函数的精简,加快数据绑定效率。
1.1 参数绑定的原则:model 定义的结构体字段和表单参数验证器结构体设置的json标签名称、数据类型一致,才可以绑定, UserModel 支持类似BaseModel等结构体组合.
1.2 gorm 的数据新增函数 Create 支持单条、批量,如果是批量,只需要定义被添加的数据为 切片即可,例如 var tmp []UsersModel , u.Create(&tmp)

  1. //新增数据
  2. func (u *UsersModel) InsertData(c *gin.Context) bool {
  3. // 注意: 必须重新定义一个 userModel 变量
  4. var tmp UsersModel
  5. // data_bind.ShouldBindFormDataToModel 函数主要按照 UsersModel 结构体指定的json标签去gin.Context上去寻找相同名称的表单数据,绑定到新定义的变量.
  6. // 这里不能使用 gin.ShouldBind 函数从上下文绑定数据,因为 UserModel 我们组合了 gorm.DB ,该函数功能太强大,会深入内部持续解析gorm.Db,产生死循环
  7. // 使用我们提供的简化版本函数(data_bind.ShouldBindFormDataToModel)代替 gin.ShouldBind 即可
  8. if err := data_bind.ShouldBindFormDataToModel(c, &tmp); err == nil {
  9. // Create 函数会将新插入的数据Id 继续更新到 tmp 结构体的主键ID 字段,这里必须传递 指针. 最终的 tmp 其实就是一条新增加的完整数据
  10. // 注意: 在本项目骨架 Create 参数必须传递 指针类型,这样才能支持gorm的回调函数
  11. if res := u.Create(&tmp); res.Error == nil {
  12. return true
  13. } else {
  14. variable.ZapLog.Error("UsersModel 数据新增出错", zap.Error(res.Error))
  15. }
  16. }else {
  17. variable.ZapLog.Error("UsersModel 数据绑定出错", zap.Error(err))
  18. }
  19. return false
  20. }

1.3 关于model变量从上下文如何绑定,附一张绑定数据的逻辑图,也可以帮助大家理解.

model(gorm ginskeleton CURD精华) - 图1

2.修改数据

2.1 gorm 的数据更新有两个函数: updates 不会处理零值字段,save 会全量覆盖式更新字段
2.2 u.Updates() 函数会根据 UsersModel 已经绑定的 TableName 函数解析对应的数据表,然后根据 tmp 结构体定义的主键Id,去更新其他字段值.
2.3 更新时可以搭配 gorm_v2 提供的 Select() 指定字段更新,例如:gorm.Db.Select(字段1,字段2,字段3..) ,也可以设置忽略特定字段更新数据,例如: gorm.Db.Omit(字段1,字段2,字段3..)

  1. //更新
  2. func (u *UsersModel) UpdateData(c *gin.Context) bool {
  3. var tmp UsersModel
  4. if err := data_bind.ShouldBindFormDataToModel(c, &tmp); err == nil {
  5. //tmp 会被自动绑定 CreatedAt、UpdatedAt 字段,更新时我们不希望更新 CreatedAt 字段,使用 Omit 语法忽略该字段
  6. // 注意: 在本项目骨架 Save、Update 参数必须传递 指针类型,这样才能支持gorm的回调函数
  7. if res := u.Omit("CreatedAt").Save(&tmp); res.Error == nil {
  8. return true
  9. } else {
  10. variable.ZapLog.Error("UsersModel 数据更更新出错", zap.Error(err))
  11. }
  12. }
  13. return false
  14. }

3.单条删除数据

UsersModel 已经绑定了函数 TableName ,所以 u.Delete(u,id) 会自动解析出需要删除的表,然后根据Id删除数据.

  1. //删除,我们根据Id删除
  2. func (u *UsersModel) DeleteData(id int) bool {
  3. if u.Delete(u, id).Error == nil {
  4. return true
  5. }
  6. return false
  7. }

4.批量删除(推荐使用第一种方法)

4.1 如果用户传递的参数是 ids 格式如右侧: “100,200,300,400”

  1. //批量删除
  2. func (i *IptvUser) BatchDeleteData(ids string) bool {
  3. if i.Where(" FIND_IN_SET(id,?)", ids).Delete(i).Error == nil {
  4. go i.syncDelTbUsers(ids)
  5. return true
  6. }
  7. return false
  8. }

4.2 批量删除数据(第二种方法)

如果用户传递的参数是 ids 格式如右侧: [100,200,300,400]

  1. //删除,我们根据Id删除
  2. func (u *UsersModel) DeleteData(ids []int) bool {
  3. // ids 格式必须是: [100,200,300,400]
  4. if u.Where("id in (?)",ids).Delete(u).Error == nil {
  5. return true
  6. }
  7. return false
  8. }

5.查询

在线代码预览

5.1 查询是sql操作最复杂的环节,如果业务复杂,那么请使用原生sql操作业务

  1. // 查询类 sql 语句
  2. u.Raw(sql语句,参数1,参数2... ... )
  3. // 执行类 sql 语句
  4. u.Exec(sql语句,参数1,参数2... ... )

5.2 接下来我们演示gorm自带查询

  1. // 第一种情况
  2. // 如果 UsersModel 结构体已经绑定 TableName 函数,那么查询语句对应的数据表名就是 tableName 的返回值;
  3. var tmp []UsersModel
  4. // Where 关键词前面没有指定表名,那么查询的数据库表名就是 tmp 对应的结构体 UsersModel 结构体绑定的 TableName 的返回值
  5. u.Where("ID = ?", user_id).Find(&tmp)
  6. // 第二种情况
  7. var tmp []UsersList
  8. // 假设 UsersList 是自定义数据类型,没有绑定 TbaleName ,那么在 where 关键词开始时就必须指定表名
  9. //指定表名 有以下两种方式:
  10. // u.Model(u) 表示从 u 结构体绑定的 tableName 函数获取对应的表名,如果 u 对应的结构体和 tmp 对应的结构体 UsersList 都没有绑定 TableName ,就会发生错误
  11. u.Model(u).Where("ID = ?", user_id).Find(&tmp)
  12. // u.Tbale(u.TableName()).Where("ID = ?", user_id).Find(&tmp)