From a5429b8a83e1b4a48c2a3b9b0b475f08a114837c Mon Sep 17 00:00:00 2001 From: yinbangzhong <zhongbangyin@126.com> Date: 星期四, 20 六月 2024 09:11:57 +0800 Subject: [PATCH] download file by attach model --- models/audio_text.go | 23 ++++++++++++++++------- 1 files changed, 16 insertions(+), 7 deletions(-) diff --git a/models/audio_text.go b/models/audio_text.go index e4ad793..611d75c 100644 --- a/models/audio_text.go +++ b/models/audio_text.go @@ -10,9 +10,8 @@ // AudioText 闊抽鏂囨湰 AudioText struct { gorm.Model - AudioID uint `gorm:"unique;type:int;not null;default:0;comment:闊抽id" json:"audioID"` // 闊抽id - AudioText string `json:"audioText"` // 闊抽鏂囨湰 - Score float64 `json:"score"` // 缃俊搴� + AudioID uint `gorm:"unique;type:int;not null;default:0;comment:闊抽id" json:"audioID"` // 闊抽id + AudioText string `json:"audioText"` // 闊抽鏂囨湰 } AudioTextSearch struct { @@ -21,6 +20,7 @@ PageNum int PageSize int Orm *gorm.DB + AudioIDs []uint } ) @@ -51,6 +51,11 @@ return slf } +func (slf *AudioTextSearch) SetAudioIDs(ids []uint) *AudioTextSearch { + slf.AudioIDs = ids + return slf +} + func (slf *AudioTextSearch) build() *gorm.DB { var db = slf.Orm.Table(slf.TableName()) @@ -60,6 +65,10 @@ if slf.Order != "" { db = db.Order(slf.Order) + } + + if len(slf.AudioIDs) > 0 { + db = db.Where("audio_id in ?", slf.AudioIDs) } return db @@ -90,7 +99,7 @@ func (slf *AudioTextSearch) Save(record *AudioText) error { var db = slf.build() - if err := db.Omit("CreatedAt").Save(record).Error; err != nil { + if err := db.Save(record).Error; err != nil { return fmt.Errorf("save err: %v, record: %+v", err, record) } @@ -124,7 +133,7 @@ func (slf *AudioTextSearch) Delete() error { var db = slf.build() - if err := db.Unscoped().Delete(&AudioText{}).Error; err != nil { + if err := db.Delete(&AudioText{}).Error; err != nil { return err } @@ -151,7 +160,7 @@ db = slf.build() ) - if err := db.Count(&total).Error; err != nil { + if err := db.Where("deleted_at IS NULL").Count(&total).Error; err != nil { return records, int(total), fmt.Errorf("find count err: %v", err) } if slf.PageNum*slf.PageSize > 0 { @@ -188,7 +197,7 @@ db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) - if err := db.Count(&total).Error; err != nil { + if err := db.Where("deleted_at IS NULL").Count(&total).Error; err != nil { return records, total, fmt.Errorf("find by query count err: %v", err) } if slf.PageNum*slf.PageSize > 0 { -- Gitblit v1.8.0