fix: db init table record

This commit is contained in:
langhuihui
2024-12-13 11:47:23 +08:00
parent af8ab607bf
commit 4a66d542ce
2 changed files with 3 additions and 2 deletions
+2 -2
View File
@@ -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))
}
+1
View File
@@ -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