base_from_kpler.go 31 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037
  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 = "2013-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. fromZoneIdsStr := ""
  154. toZoneIdsStr := ""
  155. if len(req.FromZoneIds) > 0 {
  156. for _, v := range req.FromZoneIds {
  157. fromZoneIdsStr = fmt.Sprintf("%s%d,", fromZoneIdsStr, v)
  158. }
  159. }
  160. if len(req.ToZoneIds) > 0 {
  161. for _, v := range req.ToZoneIds {
  162. toZoneIdsStr = fmt.Sprintf("%s%d,", toZoneIdsStr, 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 = fromZoneIdsStr
  173. indexObj.ToZoneId = toZoneIdsStr
  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.FromZoneId = fromZoneIdsStr
  208. indexObj.ToZoneId = toZoneIdsStr
  209. indexObj.FromZoneName = req.FromZoneNames
  210. indexObj.ToZoneName = req.ToZoneNames
  211. indexObj.FlowDirection = req.FlowDirection
  212. indexObj.Granularity = GetKplerGranularity(req.Granularity)
  213. indexObj.Split = req.Split
  214. indexObj.Unit = req.Unit
  215. indexObj.ApiQueryUrl = apiQueryUrl
  216. indexObj.ModifyTime = time.Now()
  217. //修改数据
  218. updateColsArr := make([]string, 0)
  219. updateColsArr = append(updateColsArr, "index_name")
  220. updateColsArr = append(updateColsArr, "classify_id")
  221. updateColsArr = append(updateColsArr, "product_names")
  222. updateColsArr = append(updateColsArr, "from_zone_id")
  223. updateColsArr = append(updateColsArr, "to_zone_id")
  224. updateColsArr = append(updateColsArr, "from_zone_name")
  225. updateColsArr = append(updateColsArr, "to_zone_name")
  226. updateColsArr = append(updateColsArr, "flow_direction")
  227. updateColsArr = append(updateColsArr, "granularity")
  228. updateColsArr = append(updateColsArr, "split")
  229. updateColsArr = append(updateColsArr, "frequency")
  230. updateColsArr = append(updateColsArr, "modify_time")
  231. updateColsArr = append(updateColsArr, "api_query_url")
  232. e := indexObj.Update(updateColsArr)
  233. if e != nil {
  234. fmt.Println("Index Update Err:" + e.Error())
  235. return
  236. }
  237. }
  238. dataObj := new(models.BaseFromKplerData)
  239. exitDataMap := make(map[string]*models.BaseFromKplerData)
  240. //获取已存在的所有数据
  241. var exitDataList []*models.BaseFromKplerData
  242. exitDataList, err = dataObj.GetByIndexCode(indexInfo.IndexCode)
  243. if err != nil {
  244. err = fmt.Errorf("数据源查询Kpler指标数据失败 Err:%s", err)
  245. return
  246. }
  247. fmt.Println("exitDataListLen:", len(exitDataList))
  248. for _, v := range exitDataList {
  249. dateStr := v.DataTime
  250. exitDataMap[dateStr] = v
  251. }
  252. addDataList := make([]*models.BaseFromKplerData, 0)
  253. // 遍历excel数据,然后跟现有的数据做校验,不存在则入库
  254. for _, value := range indexInfo.IndexData {
  255. date := value.DataTime
  256. value := value.Value
  257. if findData, ok := exitDataMap[date]; !ok {
  258. _, err = time.ParseInLocation(utils.FormatDate, date, time.Local)
  259. if err != nil {
  260. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  261. return
  262. }
  263. var saveDataTime time.Time
  264. if strings.Contains(date, "00:00:00") {
  265. saveDataTime, err = time.Parse(utils.FormatDateTime, date)
  266. } else {
  267. saveDataTime, err = time.Parse(utils.FormatDate, date)
  268. }
  269. if err != nil {
  270. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  271. continue
  272. }
  273. timestamp := saveDataTime.UnixNano() / 1e6
  274. dataItem := new(models.BaseFromKplerData)
  275. dataItem.BaseFromKplerIndexId = int(indexId)
  276. dataItem.IndexCode = indexInfo.IndexCode
  277. dataItem.DataTime = date
  278. dataItem.Value = value
  279. dataItem.CreateTime = time.Now()
  280. dataItem.ModifyTime = time.Now()
  281. dataItem.DataTimestamp = timestamp
  282. addDataList = append(addDataList, dataItem)
  283. if len(addDataList) > 500 {
  284. err = dataObj.AddMulti(addDataList)
  285. if err != nil {
  286. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  287. return
  288. }
  289. addDataList = make([]*models.BaseFromKplerData, 0)
  290. }
  291. } else {
  292. if findData != nil && findData.Value != value && !strings.Contains(value, "#N/A") { //修改数据
  293. // 过滤0.50和0.5的比较
  294. oldV, _ := strconv.ParseFloat(findData.Value, 64)
  295. newV, _ := strconv.ParseFloat(value, 64)
  296. if oldV == newV {
  297. continue
  298. }
  299. dataObj.BaseFromKplerIndexId = findData.BaseFromKplerIndexId
  300. dataObj.Value = value
  301. dataObj.ModifyTime = time.Now()
  302. updateDataColsArr := make([]string, 0)
  303. updateDataColsArr = append(updateDataColsArr, "value")
  304. updateDataColsArr = append(updateDataColsArr, "modify_time")
  305. dataObj.Update(updateDataColsArr)
  306. }
  307. }
  308. }
  309. if len(addDataList) > 0 {
  310. err = dataObj.AddMulti(addDataList)
  311. if err != nil {
  312. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  313. return
  314. }
  315. }
  316. var dateItem *models.EdbInfoMaxAndMinInfo
  317. dateItem, err = dataObj.GetMaxAndMinDateByIndexCode(indexInfo.IndexCode)
  318. if err != nil {
  319. err = fmt.Errorf("查询指标最新日期失败 Err:%s", err)
  320. return
  321. }
  322. go func() {
  323. indexObj.ModifyIndexMaxAndMinDate(indexInfo.IndexCode, dateItem)
  324. }()
  325. // 同步刷新ETA指标库的指标
  326. {
  327. // 获取指标详情
  328. baseObj := new(models.BaseFromKpler)
  329. var edbInfo *models.EdbInfo
  330. edbInfo, err = models.GetEdbInfoByEdbCode(baseObj.GetSource(), indexInfo.IndexCode)
  331. if err != nil {
  332. if !utils.IsErrNoRow(err) {
  333. errMsgList = append(errMsgList, fmt.Sprint("刷新ETA指标异常,指标编码:", indexInfo.IndexCode, err.Error()))
  334. return
  335. } else {
  336. err = nil
  337. }
  338. }
  339. // 已经加入到指标库的话,那么就去更新ETA指标库吧
  340. if edbInfo != nil {
  341. go logic.RefreshBaseEdbInfo(edbInfo, ``)
  342. }
  343. }
  344. }
  345. return
  346. }
  347. // HandleKplerIndex 处理Kpler的excel数据
  348. func HandleKplerIndex(req *models.HandleKplerExcelDataReq) (err error) {
  349. errMsgList := make([]string, 0)
  350. defer func() {
  351. if len(errMsgList) > 0 {
  352. msg := fmt.Sprint("数据源-Kpler数据处理失败,err:", strings.Join(errMsgList, "\n"))
  353. utils.FileLog.Info(msg)
  354. go alarm_msg.SendAlarmMsg(msg, 3)
  355. }
  356. }()
  357. // 查询所有的一级分类
  358. classifyObj := new(models.BaseFromKplerClassify)
  359. classifyList, err := classifyObj.GetParentClassify()
  360. if err != nil {
  361. err = fmt.Errorf("查询一级目录信息失败 Err:%s", err)
  362. return
  363. }
  364. classifyMap := make(map[string]int, 0)
  365. for _, v := range classifyList {
  366. classifyMap[v.ClassifyName] = int(v.ClassifyId)
  367. }
  368. for _, v := range req.List {
  369. err = handleKplerIndex(v, req.TerminalCode, classifyMap)
  370. if err != nil {
  371. errMsgList = append(errMsgList, fmt.Sprintf("新增指标异常,指标编码:%s, Err: %s", v.IndexCode, err))
  372. return
  373. }
  374. }
  375. return
  376. }
  377. // getKplerFlowDataLib 获取开普勒流数据
  378. func getKplerFlowDataLib(libUrl string, dataMap *models.KplerFlowDataLibReq) (resp *models.KplerFlowDataLibResp, err error) {
  379. postUrl := fmt.Sprintf("%s%s", libUrl, "/v1/kpler/getFlowData")
  380. postData, err := json.Marshal(dataMap)
  381. if err != nil {
  382. return
  383. }
  384. result, err := HttpPost(postUrl, string(postData), "application/json")
  385. if err != nil {
  386. return
  387. }
  388. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";" + string(postData) + ";result:" + string(result))
  389. err = json.Unmarshal(result, &resp)
  390. if err != nil {
  391. return
  392. }
  393. return resp, nil
  394. }
  395. func getKplerProductLib(libUrl string, req *models.KplerProductLibReq) (resp *models.KplerProductLibResp, err error) {
  396. postUrl := fmt.Sprintf("%s%s", libUrl, "/v1/kpler/getProductData")
  397. postData, err := json.Marshal(req)
  398. if err != nil {
  399. return
  400. }
  401. result, err := HttpPost(postUrl, string(postData), "application/json")
  402. if err != nil {
  403. return
  404. }
  405. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";result:" + string(result))
  406. err = json.Unmarshal(result, &resp)
  407. if err != nil {
  408. return
  409. }
  410. return resp, nil
  411. }
  412. func getKplerZoneLib(libUrl string) (resp *models.KplerZoneLibResp, err error) {
  413. postUrl := fmt.Sprintf("%s%s", libUrl, "/v1/kpler/getZoneData")
  414. result, err := HttpPost(postUrl, "", "application/json")
  415. if err != nil {
  416. return
  417. }
  418. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";result:" + string(result))
  419. err = json.Unmarshal(result, &resp)
  420. if err != nil {
  421. return
  422. }
  423. return resp, nil
  424. }
  425. func handleKplerIndex(req *models.HandleKplerExcelData, terminalCode string, classifyMap map[string]int) (err error) {
  426. indexName := req.IndexName
  427. indexCode := req.IndexCode
  428. excelDataMap := req.ExcelDataMap
  429. errMsgList := make([]string, 0)
  430. defer func() {
  431. if len(errMsgList) > 0 {
  432. msg := fmt.Sprint("数据源-Kpler数据处理失败,err:", strings.Join(errMsgList, "\n"))
  433. utils.FileLog.Info(msg)
  434. go alarm_msg.SendAlarmMsg(msg, 3)
  435. }
  436. }()
  437. indexObj := new(models.BaseFromKplerIndex)
  438. dataObj := new(models.BaseFromKplerData)
  439. classifyObj := new(models.BaseFromKplerClassify)
  440. var indexId int64
  441. addDataList := make([]*models.BaseFromKplerData, 0)
  442. exitDataMap := make(map[string]*models.BaseFromKplerData)
  443. // 修改指标信息
  444. if indexName == "" {
  445. utils.FileLog.Info("未刷新到指标数据:indexName:" + indexName)
  446. return
  447. }
  448. if req.ClassifyName == "" {
  449. // 获取产品的名称,把产品的第一个名称作为分类ID
  450. productNames := strings.Split(req.ProductNames, ",")
  451. if len(productNames) > 0 {
  452. req.ClassifyName = productNames[0]
  453. }
  454. }
  455. fromZoneNamesStr := req.FromZoneNames
  456. flowDirection := req.FlowDirection
  457. toZoneNamesStr := req.ToZoneNames
  458. productNamesStr := req.ProductNames
  459. // 查询fromZoneIds
  460. zoneObj := new(models.BaseFromKplerZone)
  461. fromZoneNames := strings.Split(fromZoneNamesStr, ",")
  462. fromZoneIdsStr := ""
  463. // 查询 zone
  464. if len(fromZoneNames) > 0 {
  465. fromZoneIds, er := zoneObj.GetIdsByZoneNames(fromZoneNames)
  466. if er != nil {
  467. err = fmt.Errorf("查询区域失败 Err:%s", err)
  468. return
  469. }
  470. for _, v := range fromZoneIds {
  471. fromZoneIdsStr = fmt.Sprintf("%s%d,", fromZoneIdsStr, v)
  472. }
  473. }
  474. // 查询toZoneIds
  475. toZoneNames := strings.Split(toZoneNamesStr, ",")
  476. toZoneIdsStr := ""
  477. if len(toZoneNames) > 0 {
  478. toZoneIds, er := zoneObj.GetIdsByZoneNames(toZoneNames)
  479. if er != nil {
  480. err = fmt.Errorf("查询区域失败 Err:%s", err)
  481. return
  482. }
  483. for _, v := range toZoneIds {
  484. toZoneIdsStr = fmt.Sprintf("%s%d,", toZoneIdsStr, v)
  485. }
  486. }
  487. // 拼接指标编码
  488. prefixIndexCode := fmt.Sprintf("k%s%s%s%s", fromZoneNamesStr, flowDirection, toZoneNamesStr, productNamesStr)
  489. // 获取首字母
  490. lastIndexCode := strings.ToUpper(req.Granularity[:1])
  491. indexCode = fmt.Sprintf("%s%s%s", prefixIndexCode,req.SplitName, lastIndexCode)
  492. // 判断目录是否存在
  493. var classifyId int64
  494. now := time.Now()
  495. if req.ClassifyName != "" {
  496. classifyParentId := 0
  497. level := 1
  498. classifyParentId, _ = classifyMap[req.ParentClassifyName]
  499. if classifyParentId > 0 {
  500. level = 2
  501. }
  502. classifyObj, err = classifyObj.GetByClassifyName(req.ClassifyName)
  503. if err != nil {
  504. if err.Error() == utils.ErrNoRow() {
  505. //新增分类
  506. classifyObj = &models.BaseFromKplerClassify{
  507. ClassifyName: req.ClassifyName,
  508. ClassifyNameEn: req.ClassifyName,
  509. ParentId: classifyParentId,
  510. SysUserId: 0,
  511. SysUserRealName: "",
  512. Level: level,
  513. Sort: req.ClassifySort,
  514. ModifyTime: now,
  515. CreateTime: now,
  516. }
  517. classifyId, err = classifyObj.Add()
  518. if err != nil {
  519. err = fmt.Errorf("新增分类失败 Err:%s", err)
  520. return
  521. }
  522. classifyObj.ClassifyId = int(classifyId)
  523. } else {
  524. return
  525. }
  526. }
  527. }
  528. //判断指标是否存在
  529. var isAdd int
  530. item, err := indexObj.GetByIndexCode(indexCode)
  531. if err != nil {
  532. if err.Error() == utils.ErrNoRow() {
  533. isAdd = 1
  534. err = nil
  535. } else {
  536. isAdd = -1
  537. err = fmt.Errorf("查询数据源指标库失败 GetByIndexCode Err:%s", err)
  538. return
  539. }
  540. }
  541. if item != nil && item.BaseFromKplerIndexId > 0 {
  542. fmt.Println("item:", item)
  543. isAdd = 2
  544. } else {
  545. isAdd = 1
  546. }
  547. if isAdd == 1 {
  548. indexObj.IndexCode = indexCode
  549. indexObj.IndexName = indexName
  550. indexObj.Frequency = getKplerFrequency(req.Granularity)
  551. indexObj.ClassifyId = int(classifyId)
  552. indexObj.ProductNames = req.ProductNames
  553. indexObj.FromZoneId = fromZoneIdsStr
  554. indexObj.ToZoneId = toZoneIdsStr
  555. indexObj.FromZoneName = fromZoneNamesStr
  556. indexObj.ToZoneName = toZoneNamesStr
  557. indexObj.FlowDirection = req.FlowDirection
  558. indexObj.Granularity = req.Granularity
  559. indexObj.Split = req.Split
  560. indexObj.Unit = req.Unit
  561. indexObj.Sort = req.Sort
  562. indexObj.ExcelQueryUrl = req.ExcelQueryUrl
  563. indexObj.ModifyTime = time.Now()
  564. indexObj.CreateTime = time.Now()
  565. indexObj.TerminalCode = terminalCode
  566. err = indexObj.Add()
  567. if err != nil {
  568. err = fmt.Errorf("数据源新增Kpler指标失败 Err:%s", err)
  569. return
  570. }
  571. indexId = int64(indexObj.BaseFromKplerIndexId)
  572. } else if isAdd == 2 {
  573. indexId = int64(item.BaseFromKplerIndexId)
  574. if item.TerminalCode == `` && terminalCode != `` {
  575. item.TerminalCode = terminalCode
  576. err = item.Update([]string{"TerminalCode"})
  577. if err != nil {
  578. err = fmt.Errorf("数据源更新Kpler指标失败 Err:%s", err)
  579. return
  580. }
  581. }
  582. indexObj.BaseFromKplerIndexId = item.BaseFromKplerIndexId
  583. indexObj.IndexName = indexName
  584. indexObj.Frequency = getKplerFrequency(req.Granularity)
  585. indexObj.ClassifyId = int(classifyId)
  586. indexObj.ProductNames = req.ProductNames
  587. indexObj.FromZoneName = fromZoneNamesStr
  588. indexObj.ToZoneName = toZoneNamesStr
  589. indexObj.FlowDirection = req.FlowDirection
  590. indexObj.Granularity = req.Granularity
  591. indexObj.Split = req.Split
  592. indexObj.ExcelQueryUrl = req.ExcelQueryUrl
  593. indexObj.Unit = req.Unit
  594. indexObj.Sort = req.Sort
  595. indexObj.ModifyTime = time.Now()
  596. //修改数据
  597. updateColsArr := make([]string, 0)
  598. updateColsArr = append(updateColsArr, "index_name")
  599. updateColsArr = append(updateColsArr, "classify_id")
  600. updateColsArr = append(updateColsArr, "product_names")
  601. updateColsArr = append(updateColsArr, "from_zone_id")
  602. updateColsArr = append(updateColsArr, "from_zone_name")
  603. updateColsArr = append(updateColsArr, "to_zone_id")
  604. updateColsArr = append(updateColsArr, "to_zone_name")
  605. updateColsArr = append(updateColsArr, "flow_direction")
  606. updateColsArr = append(updateColsArr, "granularity")
  607. updateColsArr = append(updateColsArr, "split")
  608. updateColsArr = append(updateColsArr, "frequency")
  609. updateColsArr = append(updateColsArr, "sort")
  610. updateColsArr = append(updateColsArr, "modify_time")
  611. e := indexObj.Update(updateColsArr)
  612. if e != nil {
  613. fmt.Println("Index Update Err:" + e.Error())
  614. return
  615. }
  616. }
  617. //获取已存在的所有数据
  618. var exitDataList []*models.BaseFromKplerData
  619. exitDataList, err = dataObj.GetByIndexCode(indexCode)
  620. if err != nil {
  621. err = fmt.Errorf("数据源查询Kpler指标数据失败 Err:%s", err)
  622. return
  623. }
  624. fmt.Println("exitDataListLen:", len(exitDataList))
  625. for _, v := range exitDataList {
  626. dateStr := v.DataTime
  627. exitDataMap[dateStr] = v
  628. }
  629. // 遍历excel数据,然后跟现有的数据做校验,不存在则入库
  630. for date, value := range excelDataMap {
  631. if findData, ok := exitDataMap[date]; !ok {
  632. _, err = time.ParseInLocation(utils.FormatDate, date, time.Local)
  633. if err != nil {
  634. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  635. return
  636. }
  637. var saveDataTime time.Time
  638. if strings.Contains(date, "00:00:00") {
  639. saveDataTime, err = time.Parse(utils.FormatDateTime, date)
  640. } else {
  641. saveDataTime, err = time.Parse(utils.FormatDate, date)
  642. }
  643. if err != nil {
  644. err = fmt.Errorf("%s 转换日期格式失败 Err:%s", date, err)
  645. continue
  646. }
  647. timestamp := saveDataTime.UnixNano() / 1e6
  648. dataItem := new(models.BaseFromKplerData)
  649. dataItem.BaseFromKplerIndexId = int(indexId)
  650. dataItem.IndexCode = indexCode
  651. dataItem.DataTime = date
  652. dataItem.Value = value
  653. dataItem.CreateTime = time.Now()
  654. dataItem.ModifyTime = time.Now()
  655. dataItem.DataTimestamp = timestamp
  656. addDataList = append(addDataList, dataItem)
  657. if len(addDataList) > 500 {
  658. err = dataObj.AddMulti(addDataList)
  659. if err != nil {
  660. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  661. return
  662. }
  663. addDataList = make([]*models.BaseFromKplerData, 0)
  664. }
  665. } else {
  666. if findData != nil && findData.Value != value && !strings.Contains(value, "#N/A") { //修改数据
  667. // 过滤0.50和0.5的比较
  668. oldV, _ := strconv.ParseFloat(findData.Value, 64)
  669. newV, _ := strconv.ParseFloat(value, 64)
  670. if oldV == newV {
  671. continue
  672. }
  673. dataObj.BaseFromKplerIndexId = findData.BaseFromKplerIndexId
  674. dataObj.Value = value
  675. dataObj.ModifyTime = time.Now()
  676. updateDataColsArr := make([]string, 0)
  677. updateDataColsArr = append(updateDataColsArr, "value")
  678. updateDataColsArr = append(updateDataColsArr, "modify_time")
  679. dataObj.Update(updateDataColsArr)
  680. }
  681. }
  682. }
  683. if len(addDataList) > 0 {
  684. err = dataObj.AddMulti(addDataList)
  685. if err != nil {
  686. err = fmt.Errorf("批量新增指标失败 Err:%s", err)
  687. return
  688. }
  689. }
  690. var dateItem *models.EdbInfoMaxAndMinInfo
  691. dateItem, err = dataObj.GetMaxAndMinDateByIndexCode(indexCode)
  692. if err != nil {
  693. err = fmt.Errorf("查询指标最新日期失败 Err:%s", err)
  694. return
  695. }
  696. go func() {
  697. indexObj.ModifyIndexMaxAndMinDate(indexCode, dateItem)
  698. }()
  699. // 同步刷新ETA指标库的指标
  700. {
  701. // 获取指标详情
  702. baseObj := new(models.BaseFromKpler)
  703. var edbInfo *models.EdbInfo
  704. edbInfo, err = models.GetEdbInfoByEdbCode(baseObj.GetSource(), indexCode)
  705. if err != nil {
  706. if !utils.IsErrNoRow(err) {
  707. errMsgList = append(errMsgList, fmt.Sprint("刷新ETA指标异常,指标编码:", indexCode, err.Error()))
  708. return
  709. } else {
  710. err = nil
  711. }
  712. }
  713. // 已经加入到指标库的话,那么就去更新ETA指标库吧
  714. if edbInfo != nil {
  715. go logic.RefreshBaseEdbInfo(edbInfo, ``)
  716. }
  717. }
  718. return
  719. }
  720. func HttpPost(url, postData string, params ...string) ([]byte, error) {
  721. body := ioutil.NopCloser(strings.NewReader(postData))
  722. client := &http.Client{}
  723. req, err := http.NewRequest("POST", url, body)
  724. if err != nil {
  725. return nil, err
  726. }
  727. contentType := "application/x-www-form-urlencoded;charset=utf-8"
  728. if len(params) > 0 && params[0] != "" {
  729. contentType = params[0]
  730. }
  731. req.Header.Set("Content-Type", contentType)
  732. req.Header.Set("authorization", utils.MD5(utils.APP_EDB_DATA_ANALYSIS+utils.EDB_DATA_ANALYSIS_Md5_KEY))
  733. resp, err := client.Do(req)
  734. if err != nil {
  735. return nil, err
  736. }
  737. defer resp.Body.Close()
  738. b, err := ioutil.ReadAll(resp.Body)
  739. utils.FileLog.Debug("HttpPost:" + string(b))
  740. return b, err
  741. }
  742. func getKplerFrequency(granularity string) (frequency string) {
  743. switch granularity {
  744. case "daily", "Daily", "days":
  745. return "日度"
  746. case "weekly", "Weekly", "weeks":
  747. return "周度"
  748. case "monthly", "Monthly", "months":
  749. return "月度"
  750. case "yearly", "Yearly", "years":
  751. return "年度"
  752. }
  753. return ""
  754. }
  755. func GetKplerGranularity(frequency string) (granularity string) {
  756. switch frequency {
  757. case "daily", "Daily", "days":
  758. return "days"
  759. case "weekly", "Weekly", "weeks":
  760. return "weeks"
  761. case "monthly", "Monthly", "months":
  762. return "months"
  763. case "yearly", "Yearly", "years":
  764. return "years"
  765. case "quarterly", "Quarters", "quarters":
  766. return "quarters"
  767. }
  768. return ""
  769. }
  770. func InitKplerProduct ()(err error) {
  771. libUrl := "http://127.0.0.1:8915"
  772. libResp, err := getKplerProductLib(libUrl, &models.KplerProductLibReq{
  773. AncestorFamilyIds: "",
  774. AncestorFamilyNames: "",
  775. AncestorGroupIds: "",
  776. AncestorGroupNames: "",
  777. AncestorProductIds: "",
  778. AncestorProductNames: "",
  779. AncestorGradeIds: "",
  780. AncestorGradeNames: "",
  781. Products: "",
  782. ProductIds: "",
  783. })
  784. if err != nil {
  785. return
  786. }
  787. if libResp.Ret != 200 {
  788. fmt.Println("获取开普勒产品库失败", libResp.ErrMsg)
  789. utils.FileLog.Info("获取开普勒产品库失败", libResp.ErrMsg)
  790. return
  791. }
  792. classifyList, err := models.GetAllBaseFromKplerClassify()
  793. if err != nil {
  794. return
  795. }
  796. classifyMap := make(map[int]*models.BaseFromKplerClassify)
  797. for _, v := range classifyList {
  798. classifyMap[v.ProductId] = v
  799. }
  800. // 批量插入分类表中
  801. // family > group(commodity) > product() > grade
  802. sort := 0
  803. for _, v := range libResp.Data {
  804. id, _ := strconv.Atoi(v.Id)
  805. if classifyObj, ok := classifyMap[id]; !ok {
  806. classifyObj = new(models.BaseFromKplerClassify)
  807. classifyObj.ClassifyName = v.Name
  808. classifyObj.ClassifyNameEn = v.Name
  809. classifyObj.ModifyTime = time.Now()
  810. classifyObj.CreateTime = time.Now()
  811. classifyObj.Sort = sort
  812. classifyObj.Level = 1
  813. classifyObj.ParentId = 0
  814. classifyObj.SysUserId = 0
  815. classifyObj.SysUserRealName = ""
  816. classifyObj.ProductId = id
  817. classifyObj.ProductName = v.Name
  818. classifyObj.AncestorFamilyId, _ = strconv.Atoi(v.FamilyId)
  819. classifyObj.AncestorFamilyName = v.Family
  820. classifyObj.AncestorGroupId, _ = strconv.Atoi(v.GroupId)
  821. classifyObj.AncestorGroupName = v.Group
  822. classifyObj.AncestorProductId, _ = strconv.Atoi(v.ProductId)
  823. classifyObj.AncestorProductName = v.Product
  824. classifyObj.AncestorGradeId, _ = strconv.Atoi(v.GradeId)
  825. classifyObj.AncestorGradeName = v.Grade
  826. classifyObj.ClassifyType = v.Type
  827. _, err = classifyObj.Add()
  828. if err != nil {
  829. fmt.Println("新增开普勒产品库失败", err)
  830. utils.FileLog.Info("新增开普勒产品库失败", err)
  831. }
  832. classifyMap[id] = classifyObj
  833. }
  834. }
  835. for _, v := range classifyMap {
  836. if v.ParentId == 0 {
  837. if v.AncestorGradeId > 0 && v.AncestorGradeName != v.ClassifyName {
  838. v.ParentId = classifyMap[v.AncestorGradeId].ClassifyId
  839. } else if v.AncestorProductId > 0 && v.AncestorProductName != v.ClassifyName {
  840. v.ParentId = classifyMap[v.AncestorProductId].ClassifyId
  841. } else if v.AncestorGroupId > 0 && v.AncestorGroupName != v.ClassifyName {
  842. v.ParentId = classifyMap[v.AncestorGroupId].ClassifyId
  843. }
  844. err = v.Update([]string{"ParentId"})
  845. if err != nil {
  846. fmt.Println("更新开普勒产品库失败", err)
  847. utils.FileLog.Info("更新开普勒产品库失败", err)
  848. }
  849. }
  850. }
  851. fmt.Println("classifyList:", classifyList)
  852. return
  853. }
  854. func InitKplerProductGrade ()(err error) {
  855. libUrl := "http://127.0.0.1:8915"
  856. classifyList, err := models.GetAllBaseFromKplerClassifyByClassifyType("subgrade4")
  857. if err != nil {
  858. return
  859. }
  860. for _, c := range classifyList {
  861. c.ModifyTime = time.Now()
  862. err = c.Update([]string{"ModifyTime"})
  863. if err != nil {
  864. fmt.Println("更新开普勒产品库失败", err)
  865. utils.FileLog.Info("更新开普勒产品库失败", err)
  866. }
  867. // 每个分类都发起分组请求
  868. libResp, er := getKplerProductLib(libUrl, &models.KplerProductLibReq{
  869. // AncestorFamilyIds: familyIds,
  870. // AncestorFamilyNames: familyNames,
  871. // AncestorGroupIds: groupIds,
  872. // AncestorGroupNames: groupNames,
  873. // AncestorProductIds: productIds,
  874. // AncestorProductNames: productNames,
  875. AncestorGradeIds: strconv.Itoa(c.ProductId),
  876. AncestorGradeNames: c.ProductName,
  877. })
  878. if er != nil {
  879. fmt.Println("获取开普勒产品库失败", er)
  880. utils.FileLog.Info("获取开普勒产品库失败", er)
  881. continue
  882. }
  883. if libResp.Ret != 200 {
  884. fmt.Println("获取开普勒产品库失败", libResp.ErrMsg)
  885. utils.FileLog.Info("获取开普勒产品库失败", libResp.ErrMsg)
  886. continue
  887. }
  888. classifyObj := new(models.BaseFromKplerClassify)
  889. for _, v := range libResp.Data {
  890. id, _ := strconv.Atoi(v.Id)
  891. if id == 0 {
  892. continue
  893. }
  894. if id == c.ProductId {
  895. continue
  896. }
  897. // 更新对应的分类等级和父级
  898. // 查找所有子分类
  899. classifyItem, er := classifyObj.GetByProductId(id)
  900. if er != nil {
  901. err = fmt.Errorf("子分类不存在 Err:%s", er)
  902. continue
  903. }
  904. classifyItem.Level = c.Level + 1
  905. classifyItem.ParentId = c.ClassifyId
  906. classifyItem.ClassifyType = "subgrade5"
  907. classifyItem.ModifyTime = time.Now()
  908. err = classifyItem.Update([]string{"Level", "ParentId", "ClassifyType", "ModifyTime"})
  909. if err != nil {
  910. fmt.Println("更新开普勒产品库失败", err)
  911. utils.FileLog.Info("更新开普勒产品库失败", err)
  912. }
  913. }
  914. }
  915. fmt.Println("classifyList:", classifyList)
  916. return
  917. }
  918. func InitKplerZone() (err error) {
  919. libUrl := "http://127.0.0.1:8915"
  920. libResp, err := getKplerZoneLib(libUrl)
  921. if err != nil {
  922. return
  923. }
  924. if libResp.Ret != 200 {
  925. fmt.Println("获取开普勒区域库失败", libResp.ErrMsg)
  926. utils.FileLog.Info("获取开普勒区域库失败", libResp.ErrMsg)
  927. return
  928. }
  929. // 批量插入区域表中
  930. zoneMap := make(map[int]*models.BaseFromKplerZone)
  931. for _, v := range libResp.Data {
  932. zoneObj := new(models.BaseFromKplerZone)
  933. descendantId, _ := strconv.Atoi(v.DescendantId)
  934. ancestorId, _ := strconv.Atoi(v.AncestorId)
  935. if _, ok := zoneMap[descendantId]; !ok {
  936. zoneObj.ZoneName = v.DescendantName
  937. //zoneObj.ZoneType = v.DescendantType
  938. zoneObj.AncestorId = ancestorId
  939. zoneObj.AncestorType = v.AncestorType
  940. zoneObj.AncestorName = v.AncestorName
  941. zoneObj.DescendantId = descendantId
  942. zoneObj.DescendantName = v.DescendantName
  943. zoneObj.ModifyTime = time.Now()
  944. zoneObj.CreateTime = time.Now()
  945. _, err = zoneObj.Add()
  946. if err != nil {
  947. fmt.Println("新增开普勒区域库失败", err)
  948. utils.FileLog.Info("新增开普勒区域库失败", err)
  949. }
  950. zoneMap[descendantId] = zoneObj
  951. if _, ok := zoneMap[ancestorId]; !ok {
  952. zoneObj := new(models.BaseFromKplerZone)
  953. zoneObj.ZoneName = v.AncestorName
  954. zoneObj.ZoneType = v.AncestorType
  955. zoneObj.DescendantId = ancestorId
  956. zoneObj.DescendantName = v.AncestorName
  957. zoneObj.ModifyTime = time.Now()
  958. zoneObj.CreateTime = time.Now()
  959. _, err = zoneObj.Add()
  960. if err != nil {
  961. fmt.Println("新增开普勒区域库失败", err)
  962. utils.FileLog.Info("新增开普勒区域库失败", err)
  963. }
  964. zoneMap[ancestorId] = zoneObj
  965. }
  966. }
  967. }
  968. return
  969. }