From 4a66d542ce2df73f0d70cba2dcf2eb5fdfc85a3e Mon Sep 17 00:00:00 2001 From: langhuihui <178529795@qq.com> Date: Fri, 13 Dec 2024 11:47:23 +0800 Subject: [PATCH] fix: db init table record --- plugin/mp4/api.go | 4 ++-- plugin/mp4/index.go | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/plugin/mp4/api.go b/plugin/mp4/api.go index 75fd885..258f24c 100644 --- a/plugin/mp4/api.go +++ b/plugin/mp4/api.go @@ -41,9 +41,9 @@ func (p *MP4Plugin) List(ctx context.Context, req *pb.ReqRecordList) (resp *pb.R var totalCount int64 //总条数 // 查询总记录数 - countQuery := p.DB.Model(m7s.RecordStream{}) + countQuery := p.DB.Model(&m7s.RecordStream{}) // 查询当前页的数据 - query := p.DB.Model(m7s.RecordStream{}) + query := p.DB.Model(&m7s.RecordStream{}) if req.PageSize > 0 { query = query.Limit(int(req.PageSize)).Offset(int(offset)) } diff --git a/plugin/mp4/index.go b/plugin/mp4/index.go index 8783b44..dbc9311 100644 --- a/plugin/mp4/index.go +++ b/plugin/mp4/index.go @@ -93,6 +93,7 @@ func (p *MP4Plugin) RegisterHandler() map[string]http.HandlerFunc { func (p *MP4Plugin) OnInit() (err error) { if p.DB != nil { err = p.DB.AutoMigrate(&Exception{}) + p.DB.AutoMigrate(&m7s.RecordStream{}) var deleteRecordTask DeleteRecordTask deleteRecordTask.DB = p.DB deleteRecordTask.DiskMaxPercent = p.DiskMaxPercent