base_from_kpler.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881
  1. package services
  2. import (
  3. "encoding/json"
  4. "eta/eta_index_lib/logic"
  5. "eta/eta_index_lib/models"
  6. "eta/eta_index_lib/services/alarm_msg"
  7. "eta/eta_index_lib/utils"
  8. "fmt"
  9. "io/ioutil"
  10. "net/http"
  11. "strconv"
  12. "strings"
  13. "time"
  14. )
  15. // GetKplerDataByApi 获取开普勒数据
  16. func GetKplerDataByApi(params models.KplerSearchEdbReq, terminalCode string) (indexes []*models.KplerIndexItem, apiQueryUrl string, terminalInfo *models.EdbTerminal, err error) {
  17. terminal, e := GetApiTerminal(utils.DATA_SOURCE_KPLER, terminalCode)
  18. if e != nil {
  19. err = fmt.Errorf("获取开普勒终端配置失败, %v", e)
  20. return
  21. }
  22. if terminal.ServerUrl == "" {
  23. err = fmt.Errorf("开普勒终端地址未配置")
  24. return
  25. }
  26. // 走API
  27. if terminal.IsApi == 1 {
  28. indexes, apiQueryUrl, err = getKplerDataByApi(params, terminal.ServerUrl)
  29. if err != nil {
  30. err = fmt.Errorf("获取开普勒指标数据失败, %v", err)
  31. return
  32. }
  33. terminalInfo = terminal
  34. return
  35. }
  36. return
  37. }
  38. // getEdbDataFromThsHfHttp API-获取高频指标数据
  39. func getKplerDataByApi(params models.KplerSearchEdbReq, serverUrl string) (list []*models.KplerIndexItem, apiQueryUrl string, err error) {
  40. defer func() {
  41. if err != nil {
  42. tips := fmt.Sprintf("开普勒指标API-getKplerDataByApi err: %v", err)
  43. utils.FileLog.Info(tips)
  44. }
  45. }()
  46. // todo 判断指标是否已存在
  47. productNamesStr := params.ProductNames
  48. fromZoneNamesStr := params.FromZoneNames
  49. toZoneNamesStr := params.ToZoneNames
  50. libreq := new(models.KplerFlowDataLibReq)
  51. libreq.Products = productNamesStr
  52. libreq.FromZones = fromZoneNamesStr
  53. libreq.ToZones = toZoneNamesStr
  54. libreq.Split = params.Split
  55. libreq.FlowDirection = params.FlowDirection
  56. libreq.Granularity = params.Granularity
  57. libreq.Unit = params.Unit
  58. libreq.WithIntraRegion = "true"
  59. libreq.WithForecast = "true"
  60. libreq.OnlyRealized = "false"
  61. libreq.StartDate = "2025-01-01"
  62. libreq.EndDate = time.Now().Format(utils.FormatDate)
  63. // 请求接口
  64. apiResp, err := getKplerFlowDataLib(serverUrl, libreq)
  65. if err != nil {
  66. err = fmt.Errorf("开普勒指标API-getKplerDataByApi err: %v", err)
  67. return
  68. }
  69. if apiResp.Ret != 200 {
  70. err = fmt.Errorf("开普勒指标API-状态码: %d, 提示信息: %s", apiResp.Ret, apiResp.ErrMsg)
  71. return
  72. }
  73. indexes := make([]*models.KplerIndexItem, 0)
  74. if len(apiResp.Data) == 0 {
  75. utils.FileLog.Info("开普勒指标API-无数据")
  76. return
  77. }
  78. indexCodeList := make([]string, 0)
  79. existIndexCodeMap := make(map[string]bool)
  80. flowDirection := params.FlowDirection
  81. prefixIndexName := fmt.Sprintf("kpler%s%s%s%s", fromZoneNamesStr, flowDirection, toZoneNamesStr, productNamesStr)
  82. lastIndexName := params.Granularity
  83. prefixIndexCode := fmt.Sprintf("k%s%s%s%s", fromZoneNamesStr, flowDirection, toZoneNamesStr, productNamesStr)
  84. // 获取首字母
  85. lastIndexCode := strings.ToUpper(params.Granularity[:1])
  86. // Tables中的每一个对应一个证券代码
  87. for _, v := range apiResp.Data {
  88. index := new(models.KplerIndexItem)
  89. indexName := ""
  90. indexCode := ""
  91. indexName = fmt.Sprintf("%s%s%s", prefixIndexName, v.SplitItem, lastIndexName)
  92. indexCode = fmt.Sprintf("%s%s%s", prefixIndexCode,v.SplitItem, lastIndexCode)
  93. indexCodeList = append(indexCodeList, indexCode)
  94. index.IndexCode = indexCode
  95. index.IndexName = indexName
  96. index.Frequency = getKplerFrequency(params.Granularity)
  97. index.Unit = params.Unit
  98. index.IndexData = v.IndexData
  99. indexes = append(indexes, index)
  100. }
  101. indexObj := new(models.BaseFromKplerIndex)
  102. // 查询指标是否存在
  103. existList, err := indexObj.GetByIndexCodes(indexCodeList)
  104. if err != nil {
  105. err = fmt.Errorf("查询指标是否存在失败, %v", err)
  106. return
  107. }
  108. for _, v := range existList {
  109. existIndexCodeMap[v.IndexCode] = true
  110. }
  111. // 过滤已经存在的指标
  112. list = make([]*models.KplerIndexItem, 0)
  113. for _, v := range indexes {
  114. if _, ok := existIndexCodeMap[v.IndexCode]; !ok {
  115. list = append(list, v)
  116. }
  117. }
  118. return
  119. }
  120. func AddKplerIndexByApi(indexList []*models.KplerIndexItem, req *models.KplerSearchEdbReq, ApiQueryUrl string, classifyId int, terminalCode string) (err error) {
  121. errMsgList := make([]string, 0)
  122. defer func() {
  123. if len(errMsgList) > 0 {
  124. msg := fmt.Sprint("数据源-Kpler数据处理失败,err:", strings.Join(errMsgList, "\n"))
  125. utils.FileLog.Info(msg)
  126. go alarm_msg.SendAlarmMsg(msg, 3)
  127. }
  128. }()
  129. maxSort := 0
  130. for _, indexInfo := range indexList {
  131. indexObj := new(models.BaseFromKplerIndex)
  132. indexId := 0
  133. //判断指标是否存在
  134. var isAdd int
  135. item, er := indexObj.GetByIndexCode(indexInfo.IndexCode)
  136. if er != nil {
  137. if er.Error() == utils.ErrNoRow() {
  138. isAdd = 1
  139. err = nil
  140. } else {
  141. errMsgList = append(errMsgList, fmt.Sprintf("查询数据源指标库失败 GetByIndexCode Err:%s", er))
  142. continue
  143. }
  144. }
  145. if item != nil && item.BaseFromKplerIndexId > 0 {
  146. fmt.Println("item:", item)
  147. isAdd = 2
  148. } else {
  149. isAdd = 1
  150. }
  151. // 批量新增指标
  152. if isAdd == 1 {
  153. indexObj.IndexCode = indexInfo.IndexCode
  154. indexObj.IndexName = indexInfo.IndexName
  155. indexObj.Frequency = indexInfo.Frequency
  156. indexObj.ClassifyId = int(classifyId)
  157. indexObj.FromZoneName = req.FromZoneNames
  158. indexObj.ToZoneName = req.ToZoneNames
  159. indexObj.FlowDirection = req.FlowDirection
  160. indexObj.Granularity = req.Granularity
  161. indexObj.Split = req.Split
  162. indexObj.Unit = req.Unit
  163. indexObj.Sort = maxSort
  164. indexObj.ApiQueryUrl = ApiQueryUrl
  165. indexObj.ModifyTime = time.Now()
  166. indexObj.CreateTime = time.Now()
  167. indexObj.TerminalCode = terminalCode
  168. err = indexObj.Add()
  169. if err != nil {
  170. err = fmt.Errorf("数据源新增Kpler指标失败 Err:%s", err)
  171. return
  172. }
  173. indexId = indexObj.BaseFromKplerIndexId
  174. maxSort++
  175. } else if isAdd == 2 {
  176. indexId = item.BaseFromKplerIndexId
  177. if item.TerminalCode == `` && terminalCode != `` {
  178. item.TerminalCode = terminalCode
  179. err = item.Update([]string{"TerminalCode"})
  180. if err != nil {
  181. err = fmt.Errorf("数据源更新Kpler指标失败 Err:%s", err)
  182. return
  183. }
  184. }
  185. indexObj.BaseFromKplerIndexId = item.BaseFromKplerIndexId
  186. indexObj.IndexName = indexInfo.IndexName
  187. indexObj.Frequency = indexInfo.Frequency
  188. indexObj.ClassifyId = int(classifyId)
  189. indexObj.ProductNames = req.ProductNames
  190. indexObj.FromZoneName = req.FromZoneNames
  191. indexObj.ToZoneName = req.ToZoneNames
  192. indexObj.FlowDirection = req.FlowDirection
  193. indexObj.Granularity = req.Granularity
  194. indexObj.Split = req.Split
  195. indexObj.Unit = req.Unit
  196. indexObj.ApiQueryUrl = ApiQueryUrl
  197. indexObj.ModifyTime = time.Now()
  198. //修改数据
  199. updateColsArr := make([]string, 0)
  200. updateColsArr = append(updateColsArr, "index_name")
  201. updateColsArr = append(updateColsArr, "classify_id")
  202. updateColsArr = append(updateColsArr, "product_names")
  203. updateColsArr = append(updateColsArr, "from_zone_name")
  204. updateColsArr = append(updateColsArr, "to_zone_name")
  205. updateColsArr = append(updateColsArr, "flow_direction")
  206. updateColsArr = append(updateColsArr, "granularity")
  207. updateColsArr = append(updateColsArr, "split")
  208. updateColsArr = append(updateColsArr, "frequency")
  209. updateColsArr = append(updateColsArr, "modify_time")
  210. e := indexObj.Update(updateColsArr)
  211. if e != nil {
  212. fmt.Println("Index Update Err:" + e.Error())
  213. return
  214. }
  215. }
  216. dataObj := new(models.BaseFromKplerData)
  217. exitDataMap := make(map[string]*models.BaseFromKplerData)
  218. //获取已存在的所有数据
  219. var exitDataList []*models.BaseFromKplerData
  220. exitDataList, err = dataObj.GetByIndexCode(indexInfo.IndexCode)
  221. if err != nil {
  222. err = fmt.Errorf("数据源查询Kpler指标数据失败 Err:%s", err)
  223. return
  224. }
  225. fmt.Println("exitDataListLen:", len(exitDataList))
  226. for _, v := range exitDataList {
  227. dateStr := v.DataTime
  228. exitDataMap[dateStr] = v
  229. }
  230. addDataList := make([]*models.BaseFromKplerData, 0)
  231. // 遍历excel数据,然后跟现有的数据做校验,不存在则入库
  232. for _, value := range indexInfo.IndexData {
  233. date := value.DataTime
  234. value := value.Value
  235. if findData, ok := exitDataMap[date]; !ok {
  236. _, err = time.ParseInLocation(utils.FormatDate, date, time.Local)
  237. if err != nil {
  238. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  239. return
  240. }
  241. var saveDataTime time.Time
  242. if strings.Contains(date, "00:00:00") {
  243. saveDataTime, err = time.Parse(utils.FormatDateTime, date)
  244. } else {
  245. saveDataTime, err = time.Parse(utils.FormatDate, date)
  246. }
  247. if err != nil {
  248. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  249. continue
  250. }
  251. timestamp := saveDataTime.UnixNano() / 1e6
  252. dataItem := new(models.BaseFromKplerData)
  253. dataItem.BaseFromKplerIndexId = int(indexId)
  254. dataItem.IndexCode = indexInfo.IndexCode
  255. dataItem.DataTime = date
  256. dataItem.Value = value
  257. dataItem.CreateTime = time.Now()
  258. dataItem.ModifyTime = time.Now()
  259. dataItem.DataTimestamp = timestamp
  260. addDataList = append(addDataList, dataItem)
  261. if len(addDataList) > 500 {
  262. err = dataObj.AddMulti(addDataList)
  263. if err != nil {
  264. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  265. return
  266. }
  267. addDataList = make([]*models.BaseFromKplerData, 0)
  268. }
  269. } else {
  270. if findData != nil && findData.Value != value && !strings.Contains(value, "#N/A") { //修改数据
  271. // 过滤0.50和0.5的比较
  272. oldV, _ := strconv.ParseFloat(findData.Value, 64)
  273. newV, _ := strconv.ParseFloat(value, 64)
  274. if oldV == newV {
  275. continue
  276. }
  277. dataObj.BaseFromKplerIndexId = findData.BaseFromKplerIndexId
  278. dataObj.Value = value
  279. dataObj.ModifyTime = time.Now()
  280. updateDataColsArr := make([]string, 0)
  281. updateDataColsArr = append(updateDataColsArr, "value")
  282. updateDataColsArr = append(updateDataColsArr, "modify_time")
  283. dataObj.Update(updateDataColsArr)
  284. }
  285. }
  286. }
  287. if len(addDataList) > 0 {
  288. err = dataObj.AddMulti(addDataList)
  289. if err != nil {
  290. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  291. return
  292. }
  293. }
  294. var dateItem *models.EdbInfoMaxAndMinInfo
  295. dateItem, err = dataObj.GetMaxAndMinDateByIndexCode(indexInfo.IndexCode)
  296. if err != nil {
  297. err = fmt.Errorf("查询指标最新日期失败 Err:%s", err)
  298. return
  299. }
  300. go func() {
  301. indexObj.ModifyIndexMaxAndMinDate(indexInfo.IndexCode, dateItem)
  302. }()
  303. // 同步刷新ETA指标库的指标
  304. {
  305. // 获取指标详情
  306. baseObj := new(models.BaseFromKpler)
  307. var edbInfo *models.EdbInfo
  308. edbInfo, err = models.GetEdbInfoByEdbCode(baseObj.GetSource(), indexInfo.IndexCode)
  309. if err != nil {
  310. if !utils.IsErrNoRow(err) {
  311. errMsgList = append(errMsgList, fmt.Sprint("刷新ETA指标异常,指标编码:", indexInfo.IndexCode, err.Error()))
  312. return
  313. } else {
  314. err = nil
  315. }
  316. }
  317. // 已经加入到指标库的话,那么就去更新ETA指标库吧
  318. if edbInfo != nil {
  319. go logic.RefreshBaseEdbInfo(edbInfo, ``)
  320. }
  321. }
  322. }
  323. return
  324. }
  325. // HandleKplerIndex 处理Kpler的excel数据
  326. func HandleKplerIndex(req *models.HandleKplerExcelDataReq) (err error) {
  327. errMsgList := make([]string, 0)
  328. defer func() {
  329. if len(errMsgList) > 0 {
  330. msg := fmt.Sprint("数据源-Kpler数据处理失败,err:", strings.Join(errMsgList, "\n"))
  331. utils.FileLog.Info(msg)
  332. go alarm_msg.SendAlarmMsg(msg, 3)
  333. }
  334. }()
  335. // 查询所有的一级分类
  336. classifyObj := new(models.BaseFromKplerClassify)
  337. classifyList, err := classifyObj.GetParentClassify()
  338. if err != nil {
  339. err = fmt.Errorf("查询一级目录信息失败 Err:%s", err)
  340. return
  341. }
  342. classifyMap := make(map[string]int, 0)
  343. for _, v := range classifyList {
  344. classifyMap[v.ClassifyName] = int(v.ClassifyId)
  345. }
  346. for _, v := range req.List {
  347. if v.IndexName == "" || v.IndexCode == "" {
  348. errMsgList = append(errMsgList, fmt.Sprintf("新增指标异常,指标编码%s或者指标ID%s为空:", v.IndexCode, v.IndexName))
  349. continue
  350. }
  351. err = handleKplerIndex(v, req.TerminalCode, classifyMap)
  352. if err != nil {
  353. errMsgList = append(errMsgList, fmt.Sprintf("新增指标异常,指标编码:%s, Err: %s", v.IndexCode, err))
  354. return
  355. }
  356. }
  357. return
  358. }
  359. // getKplerFlowDataLib 获取开普勒流数据
  360. func getKplerFlowDataLib(libUrl string, dataMap *models.KplerFlowDataLibReq) (resp *models.KplerFlowDataLibResp, err error) {
  361. postUrl := fmt.Sprintf("%s%s", libUrl, "/v1/kpler/getFlowData")
  362. postData, err := json.Marshal(dataMap)
  363. if err != nil {
  364. return
  365. }
  366. result, err := HttpPost(postUrl, string(postData), "application/json")
  367. if err != nil {
  368. return
  369. }
  370. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";" + string(postData) + ";result:" + string(result))
  371. err = json.Unmarshal(result, &resp)
  372. if err != nil {
  373. return
  374. }
  375. return resp, nil
  376. }
  377. func getKplerProductLib(libUrl string) (resp *models.KplerProductLibResp, err error) {
  378. postUrl := fmt.Sprintf("%s%s", libUrl, "/v1/kpler/getProductData")
  379. result, err := HttpPost(postUrl, "", "application/json")
  380. if err != nil {
  381. return
  382. }
  383. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";result:" + string(result))
  384. err = json.Unmarshal(result, &resp)
  385. if err != nil {
  386. return
  387. }
  388. return resp, nil
  389. }
  390. func getKplerZoneLib(libUrl string) (resp *models.KplerZoneLibResp, err error) {
  391. postUrl := fmt.Sprintf("%s%s", libUrl, "/v1/kpler/getZoneData")
  392. result, err := HttpPost(postUrl, "", "application/json")
  393. if err != nil {
  394. return
  395. }
  396. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";result:" + string(result))
  397. err = json.Unmarshal(result, &resp)
  398. if err != nil {
  399. return
  400. }
  401. return resp, nil
  402. }
  403. func handleKplerIndex(req *models.HandleKplerExcelData, terminalCode string, classifyMap map[string]int) (err error) {
  404. indexName := req.IndexName
  405. indexCode := req.IndexCode
  406. excelDataMap := req.ExcelDataMap
  407. errMsgList := make([]string, 0)
  408. defer func() {
  409. if len(errMsgList) > 0 {
  410. msg := fmt.Sprint("数据源-Kpler数据处理失败,err:", strings.Join(errMsgList, "\n"))
  411. utils.FileLog.Info(msg)
  412. go alarm_msg.SendAlarmMsg(msg, 3)
  413. }
  414. }()
  415. indexObj := new(models.BaseFromKplerIndex)
  416. dataObj := new(models.BaseFromKplerData)
  417. classifyObj := new(models.BaseFromKplerClassify)
  418. var indexId int64
  419. addDataList := make([]*models.BaseFromKplerData, 0)
  420. exitDataMap := make(map[string]*models.BaseFromKplerData)
  421. // 修改指标信息
  422. if indexName == "" {
  423. utils.FileLog.Info("未刷新到指标数据:indexName:" + indexName)
  424. return
  425. }
  426. // 判断目录是否存在
  427. var classifyId int64
  428. now := time.Now()
  429. if req.ClassifyName != "" {
  430. classifyParentId := 0
  431. level := 1
  432. classifyParentId, _ = classifyMap[req.ParentClassifyName]
  433. if classifyParentId > 0 {
  434. level = 2
  435. }
  436. classifyObj, err = classifyObj.GetByClassifyName(req.ClassifyName)
  437. if err != nil {
  438. if err.Error() == utils.ErrNoRow() {
  439. //新增分类
  440. classifyObj = &models.BaseFromKplerClassify{
  441. ClassifyName: req.ClassifyName,
  442. ClassifyNameEn: req.ClassifyName,
  443. ParentId: classifyParentId,
  444. SysUserId: 0,
  445. SysUserRealName: "",
  446. Level: level,
  447. Sort: req.ClassifySort,
  448. ModifyTime: now,
  449. CreateTime: now,
  450. }
  451. classifyId, err = classifyObj.Add()
  452. if err != nil {
  453. err = fmt.Errorf("新增分类失败 Err:%s", err)
  454. return
  455. }
  456. classifyObj.ClassifyId = int(classifyId)
  457. } else {
  458. return
  459. }
  460. } else {
  461. classifyId = int64(classifyObj.ClassifyId)
  462. classifyObj.ModifyTime = now
  463. classifyObj.ParentId = classifyParentId
  464. e := classifyObj.Update([]string{"ParentId", "ModifyTime"})
  465. if e != nil {
  466. fmt.Println("classifyObj Update Err:" + e.Error())
  467. return
  468. }
  469. }
  470. }
  471. //判断指标是否存在
  472. var isAdd int
  473. item, err := indexObj.GetByIndexCode(indexCode)
  474. if err != nil {
  475. if err.Error() == utils.ErrNoRow() {
  476. isAdd = 1
  477. err = nil
  478. } else {
  479. isAdd = -1
  480. err = fmt.Errorf("查询数据源指标库失败 GetByIndexCode Err:%s", err)
  481. return
  482. }
  483. }
  484. if item != nil && item.BaseFromKplerIndexId > 0 {
  485. fmt.Println("item:", item)
  486. isAdd = 2
  487. } else {
  488. isAdd = 1
  489. }
  490. if isAdd == 1 {
  491. indexObj.IndexCode = indexCode
  492. indexObj.IndexName = indexName
  493. indexObj.Frequency = req.Frequency
  494. indexObj.ClassifyId = int(classifyId)
  495. indexObj.ProductNames = req.ProductNames
  496. indexObj.FromZoneId = req.FromZoneId
  497. indexObj.FromZoneName = req.FromZoneName
  498. indexObj.ToZoneId = req.ToZoneId
  499. indexObj.ToZoneName = req.ToZoneName
  500. indexObj.FlowDirection = req.FlowDirection
  501. indexObj.Granularity = req.Granularity
  502. indexObj.Split = req.Split
  503. indexObj.Unit = req.Unit
  504. indexObj.Sort = req.Sort
  505. indexObj.ModifyTime = time.Now()
  506. indexObj.CreateTime = time.Now()
  507. indexObj.TerminalCode = terminalCode
  508. err = indexObj.Add()
  509. if err != nil {
  510. err = fmt.Errorf("数据源新增Kpler指标失败 Err:%s", err)
  511. return
  512. }
  513. indexId = int64(indexObj.BaseFromKplerIndexId)
  514. } else if isAdd == 2 {
  515. indexId = int64(item.BaseFromKplerIndexId)
  516. if item.TerminalCode == `` && terminalCode != `` {
  517. item.TerminalCode = terminalCode
  518. err = item.Update([]string{"TerminalCode"})
  519. if err != nil {
  520. err = fmt.Errorf("数据源更新Kpler指标失败 Err:%s", err)
  521. return
  522. }
  523. }
  524. indexObj.BaseFromKplerIndexId = item.BaseFromKplerIndexId
  525. indexObj.IndexName = indexName
  526. indexObj.Frequency = req.Frequency
  527. indexObj.ClassifyId = int(classifyId)
  528. indexObj.ProductNames = req.ProductNames
  529. indexObj.FromZoneId = req.FromZoneId
  530. indexObj.FromZoneName = req.FromZoneName
  531. indexObj.ToZoneId = req.ToZoneId
  532. indexObj.ToZoneName = req.ToZoneName
  533. indexObj.FlowDirection = req.FlowDirection
  534. indexObj.Granularity = req.Granularity
  535. indexObj.Split = req.Split
  536. indexObj.Unit = req.Unit
  537. indexObj.Sort = req.Sort
  538. indexObj.ModifyTime = time.Now()
  539. //修改数据
  540. updateColsArr := make([]string, 0)
  541. updateColsArr = append(updateColsArr, "index_name")
  542. updateColsArr = append(updateColsArr, "classify_id")
  543. updateColsArr = append(updateColsArr, "product_names")
  544. updateColsArr = append(updateColsArr, "from_zone_id")
  545. updateColsArr = append(updateColsArr, "from_zone_name")
  546. updateColsArr = append(updateColsArr, "to_zone_id")
  547. updateColsArr = append(updateColsArr, "to_zone_name")
  548. updateColsArr = append(updateColsArr, "flow_direction")
  549. updateColsArr = append(updateColsArr, "granularity")
  550. updateColsArr = append(updateColsArr, "split")
  551. updateColsArr = append(updateColsArr, "frequency")
  552. updateColsArr = append(updateColsArr, "sort")
  553. updateColsArr = append(updateColsArr, "modify_time")
  554. e := indexObj.Update(updateColsArr)
  555. if e != nil {
  556. fmt.Println("Index Update Err:" + e.Error())
  557. return
  558. }
  559. }
  560. //获取已存在的所有数据
  561. var exitDataList []*models.BaseFromKplerData
  562. exitDataList, err = dataObj.GetByIndexCode(indexCode)
  563. if err != nil {
  564. err = fmt.Errorf("数据源查询Kpler指标数据失败 Err:%s", err)
  565. return
  566. }
  567. fmt.Println("exitDataListLen:", len(exitDataList))
  568. for _, v := range exitDataList {
  569. dateStr := v.DataTime
  570. exitDataMap[dateStr] = v
  571. }
  572. // 遍历excel数据,然后跟现有的数据做校验,不存在则入库
  573. for date, value := range excelDataMap {
  574. if findData, ok := exitDataMap[date]; !ok {
  575. _, err = time.ParseInLocation(utils.FormatDate, date, time.Local)
  576. if err != nil {
  577. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  578. return
  579. }
  580. var saveDataTime time.Time
  581. if strings.Contains(date, "00:00:00") {
  582. saveDataTime, err = time.Parse(utils.FormatDateTime, date)
  583. } else {
  584. saveDataTime, err = time.Parse(utils.FormatDate, date)
  585. }
  586. if err != nil {
  587. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  588. continue
  589. }
  590. timestamp := saveDataTime.UnixNano() / 1e6
  591. dataItem := new(models.BaseFromKplerData)
  592. dataItem.BaseFromKplerIndexId = int(indexId)
  593. dataItem.IndexCode = indexCode
  594. dataItem.DataTime = date
  595. dataItem.Value = value
  596. dataItem.CreateTime = time.Now()
  597. dataItem.ModifyTime = time.Now()
  598. dataItem.DataTimestamp = timestamp
  599. addDataList = append(addDataList, dataItem)
  600. if len(addDataList) > 500 {
  601. err = dataObj.AddMulti(addDataList)
  602. if err != nil {
  603. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  604. return
  605. }
  606. addDataList = make([]*models.BaseFromKplerData, 0)
  607. }
  608. } else {
  609. if findData != nil && findData.Value != value && !strings.Contains(value, "#N/A") { //修改数据
  610. // 过滤0.50和0.5的比较
  611. oldV, _ := strconv.ParseFloat(findData.Value, 64)
  612. newV, _ := strconv.ParseFloat(value, 64)
  613. if oldV == newV {
  614. continue
  615. }
  616. dataObj.BaseFromKplerIndexId = findData.BaseFromKplerIndexId
  617. dataObj.Value = value
  618. dataObj.ModifyTime = time.Now()
  619. updateDataColsArr := make([]string, 0)
  620. updateDataColsArr = append(updateDataColsArr, "value")
  621. updateDataColsArr = append(updateDataColsArr, "modify_time")
  622. dataObj.Update(updateDataColsArr)
  623. }
  624. }
  625. }
  626. if len(addDataList) > 0 {
  627. err = dataObj.AddMulti(addDataList)
  628. if err != nil {
  629. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  630. return
  631. }
  632. }
  633. var dateItem *models.EdbInfoMaxAndMinInfo
  634. dateItem, err = dataObj.GetMaxAndMinDateByIndexCode(indexCode)
  635. if err != nil {
  636. err = fmt.Errorf("查询指标最新日期失败 Err:%s", err)
  637. return
  638. }
  639. go func() {
  640. indexObj.ModifyIndexMaxAndMinDate(indexCode, dateItem)
  641. }()
  642. // 同步刷新ETA指标库的指标
  643. {
  644. // 获取指标详情
  645. baseObj := new(models.BaseFromKpler)
  646. var edbInfo *models.EdbInfo
  647. edbInfo, err = models.GetEdbInfoByEdbCode(baseObj.GetSource(), indexCode)
  648. if err != nil {
  649. if !utils.IsErrNoRow(err) {
  650. errMsgList = append(errMsgList, fmt.Sprint("刷新ETA指标异常,指标编码:", indexCode, err.Error()))
  651. return
  652. } else {
  653. err = nil
  654. }
  655. }
  656. // 已经加入到指标库的话,那么就去更新ETA指标库吧
  657. if edbInfo != nil {
  658. go logic.RefreshBaseEdbInfo(edbInfo, ``)
  659. }
  660. }
  661. return
  662. }
  663. func HttpPost(url, postData string, params ...string) ([]byte, error) {
  664. body := ioutil.NopCloser(strings.NewReader(postData))
  665. client := &http.Client{}
  666. req, err := http.NewRequest("POST", url, body)
  667. if err != nil {
  668. return nil, err
  669. }
  670. contentType := "application/x-www-form-urlencoded;charset=utf-8"
  671. if len(params) > 0 && params[0] != "" {
  672. contentType = params[0]
  673. }
  674. req.Header.Set("Content-Type", contentType)
  675. req.Header.Set("authorization", utils.MD5(utils.APP_EDB_DATA_ANALYSIS+utils.EDB_DATA_ANALYSIS_Md5_KEY))
  676. resp, err := client.Do(req)
  677. if err != nil {
  678. return nil, err
  679. }
  680. defer resp.Body.Close()
  681. b, err := ioutil.ReadAll(resp.Body)
  682. utils.FileLog.Debug("HttpPost:" + string(b))
  683. return b, err
  684. }
  685. func getKplerFrequency(granularity string) (frequency string) {
  686. switch granularity {
  687. case "daily":
  688. return "日度"
  689. case "weekly":
  690. return "周度"
  691. case "monthly":
  692. return "月度"
  693. case "yearly":
  694. return "年度"
  695. }
  696. return ""
  697. }
  698. func InitKplerProduct ()(err error) {
  699. libUrl := "http://127.0.0.1:8915"
  700. libResp, err := getKplerProductLib(libUrl)
  701. if err != nil {
  702. return
  703. }
  704. if libResp.Ret != 200 {
  705. fmt.Println("获取开普勒产品库失败", libResp.ErrMsg)
  706. utils.FileLog.Info("获取开普勒产品库失败", libResp.ErrMsg)
  707. return
  708. }
  709. classifyList, err := models.GetAllBaseFromKplerClassify()
  710. if err != nil {
  711. return
  712. }
  713. classifyMap := make(map[int]*models.BaseFromKplerClassify)
  714. for _, v := range classifyList {
  715. classifyMap[v.ProductId] = v
  716. }
  717. // 批量插入分类表中
  718. // family > group(commodity) > product() > grade
  719. sort := 0
  720. for _, v := range libResp.Data {
  721. id, _ := strconv.Atoi(v.Id)
  722. if classifyObj, ok := classifyMap[id]; !ok {
  723. classifyObj = new(models.BaseFromKplerClassify)
  724. classifyObj.ClassifyName = v.Name
  725. classifyObj.ClassifyNameEn = v.Name
  726. classifyObj.ModifyTime = time.Now()
  727. classifyObj.CreateTime = time.Now()
  728. classifyObj.Sort = sort
  729. classifyObj.Level = 1
  730. classifyObj.ParentId = 0
  731. classifyObj.SysUserId = 0
  732. classifyObj.SysUserRealName = ""
  733. classifyObj.ProductId = id
  734. classifyObj.ProductName = v.Name
  735. classifyObj.AncestorFamilyId, _ = strconv.Atoi(v.FamilyId)
  736. classifyObj.AncestorFamilyName = v.Family
  737. classifyObj.AncestorGroupId, _ = strconv.Atoi(v.GroupId)
  738. classifyObj.AncestorGroupName = v.Group
  739. classifyObj.AncestorProductId, _ = strconv.Atoi(v.ProductId)
  740. classifyObj.AncestorProductName = v.Product
  741. classifyObj.AncestorGradeId, _ = strconv.Atoi(v.GradeId)
  742. classifyObj.AncestorGradeName = v.Grade
  743. classifyObj.ClassifyType = v.Type
  744. _, err = classifyObj.Add()
  745. if err != nil {
  746. fmt.Println("新增开普勒产品库失败", err)
  747. utils.FileLog.Info("新增开普勒产品库失败", err)
  748. }
  749. classifyMap[id] = classifyObj
  750. }
  751. }
  752. for _, v := range classifyMap {
  753. if v.ParentId == 0 {
  754. if v.AncestorGradeId > 0 && v.AncestorGradeName != v.ClassifyName {
  755. v.ParentId = classifyMap[v.AncestorGradeId].ClassifyId
  756. } else if v.AncestorProductId > 0 && v.AncestorProductName != v.ClassifyName {
  757. v.ParentId = classifyMap[v.AncestorProductId].ClassifyId
  758. } else if v.AncestorGroupId > 0 && v.AncestorGroupName != v.ClassifyName {
  759. v.ParentId = classifyMap[v.AncestorGroupId].ClassifyId
  760. }
  761. err = v.Update([]string{"ParentId"})
  762. if err != nil {
  763. fmt.Println("更新开普勒产品库失败", err)
  764. utils.FileLog.Info("更新开普勒产品库失败", err)
  765. }
  766. }
  767. }
  768. fmt.Println("classifyList:", classifyList)
  769. return
  770. }
  771. func InitKplerZone() (err error) {
  772. libUrl := "http://127.0.0.1:8915"
  773. libResp, err := getKplerZoneLib(libUrl)
  774. if err != nil {
  775. return
  776. }
  777. if libResp.Ret != 200 {
  778. fmt.Println("获取开普勒区域库失败", libResp.ErrMsg)
  779. utils.FileLog.Info("获取开普勒区域库失败", libResp.ErrMsg)
  780. return
  781. }
  782. // 批量插入区域表中
  783. zoneMap := make(map[int]*models.BaseFromKplerZone)
  784. for _, v := range libResp.Data {
  785. zoneObj := new(models.BaseFromKplerZone)
  786. descendantId, _ := strconv.Atoi(v.DescendantId)
  787. ancestorId, _ := strconv.Atoi(v.AncestorId)
  788. if _, ok := zoneMap[descendantId]; !ok {
  789. zoneObj.ZoneName = v.DescendantName
  790. //zoneObj.ZoneType = v.DescendantType
  791. zoneObj.AncestorId = ancestorId
  792. zoneObj.AncestorType = v.AncestorType
  793. zoneObj.AncestorName = v.AncestorName
  794. zoneObj.DescendantId = descendantId
  795. zoneObj.DescendantName = v.DescendantName
  796. zoneObj.ModifyTime = time.Now()
  797. zoneObj.CreateTime = time.Now()
  798. _, err = zoneObj.Add()
  799. if err != nil {
  800. fmt.Println("新增开普勒区域库失败", err)
  801. utils.FileLog.Info("新增开普勒区域库失败", err)
  802. }
  803. zoneMap[descendantId] = zoneObj
  804. if _, ok := zoneMap[ancestorId]; !ok {
  805. zoneObj := new(models.BaseFromKplerZone)
  806. zoneObj.ZoneName = v.AncestorName
  807. zoneObj.ZoneType = v.AncestorType
  808. zoneObj.DescendantId = ancestorId
  809. zoneObj.DescendantName = v.AncestorName
  810. zoneObj.ModifyTime = time.Now()
  811. zoneObj.CreateTime = time.Now()
  812. _, err = zoneObj.Add()
  813. if err != nil {
  814. fmt.Println("新增开普勒区域库失败", err)
  815. utils.FileLog.Info("新增开普勒区域库失败", err)
  816. }
  817. zoneMap[ancestorId] = zoneObj
  818. }
  819. }
  820. }
  821. return
  822. }