浏览代码

fix: AI预测图表

hsun 1 月之前
父节点
当前提交
af69eaf805
共有 2 个文件被更改,包括 14 次插入0 次删除
  1. 7 0
      controllers/chart_auth.go
  2. 7 0
      controllers/chart_common.go

+ 7 - 0
controllers/chart_auth.go

@@ -106,6 +106,13 @@ func (this *ChartAuthController) ChartInfoDetail() {
 			br.ErrMsg = errMsg
 			return
 		}
+	case utils.CHART_SOURCE_AI_PREDICT_MODEL_DAILY, utils.CHART_SOURCE_AI_PREDICT_MODEL_MONTHLY:
+		resp, isOk, msg, errMsg = GetAiPredictChartInfoDetailFromUniqueCode(chartInfo, key)
+		if !isOk {
+			br.Msg = msg
+			br.ErrMsg = errMsg
+			return
+		}
 	default:
 		br.Msg = "错误的图表"
 		br.ErrMsg = "错误的图表"

+ 7 - 0
controllers/chart_common.go

@@ -139,6 +139,13 @@ func (this *ChartController) CommonChartInfoDetailFromUniqueCode() {
 			br.ErrMsg = errMsg
 			return
 		}
+	case utils.CHART_SOURCE_AI_PREDICT_MODEL_DAILY, utils.CHART_SOURCE_AI_PREDICT_MODEL_MONTHLY:
+		resp, isOk, msg, errMsg = GetAiPredictChartInfoDetailFromUniqueCode(chartInfo, key)
+		if !isOk {
+			br.Msg = msg
+			br.ErrMsg = errMsg
+			return
+		}
 	default:
 		br.Msg = "错误的图表"
 		br.ErrMsg = "错误的图表"