Browse Source

同步指标数据

xyxie 3 months ago
parent
commit
ddc84cae2a
2 changed files with 20 additions and 7 deletions
  1. 6 0
      models/data_manage/base_from_gpr_risk.go
  2. 14 7
      services/gpr_risk.go

+ 6 - 0
models/data_manage/base_from_gpr_risk.go

@@ -146,3 +146,9 @@ func MultiAddBaseFromGprRiskDataIndex(items []*BaseFromGprRiskData) (lastId int6
 
 	return
 }
+
+func (item *BaseFromGprRiskData) Update() (lastId int64, err error) {
+	o := orm.NewOrm()
+	lastId, err = o.Update(item)
+	return
+}

+ 14 - 7
services/gpr_risk.go

@@ -101,15 +101,15 @@ func SyncBaseFromGprRiskData() (err error) {
 
 	method := `index/data/list_page`
 
-	existDataMap := make(map[string]*data_manage.BaseFromGprRiskData)
-	allData, err1 := data_manage.GetAllBaseFromGprRiskDataList(startDate)
+	existDataMap := make(map[int]*data_manage.BaseFromGprRiskData)
+	startDateBefore := time.Now().AddDate(0, -3, 0).Format(utils.FormatDate) + " 00:00:00"
+	allData, err1 := data_manage.GetAllBaseFromGprRiskDataList(startDateBefore)
 	if err1 != nil {
 		fmt.Println("get GetAllBaseFromGprRiskDataList err:" + err1.Error())
 		return
 	}
 	for _, dv := range allData {
-		tmpKey := dv.IndexCode + "_" + dv.DataTime
-		existDataMap[tmpKey] = dv
+		existDataMap[dv.BaseFromGprRiskDataId] = dv
 	}
 
 	//获取所有指标信息  某一天的
@@ -143,10 +143,16 @@ func SyncBaseFromGprRiskData() (err error) {
 		addDataList := make([]*data_manage.BaseFromGprRiskData, 0)
 		if respObj.Data.List != nil && len(respObj.Data.List) > 0 {
 			for _, dv := range respObj.Data.List {
-				tmpKey := dv.IndexCode + "_" + dv.DataTime
-				if _, ok := existDataMap[tmpKey]; !ok {
+				//tmpKey := dv.IndexCode + "_" + dv.DataTime
+				if _, ok := existDataMap[dv.BaseFromGprRiskDataId]; !ok {
 					addDataList = append(addDataList, dv)
-					existDataMap[tmpKey] = dv
+				} else {
+					//更新指标数据
+					_, err = dv.Update()
+					if err != nil {
+						utils.FileLog.Info("UpdateBaseFromGprRiskData error:" + err.Error())
+						fmt.Println("UpdateBaseFromGprRiskDataIndex error:", err)
+					}
 				}
 			}
 		}
@@ -155,6 +161,7 @@ func SyncBaseFromGprRiskData() (err error) {
 		if len(addDataList) > 0 {
 			_, err = data_manage.MultiAddBaseFromGprRiskDataIndex(addDataList)
 			if err != nil {
+				utils.FileLog.Info("MultiAddBaseFromGprRiskDataIndex error:" + err.Error())
 				fmt.Println("MultiAddBaseFromGprRiskDataIndex error:", err)
 			}
 		}