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

原创

TSINGSEE青犀视频

修改于 2021-02-05 14:43:40

4090

修改于 2021-02-05 14:43:40

文章被收录于专栏：TSINGSEE青犀视频

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

在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
}

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