|
@@ -6,6 +6,7 @@ import (
|
|
|
"eta/eta_index_lib/models"
|
|
|
"eta/eta_index_lib/services"
|
|
|
"eta/eta_index_lib/utils"
|
|
|
+ "fmt"
|
|
|
"strconv"
|
|
|
"time"
|
|
|
)
|
|
@@ -62,13 +63,10 @@ func (this *JiaYueController) Add() {
|
|
|
}()
|
|
|
|
|
|
// 校验数据源
|
|
|
- cond := ` AND edb_source_id = ?`
|
|
|
- pars := make([]interface{}, 0)
|
|
|
- pars = append(pars, req.Source)
|
|
|
- sourceItem, e := models.GetEdbSourceItemByCondition(cond, pars)
|
|
|
- if e != nil {
|
|
|
+ sourceItem := models.EdbSourceIdMap[req.Source]
|
|
|
+ if sourceItem == nil {
|
|
|
br.Msg = "数据源有误"
|
|
|
- br.ErrMsg = "获取数据源失败, Err: " + e.Error()
|
|
|
+ br.ErrMsg = fmt.Sprintf("数据源有误, Source: %d", req.Source)
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -78,6 +76,7 @@ func (this *JiaYueController) Add() {
|
|
|
params.SourceExtend = sourceItem.SourceExtend
|
|
|
params.StartDate = utils.BASE_START_DATE
|
|
|
params.EndDate = utils.BASE_END_DATE
|
|
|
+ params.IndexCodeRequired = sourceItem.EdbCodeRequired
|
|
|
indexData, e := services.GetJiaYueIndexDataFromBridge(params)
|
|
|
if e != nil {
|
|
|
br.Msg = "获取指标失败"
|
|
@@ -165,13 +164,10 @@ func (this *JiaYueController) Refresh() {
|
|
|
br.ErrMsg = "查询指标信息失败, Err: " + e.Error()
|
|
|
return
|
|
|
}
|
|
|
- cond := ` AND edb_source_id = ?`
|
|
|
- pars := make([]interface{}, 0)
|
|
|
- pars = append(pars, req.Source)
|
|
|
- sourceItem, e := models.GetEdbSourceItemByCondition(cond, pars)
|
|
|
- if e != nil {
|
|
|
+ sourceItem := models.EdbSourceIdMap[req.Source]
|
|
|
+ if sourceItem == nil {
|
|
|
br.Msg = "数据源有误"
|
|
|
- br.ErrMsg = "获取数据源失败, Err: " + e.Error()
|
|
|
+ br.ErrMsg = fmt.Sprintf("数据源有误, Source: %d", req.Source)
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -181,6 +177,7 @@ func (this *JiaYueController) Refresh() {
|
|
|
params.SourceExtend = sourceItem.SourceExtend
|
|
|
params.StartDate = startDate
|
|
|
params.EndDate = utils.BASE_END_DATE
|
|
|
+ params.IndexCodeRequired = sourceItem.EdbCodeRequired
|
|
|
indexData, e := services.GetJiaYueIndexDataFromBridge(params)
|
|
|
if e != nil {
|
|
|
br.Msg = "获取指标失败"
|