浏览代码

fix:stl图存储

zqbao 5 月之前
父节点
当前提交
817b6e39be
共有 2 个文件被更改,包括 15 次插入3 次删除
  1. 12 0
      models/db.go
  2. 3 3
      services/data/stl/stl.go

+ 12 - 0
models/db.go

@@ -11,6 +11,7 @@ import (
 	"eta/eta_api/models/data_manage/edb_refresh"
 	"eta/eta_api/models/data_manage/excel"
 	future_good2 "eta/eta_api/models/data_manage/future_good"
+	"eta/eta_api/models/data_manage/stl"
 	"eta/eta_api/models/data_manage/supply_analysis"
 	"eta/eta_api/models/data_stat"
 	"eta/eta_api/models/eta_trial"
@@ -204,6 +205,9 @@ func init() {
 	// 初始化因子指标系列
 	initFactorEdbSeries()
 
+	// 初始化STL指标系列
+	initStlEdbInfo()
+
 	// 初始化部分数据表变量(直接init会有顺序问题=_=!)
 	afterInitTable()
 }
@@ -636,6 +640,14 @@ func initFactorEdbSeries() {
 	)
 }
 
+func initStlEdbInfo() {
+	orm.RegisterModel(
+		new(stl.CalculateStlConfig),        // STL指标配置
+		new(stl.CalculateStlConfigMapping), // STL指标映射
+		new(stl.EdbDataCalculateStl),       // STL指标计算数据
+	)
+}
+
 // afterInitTable
 // @Description: 初始化表结构的的后置操作
 // @author: Roc

+ 3 - 3
services/data/stl/stl.go

@@ -538,7 +538,7 @@ func SaveStlEdbInfo(req *request.SaveStlEdbInfoReq, adminId int, adminRealName,
 	switch req.StlEdbType {
 	case 1:
 		// 趋势指标
-		if ok := utils.Rc.IsExist(EDB_DATA_CALCULATE_STL_TREND_CACHE + strconv.Itoa(req.CalculateStlConfigId)); ok {
+		if ok := utils.Rc.IsExist(EDB_DATA_CALCULATE_STL_TREND_CACHE + strconv.Itoa(req.CalculateStlConfigId)); !ok {
 			msg = "计算已过期,请重新计算"
 			err = fmt.Errorf("not found")
 			return
@@ -556,7 +556,7 @@ func SaveStlEdbInfo(req *request.SaveStlEdbInfoReq, adminId int, adminRealName,
 		}
 	case 2:
 		// 季节性指标
-		if ok := utils.Rc.IsExist(EDB_DATA_CALCULATE_STL_SEASONAL_CACHE + strconv.Itoa(req.CalculateStlConfigId)); ok {
+		if ok := utils.Rc.IsExist(EDB_DATA_CALCULATE_STL_SEASONAL_CACHE + strconv.Itoa(req.CalculateStlConfigId)); !ok {
 			msg = "计算已过期,请重新计算"
 			err = fmt.Errorf("not found")
 			return
@@ -574,7 +574,7 @@ func SaveStlEdbInfo(req *request.SaveStlEdbInfoReq, adminId int, adminRealName,
 		}
 	case 3:
 		// 残差性指标
-		if ok := utils.Rc.IsExist(EDB_DATA_CALCULATE_STL_RESIDUAL_CACHE + strconv.Itoa(req.CalculateStlConfigId)); ok {
+		if ok := utils.Rc.IsExist(EDB_DATA_CALCULATE_STL_RESIDUAL_CACHE + strconv.Itoa(req.CalculateStlConfigId)); !ok {
 			msg = "计算已过期,请重新计算"
 			err = fmt.Errorf("not found")
 			return