Browse Source

调整数据来源

xyxie 1 year ago
parent
commit
02215a44a3
2 changed files with 19 additions and 2 deletions
  1. 10 2
      services/index_data/jiayue_platform.go
  2. 9 0
      utils/common.go

+ 10 - 2
services/index_data/jiayue_platform.go

@@ -17,9 +17,17 @@ func GetIndexFromJiaYue(indexCode, source, startDate, endDate string) (data *res
 			return
 		}
 	}()
+	var sourceArray []string
+	if source == "wind" {
+		sourceArray = append(sourceArray, "wind", "wind_stop", "wind_p", "wind_tmp")
+	} else if source == "manual" {
+		sourceArray = append(sourceArray, "manual", "manul", "munal")
+	} else {
+		sourceArray = append(sourceArray, source)
+	}
 	var indexPars []interface{}
-	indexCond := " SOURCE_TYPE = :1"
-	indexPars = append(indexPars, source)
+	indexCond := ` SOURCE_TYPE IN (` + utils.GetOracleInReplace(len(sourceArray)) + `)`
+	indexPars = append(indexPars, sourceArray)
 
 	if source == "wind" {
 		indexCond += " AND SOURCE_CODE = :1 "

+ 9 - 0
utils/common.go

@@ -1071,3 +1071,12 @@ func GetWeekDay() (string, string) {
 	l := lastOfWeeK.Unix()
 	return time.Unix(f, 0).Format("2006-01-02") + " 00:00:00", time.Unix(l, 0).Format("2006-01-02") + " 23:59:59"
 }
+
+// GetOracleInReplace 获取oracle sql的in查询替换:1的方法
+func GetOracleInReplace(num int) string {
+	template := make([]string, num)
+	for i := 0; i < num; i++ {
+		template[i] = ":1"
+	}
+	return strings.Join(template, ",")
+}