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