소스 검색

音频检索增加时间范围和是否关注的搜索条件

zhangqian 2 년 전
부모
커밋
dc962c6426
6개의 변경된 파일131개의 추가작업 그리고 12개의 파일을 삭제
  1. 3 0
      controllers/audio.go
  2. 30 0
      docs/docs.go
  3. 30 0
      docs/swagger.json
  4. 21 0
      docs/swagger.yaml
  5. 35 6
      models/audio.go
  6. 12 6
      request/audio.go

+ 3 - 0
controllers/audio.go

@@ -178,6 +178,9 @@ func (slf AudioCtl) List(c *gin.Context) {
 		SetTrainNumber(params.TrainNumber).
 		SetDriverNumber(params.DriverNumber).
 		SetStation(params.StationNumber).
+		SetBeginTime(params.BeginTime).
+		SetEndTime(params.EndTime).
+		SetIsFollowed(params.IsFollowed).
 		Find()
 
 	if err != nil {

+ 30 - 0
docs/docs.go

@@ -197,12 +197,42 @@ const docTemplate = `{
                 ],
                 "summary": "音频分析检索",
                 "parameters": [
+                    {
+                        "type": "string",
+                        "description": "开始时间",
+                        "name": "beginTime",
+                        "in": "query"
+                    },
                     {
                         "type": "string",
                         "description": "司机号",
                         "name": "driverNumber",
                         "in": "query"
                     },
+                    {
+                        "type": "string",
+                        "description": "结束时间",
+                        "name": "endTime",
+                        "in": "query"
+                    },
+                    {
+                        "enum": [
+                            1,
+                            2
+                        ],
+                        "type": "integer",
+                        "x-enum-comments": {
+                            "BoolTypeFalse": "false",
+                            "BoolTypeTrue": "true"
+                        },
+                        "x-enum-varnames": [
+                            "BoolTypeTrue",
+                            "BoolTypeFalse"
+                        ],
+                        "description": "是否关注 1关注 2未关注",
+                        "name": "isFollowed",
+                        "in": "query"
+                    },
                     {
                         "type": "string",
                         "description": "关键字",

+ 30 - 0
docs/swagger.json

@@ -185,12 +185,42 @@
                 ],
                 "summary": "音频分析检索",
                 "parameters": [
+                    {
+                        "type": "string",
+                        "description": "开始时间",
+                        "name": "beginTime",
+                        "in": "query"
+                    },
                     {
                         "type": "string",
                         "description": "司机号",
                         "name": "driverNumber",
                         "in": "query"
                     },
+                    {
+                        "type": "string",
+                        "description": "结束时间",
+                        "name": "endTime",
+                        "in": "query"
+                    },
+                    {
+                        "enum": [
+                            1,
+                            2
+                        ],
+                        "type": "integer",
+                        "x-enum-comments": {
+                            "BoolTypeFalse": "false",
+                            "BoolTypeTrue": "true"
+                        },
+                        "x-enum-varnames": [
+                            "BoolTypeTrue",
+                            "BoolTypeFalse"
+                        ],
+                        "description": "是否关注 1关注 2未关注",
+                        "name": "isFollowed",
+                        "in": "query"
+                    },
                     {
                         "type": "string",
                         "description": "关键字",

+ 21 - 0
docs/swagger.yaml

@@ -313,10 +313,31 @@ paths:
   /api-sa/v1/audio/list:
     get:
       parameters:
+      - description: 开始时间
+        in: query
+        name: beginTime
+        type: string
       - description: 司机号
         in: query
         name: driverNumber
         type: string
+      - description: 结束时间
+        in: query
+        name: endTime
+        type: string
+      - description: 是否关注 1关注 2未关注
+        enum:
+        - 1
+        - 2
+        in: query
+        name: isFollowed
+        type: integer
+        x-enum-comments:
+          BoolTypeFalse: "false"
+          BoolTypeTrue: "true"
+        x-enum-varnames:
+        - BoolTypeTrue
+        - BoolTypeFalse
       - description: 关键字
         in: query
         name: keyword

+ 35 - 6
models/audio.go

@@ -32,12 +32,14 @@ type (
 
 	AudioSearch struct {
 		Audio
-		Order    string
-		PageNum  int
-		PageSize int
-		Orm      *gorm.DB
-		Keyword  string
-		IDs      []uint
+		Order     string
+		PageNum   int
+		PageSize  int
+		Orm       *gorm.DB
+		Keyword   string
+		IDs       []uint
+		BeginTime time.Time
+		EndTime   time.Time
 	}
 )
 
@@ -113,6 +115,21 @@ func (slf *AudioSearch) SetStation(name string) *AudioSearch {
 	return slf
 }
 
+func (slf *AudioSearch) SetBeginTime(time time.Time) *AudioSearch {
+	slf.BeginTime = time
+	return slf
+}
+
+func (slf *AudioSearch) SetEndTime(time time.Time) *AudioSearch {
+	slf.EndTime = time
+	return slf
+}
+
+func (slf *AudioSearch) SetIsFollowed(isFollowed constvar.BoolType) *AudioSearch {
+	slf.IsFollowed = isFollowed
+	return slf
+}
+
 func (slf *AudioSearch) build() *gorm.DB {
 	var db = slf.Orm.Table(slf.TableName())
 
@@ -153,6 +170,18 @@ func (slf *AudioSearch) build() *gorm.DB {
 		db = db.Where("id in ?", slf.IDs)
 	}
 
+	if !slf.BeginTime.IsZero() {
+		db = db.Where("occurrence_at >= ?", slf.BeginTime)
+	}
+
+	if !slf.EndTime.IsZero() {
+		db = db.Where("occurrence_at <= ?", slf.EndTime)
+	}
+
+	if slf.IsFollowed != 0 {
+		db = db.Where("is_followed = ?", slf.IsFollowed)
+	}
+
 	return db
 }
 

+ 12 - 6
request/audio.go

@@ -1,14 +1,20 @@
 package request
 
-import "speechAnalysis/constvar"
+import (
+	"speechAnalysis/constvar"
+	"time"
+)
 
 type GetAudioList struct {
 	PageInfo
-	Keyword          string `form:"keyword"`                                                                         // 关键字
-	LocomotiveNumber string `gorm:"index;type:varchar(255);not null;default:'';comment:机车号" form:"locomotiveNumber"` // 机车号
-	TrainNumber      string `gorm:"index;type:varchar(255);not null;default:'';comment:车次" form:"trainNumber"`       // 车次
-	DriverNumber     string `gorm:"index;type:varchar(255);not null;default:'';comment:司机号" form:"driverNumber"`     // 司机号
-	StationNumber    string `gorm:"index;type:varchar(255);not null;default:'';comment:车站号" form:"stationNumber"`    // 车站号
+	Keyword          string            `form:"keyword"`                                                                            // 关键字
+	LocomotiveNumber string            `gorm:"index;type:varchar(255);not null;default:'';comment:机车号" form:"locomotiveNumber"` // 机车号
+	TrainNumber      string            `gorm:"index;type:varchar(255);not null;default:'';comment:车次" form:"trainNumber"`        // 车次
+	DriverNumber     string            `gorm:"index;type:varchar(255);not null;default:'';comment:司机号" form:"driverNumber"`     // 司机号
+	StationNumber    string            `gorm:"index;type:varchar(255);not null;default:'';comment:车站号" form:"stationNumber"`    // 车站号
+	BeginTime        time.Time         `json:"beginTime"`                                                                          //开始时间
+	EndTime          time.Time         `json:"endTime"`                                                                            //结束时间
+	IsFollowed       constvar.BoolType `json:"isFollowed"`                                                                         //是否关注 1关注 2未关注
 }
 
 type ProcessAudio struct {