yinbangzhong
2024-07-29 9f78e3b126b15a9b331c3a1a318da1ceea30114c
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 {