RTMP协议视频平台EasyDSS开发中如何通过Go语言 gorm 框架钩子函数精简代码?

EasyDSS视频平台除了提供网页直播外,还可上传视频进行点播,上传视频后一键生成视频地址,以供用户分享、嵌入到网站、产品软件系统中。

EasyDSS架构绿.png

在EasyDSS开发过程中,将数据写入到数据库时,有大量相同的操作。如以下两个步骤设置更新时间和创建时间。

u.CreateAt = time.Now()
u.UpdateAt = time.Now()

因为很多地方需要更新这两个字段,在整个项目中出现了很多相同的代码,都是以上两行代码。因此为了精简代码,优化结构,可以使用gorm 库的hooks钩子函数功能进行优化。

gorm 为 Go 语言的数据库 orm 框架,需要在项目中创建对应的结构体,结构体如下:

// 基础表
type Table struct {
   // id 为字符串,使用 uuid 或者 xid 进行设置,为了后期做分布式的可能做此处理,因此不使用 int 类型
   ID       string    `json:"id" gorm:"PRIMARY_KEY;Size:32;NOT NULL;"`
   CreateAt time.Time `json:"createAt"  gorm:"type:datetime"`
   UpdateAt time.Time `json:"updateAt"  gorm:"type:datetime"`
}

现在的需求是希望在创建和更新数据的时候自动更新 CreateAt 和 UpdateAt 字段,而不用每次都显示调用赋值语句。gorm1.20.x 以上的版本中提供了 BeforeCreate、BeforeUpdate、AfterCreate 等钩子函数可以实现此功能。代码如下:

// 在创建前更新 create_at 字段
func (u *Table) BeforeCreate(tx *gorm.DB) (err error) {
   u.CreateAt = time.Now()
   u.UpdateAt = time.Now()
   return
}

// 在更前前更新 update_at 字段
func (u *Table) BeforeUpdate(tx *gorm.DB) (err error) {
   u.UpdateAt = time.Now()
   return
}

以上代码就实现了对应的功能,在创建数据和更新数据之前,均会更新已经设置好的字段。

70.png

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值