From 8adde93b8f2af29fb8a6405efaa49a41a1c22f8d Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 15 五月 2024 20:15:31 +0800
Subject: [PATCH] 上传音频时保存车站号,车次,车站到火车信息表,提供火车信息查询接口

---
 models/audio_text.go |   19 ++++++++++++++-----
 1 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/models/audio_text.go b/models/audio_text.go
index e4ad793..3e5d4b2 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
 	}
 

--
Gitblit v1.8.0