base_from_kpler.go 28 KB

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