base_edb_lib.go 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582
  1. package data
  2. import (
  3. "encoding/json"
  4. "eta_gn/eta_api/models"
  5. "eta_gn/eta_api/models/data_manage"
  6. "eta_gn/eta_api/models/data_manage/response"
  7. "eta_gn/eta_api/utils"
  8. "fmt"
  9. "io/ioutil"
  10. "net/http"
  11. "strings"
  12. )
  13. // AddEdbData 新增指标数据
  14. func AddEdbData(source int, edbCode, frequency string) (resp *models.BaseResponse, err error) {
  15. param := make(map[string]interface{})
  16. param["EdbCode"] = edbCode
  17. param["Source"] = source
  18. param["Frequency"] = frequency
  19. urlStr := ``
  20. switch source {
  21. //case utils.DATA_SOURCE_THS:
  22. // urlStr = "ths/add"
  23. //case utils.DATA_SOURCE_WIND:
  24. // urlStr = "wind/add"
  25. //case utils.DATA_SOURCE_PB:
  26. // urlStr = "pb/add"
  27. //case utils.DATA_SOURCE_PB_FINANCE:
  28. // urlStr = "pb_finance/add"
  29. //case utils.DATA_SOURCE_MANUAL:
  30. // urlStr = "manual/add"
  31. //case utils.DATA_SOURCE_LZ:
  32. // urlStr = "lz/add"
  33. //case utils.DATA_SOURCE_YS:
  34. // urlStr = "smm/add"
  35. //case utils.DATA_SOURCE_GL:
  36. // urlStr = "mysteel/add"
  37. //case utils.DATA_SOURCE_ZZ:
  38. // urlStr = "zz/add"
  39. //case utils.DATA_SOURCE_DL:
  40. // urlStr = "dl/add"
  41. //case utils.DATA_SOURCE_SH:
  42. // urlStr = "sh/add"
  43. //case utils.DATA_SOURCE_CFFEX:
  44. // urlStr = "cffex/add"
  45. //case utils.DATA_SOURCE_SHFE:
  46. // urlStr = "shfe/add"
  47. //case utils.DATA_SOURCE_GIE:
  48. // urlStr = "gie/add"
  49. //case utils.DATA_SOURCE_LT:
  50. // urlStr = "lt/add"
  51. //case utils.DATA_SOURCE_COAL:
  52. // urlStr = "coal/add"
  53. //case utils.DATA_SOURCE_GOOGLE_TRAVEL:
  54. // urlStr = "google_travel/add"
  55. //case utils.DATA_SOURCE_MYSTEEL_CHEMICAL:
  56. // urlStr = "mysteel_chemical/add"
  57. //case utils.DATA_SOURCE_EIA_STEO:
  58. // urlStr = "eia_steo/add"
  59. //case utils.DATA_SOURCE_COM_TRADE:
  60. // urlStr = "com_trade/add"
  61. //case utils.DATA_SOURCE_SCI:
  62. // urlStr = "sci/add"
  63. //case utils.DATA_SOURCE_BAIINFO:
  64. // urlStr = "baiinfo/add"
  65. //case utils.DATA_SOURCE_NATIONAL_STATISTICS:
  66. // urlStr = "national_statistics/add"
  67. //case utils.DATA_SOURCE_FUBAO:
  68. // urlStr = "fubao/add"
  69. //case utils.DATA_SOURCE_GFEX:
  70. // urlStr = "gz/add"
  71. //case utils.DATA_SOURCE_ICPI:
  72. // urlStr = "icpi/add"
  73. //case utils.DATA_SOURCE_SCI99:
  74. // urlStr = "sci99/add"
  75. //case utils.DATA_SOURCE_SCI_HQ:
  76. // urlStr = "sci_hq/add"
  77. default:
  78. edbSource := data_manage.GetEdbSourceBySourceId(source)
  79. if edbSource != nil {
  80. urlStr = edbSource.EdbAddMethod
  81. }
  82. }
  83. if urlStr == "" {
  84. err = fmt.Errorf("未实现该指标的刷新接口,请联系管理员")
  85. return
  86. }
  87. resp, err = postRefreshEdbData(param, urlStr)
  88. return
  89. }
  90. // AddEdbData 新增指标数据
  91. func AddEdbDataWindWsd(source int, stockCode, edbCode string) (resp *models.BaseResponse, err error) {
  92. param := make(map[string]interface{})
  93. param["EdbCode"] = edbCode
  94. param["StockCode"] = stockCode
  95. param["Source"] = source
  96. urlStr := `wind/wsd/add`
  97. if urlStr == "" {
  98. err = fmt.Errorf("未实现该指标的刷新接口,请联系管理员")
  99. return
  100. }
  101. resp, err = postRefreshEdbData(param, urlStr)
  102. return
  103. }
  104. // AddEdbDataThsDs 新增指标数据
  105. func AddEdbDataThsDs(source int, stockCode, edbCode, extraPars string) (resp *models.BaseResponse, err error) {
  106. param := make(map[string]interface{})
  107. param["EdbCode"] = edbCode
  108. param["StockCode"] = stockCode
  109. param["Source"] = source
  110. param["ExtraPars"] = extraPars
  111. urlStr := `ths/ds/add`
  112. resp, err = postRefreshEdbData(param, urlStr)
  113. return
  114. }
  115. type AddPredictEdbDataResponse struct {
  116. Ret int
  117. Msg string
  118. ErrMsg string
  119. ErrCode string
  120. Data data_manage.AddEdbInfoResp
  121. Success bool `description:"true 执行成功,false 执行失败"`
  122. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  123. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  124. }
  125. // EdbCalculateSaveMultiResponse 批量计算返回值
  126. type EdbCalculateSaveMultiResponse struct {
  127. Ret int
  128. Msg string
  129. ErrMsg string
  130. ErrCode string
  131. Data data_manage.BatchEdbInfoCalculateBatchSaveResp
  132. Success bool `description:"true 执行成功,false 执行失败"`
  133. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  134. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  135. }
  136. // SaveBasePredictEdbData 新增/编辑预测指标运算
  137. func SaveBasePredictEdbData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  138. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "predict/save", lang)
  139. err = json.Unmarshal(resultByte, &resp)
  140. if err != nil {
  141. return
  142. }
  143. return
  144. }
  145. type PredictRuleCalculateByNineRespResponse struct {
  146. Ret int
  147. Msg string
  148. ErrMsg string
  149. ErrCode string
  150. Data response.PredictRuleCalculateByNineResp
  151. Success bool `description:"true 执行成功,false 执行失败"`
  152. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  153. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  154. }
  155. // PredictCalculateByNinePreview 预测指标动态环差演算
  156. func PredictCalculateByNinePreview(RuleConfigReqStr, lang string) (resp *PredictRuleCalculateByNineRespResponse, err error) {
  157. _, resultByte, err := postAddEdbData(RuleConfigReqStr, "predict/calculate_by_nine/preview", lang)
  158. err = json.Unmarshal(resultByte, &resp)
  159. if err != nil {
  160. return
  161. }
  162. return
  163. }
  164. // SavePredictEdbData 新增/编辑预测指标运算
  165. func SavePredictEdbData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  166. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "predict_calculate/save", lang)
  167. err = json.Unmarshal(resultByte, &resp)
  168. if err != nil {
  169. return
  170. }
  171. return
  172. }
  173. // BatchSavePredictEdbData 新增/编辑 预测指标运算(同比、同差)
  174. func BatchSavePredictEdbData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  175. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "predict_calculate/batch/save", lang)
  176. if err != nil {
  177. return
  178. }
  179. err = json.Unmarshal(resultByte, &resp)
  180. if err != nil {
  181. return
  182. }
  183. return
  184. }
  185. // RefreshEdbData 刷新指标数据
  186. func RefreshEdbData(edbInfoId, source, subSource int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  187. param := make(map[string]interface{})
  188. param["EdbCode"] = edbCode
  189. param["EdbInfoId"] = edbInfoId
  190. param["StartDate"] = startDate
  191. param["Source"] = source
  192. urlStr := ``
  193. //switch source {
  194. //case utils.DATA_SOURCE_THS:
  195. // switch subSource {
  196. // case utils.DATA_SUB_SOURCE_DATE:
  197. // urlStr = "ths/ds/refresh"
  198. // case utils.DATA_SUB_SOURCE_HIGH_FREQUENCY:
  199. // urlStr = "ths/hf/edb/refresh"
  200. // default:
  201. // urlStr = "ths/refresh"
  202. // }
  203. //case utils.DATA_SOURCE_WIND:
  204. // if subSource == 0 {
  205. // urlStr = "wind/refresh"
  206. // } else {
  207. // urlStr = "wind/wsd/refresh"
  208. // }
  209. //case utils.DATA_SOURCE_PB:
  210. // urlStr = "pb/refresh"
  211. //case utils.DATA_SOURCE_PB_FINANCE:
  212. // urlStr = "pb_finance/refresh"
  213. //case utils.DATA_SOURCE_MANUAL:
  214. // urlStr = "manual/refresh"
  215. //case utils.DATA_SOURCE_LZ:
  216. // urlStr = "lz/refresh"
  217. //case utils.DATA_SOURCE_YS:
  218. // urlStr = "smm/refresh"
  219. //case utils.DATA_SOURCE_GL:
  220. // urlStr = "mysteel/refresh"
  221. //case utils.DATA_SOURCE_ZZ:
  222. // urlStr = "zz/refresh"
  223. //case utils.DATA_SOURCE_DL:
  224. // urlStr = "dl/refresh"
  225. //case utils.DATA_SOURCE_SH:
  226. // urlStr = "sh/refresh"
  227. //case utils.DATA_SOURCE_CFFEX:
  228. // urlStr = "cffex/refresh"
  229. //case utils.DATA_SOURCE_SHFE:
  230. // urlStr = "shfe/refresh"
  231. //case utils.DATA_SOURCE_GIE:
  232. // urlStr = "gie/refresh"
  233. //case utils.DATA_SOURCE_LT:
  234. // urlStr = "lt/refresh"
  235. //case utils.DATA_SOURCE_COAL:
  236. // urlStr = "coal/refresh"
  237. //case utils.DATA_SOURCE_GOOGLE_TRAVEL:
  238. // urlStr = "google_travel/refresh"
  239. //case utils.DATA_SOURCE_MYSTEEL_CHEMICAL:
  240. // urlStr = "mysteel_chemical/refresh"
  241. //case utils.DATA_SOURCE_EIA_STEO:
  242. // urlStr = "eia_steo/refresh"
  243. //case utils.DATA_SOURCE_PREDICT:
  244. // urlStr = "predict/refresh"
  245. //case utils.DATA_SOURCE_COM_TRADE:
  246. // urlStr = "com_trade/refresh"
  247. //case utils.DATA_SOURCE_SCI:
  248. // urlStr = "sci/refresh"
  249. //case utils.DATA_SOURCE_BAIINFO:
  250. // urlStr = "baiinfo/refresh"
  251. //case utils.DATA_SOURCE_STOCK_PLANT:
  252. // urlStr = "stock_plant/refresh"
  253. //case utils.DATA_SOURCE_NATIONAL_STATISTICS:
  254. // urlStr = "national_statistics/refresh"
  255. //case utils.DATA_SOURCE_FUBAO:
  256. // urlStr = "fubao/refresh"
  257. //case utils.DATA_SOURCE_GFEX:
  258. // urlStr = "gz/refresh"
  259. //case utils.DATA_SOURCE_ICPI:
  260. // urlStr = "icpi/refresh"
  261. //case utils.DATA_SOURCE_SCI99:
  262. // urlStr = "sci99/refresh"
  263. //case utils.DATA_SOURCE_SCI_HQ:
  264. // urlStr = "sci_hq/refresh"
  265. //default:
  266. // edbSource := data_manage.EdbSourceIdMap[source]
  267. // if edbSource != nil {
  268. // urlStr = edbSource.EdbRefreshMethod
  269. // }
  270. //}
  271. edbSource := data_manage.GetEdbSourceBySourceId(source)
  272. if edbSource != nil {
  273. urlStr = edbSource.EdbRefreshMethod
  274. }
  275. if urlStr == "" {
  276. err = fmt.Errorf(fmt.Sprint("source:", source, ";未实现该指标的刷新接口,请联系管理员"))
  277. return
  278. }
  279. resp, err = postRefreshEdbData(param, urlStr)
  280. return
  281. }
  282. // RefreshEdbCalculateData 刷新普通计算指标数据请求
  283. func RefreshEdbCalculateData(edbInfoId int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  284. param := make(map[string]interface{})
  285. param["EdbCode"] = edbCode
  286. param["EdbInfoId"] = edbInfoId
  287. param["StartDate"] = startDate
  288. resp, err = postRefreshEdbData(param, "calculate/refresh")
  289. return
  290. }
  291. // RefreshPredictEdbCalculateData 刷新 预测计算指标 数据请求
  292. func RefreshPredictEdbCalculateData(edbInfoId int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  293. param := make(map[string]interface{})
  294. param["EdbCode"] = edbCode
  295. param["EdbInfoId"] = edbInfoId
  296. param["StartDate"] = startDate
  297. resp, err = postRefreshEdbData(param, "predict_calculate/refresh")
  298. return
  299. }
  300. // AddEdbCalculateData 新增 累计值转月-同比值-同差等计算新增
  301. func AddEdbCalculateData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  302. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/add", lang)
  303. err = json.Unmarshal(resultByte, &resp)
  304. if err != nil {
  305. return
  306. }
  307. return
  308. }
  309. // EditEdbCalculateData 编辑 累计值转月-同比值-同差等计算新增
  310. func EditEdbCalculateData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  311. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/edit", lang)
  312. err = json.Unmarshal(resultByte, &resp)
  313. if err != nil {
  314. return
  315. }
  316. return
  317. }
  318. // BatchSaveEdbCalculateData 新增 累计值转月-同比值-同差等计算新增
  319. func BatchSaveEdbCalculateData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  320. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/batch/save", lang)
  321. err = json.Unmarshal(resultByte, &resp)
  322. if err != nil {
  323. return
  324. }
  325. return
  326. }
  327. // BatchEditEdbCalculateData 编辑 累计值转月-同比值-同差等计算新增
  328. func BatchEditEdbCalculateData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  329. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/batch/edit", lang)
  330. err = json.Unmarshal(resultByte, &resp)
  331. if err != nil {
  332. return
  333. }
  334. return
  335. }
  336. // BatchSaveEdbCalculateMultiData 批量新增 累计值转月-同比值-同差等计算新增
  337. func BatchSaveEdbCalculateMultiData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *EdbCalculateSaveMultiResponse, err error) {
  338. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/batch/save/multi", lang)
  339. err = json.Unmarshal(resultByte, &resp)
  340. if err != nil {
  341. return
  342. }
  343. return
  344. }
  345. // BatchEditEdbCalculateMultiData 批量编辑 累计值转月-同比值-同差等计算新增
  346. func BatchEditEdbCalculateMultiData(edbInfoCalculateBatchSaveReqStr, lang string) (resp *EdbCalculateSaveMultiResponse, err error) {
  347. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/batch/edit/multi", lang)
  348. err = json.Unmarshal(resultByte, &resp)
  349. if err != nil {
  350. return
  351. }
  352. return
  353. }
  354. //// EditEdbCalculateData 修改计算指标数据请求
  355. //func EditEdbCalculateData(edbInfoCalculateBatchSaveReqStr string) (resp *models.BaseResponse, err error) {
  356. // resp, _, err = postAddEdbData(edbInfoCalculateBatchSaveReqStr, "batch/save")
  357. // return
  358. //}
  359. // ExecPythonCode 执行python代码
  360. func ExecPythonCode(pythonCode string) (resp *models.BaseResponse, err error) {
  361. param := make(map[string]interface{})
  362. param["PythonCode"] = pythonCode
  363. urlStr := `/python/exec`
  364. resp, err = postRefreshEdbData(param, urlStr)
  365. return
  366. }
  367. // AddPythonEdbData 新增python指标
  368. func AddPythonEdbData(param, lang string) (resp *models.BaseResponse, err error) {
  369. urlStr := ``
  370. urlStr = "python/add"
  371. resp, _, err = postAddEdbData(param, urlStr, lang)
  372. return
  373. }
  374. // EditPythonEdbData 编辑python指标
  375. func EditPythonEdbData(param, lang string) (resp *models.BaseResponse, err error) {
  376. urlStr := ``
  377. urlStr = "python/edit"
  378. resp, _, err = postAddEdbData(param, urlStr, lang)
  379. return
  380. }
  381. // SaveAdjustEdbInfo 保存数据调整指标
  382. func SaveAdjustEdbInfo(param, lang string) (resp *models.BaseResponse, err error) {
  383. urlStr := ``
  384. urlStr = "calculate/adjust/save"
  385. resp, _, err = postAddEdbData(param, urlStr, lang)
  386. return
  387. }
  388. // ResetCustomAnalysisData 重置自定义表格的数据
  389. func ResetCustomAnalysisData(reqStr, lang string) (resp *AddPredictEdbDataResponse, err error) {
  390. _, resultByte, err := postAddEdbData(reqStr, "calculate/custom_analysis/reset", lang)
  391. err = json.Unmarshal(resultByte, &resp)
  392. if err != nil {
  393. return
  394. }
  395. return
  396. }
  397. // CalculateComputeCorrelationResp 拟合残差计算相关性的值返回
  398. type CalculateComputeCorrelationResp struct {
  399. Ret int
  400. Msg string
  401. ErrMsg string
  402. ErrCode string
  403. Data string
  404. Success bool `description:"true 执行成功,false 执行失败"`
  405. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  406. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  407. }
  408. // CalculateComputeCorrelation 拟合残差计算相关性的值
  409. func CalculateComputeCorrelation(edbInfoCalculateBatchSaveReqStr, lang string) (resp *CalculateComputeCorrelationResp, err error) {
  410. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/compute_correlation", lang)
  411. err = json.Unmarshal(resultByte, &resp)
  412. if err != nil {
  413. return
  414. }
  415. return
  416. }
  417. // PredictCalculateComputeCorrelation 拟合残差计算相关性的值(预测指标)
  418. func PredictCalculateComputeCorrelation(edbInfoCalculateBatchSaveReqStr, lang string) (resp *CalculateComputeCorrelationResp, err error) {
  419. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "predict_calculate/compute_correlation", lang)
  420. err = json.Unmarshal(resultByte, &resp)
  421. if err != nil {
  422. return
  423. }
  424. return
  425. }
  426. // BaseCalculateResp 拟合残差计算相关性的值返回
  427. type BaseCalculateResp struct {
  428. Ret int
  429. Msg string
  430. ErrMsg string
  431. ErrCode string
  432. Data BaseCalculateDataResp
  433. Success bool `description:"true 执行成功,false 执行失败"`
  434. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  435. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  436. }
  437. // BaseCalculateDataResp
  438. // @Description: 基础计算的返回结果
  439. type BaseCalculateDataResp struct {
  440. DataMap map[string]float64
  441. DateList []string
  442. }
  443. // BaseCalculate 基础计算
  444. func BaseCalculate(param, lang string) (resp *BaseCalculateResp, err error) {
  445. urlStr := "calculate/base"
  446. _, resultByte, err := postAddEdbData(param, urlStr, lang)
  447. err = json.Unmarshal(resultByte, &resp)
  448. if err != nil {
  449. return
  450. }
  451. return
  452. }
  453. // AddBaseEdbInfo 新增基础指标
  454. func AddBaseEdbInfo(addBaseEdbInfoReqStr string, source, subSource int, lang string) (resp *AddPredictEdbDataResponse, err error) {
  455. urlStr := ``
  456. switch source {
  457. case utils.DATA_SOURCE_BUSINESS:
  458. urlStr = "business_index/add"
  459. //case utils.DATA_SOURCE_THS:
  460. // if subSource == utils.DATA_SUB_SOURCE_HIGH_FREQUENCY {
  461. // urlStr = "ths/hf/edb/add"
  462. // }
  463. default:
  464. edbSource := data_manage.GetEdbSourceBySourceId(source)
  465. if edbSource != nil {
  466. urlStr = edbSource.EdbRefreshMethod
  467. }
  468. }
  469. if urlStr == "" {
  470. err = fmt.Errorf(fmt.Sprint("source:", source, ";未实现该指标的刷新接口,请联系管理员"))
  471. return
  472. }
  473. _, resultByte, err := postAddEdbData(addBaseEdbInfoReqStr, urlStr, lang)
  474. if err != nil {
  475. return
  476. }
  477. err = json.Unmarshal(resultByte, &resp)
  478. return
  479. }
  480. // postRefreshEdbData 刷新指标数据
  481. func postRefreshEdbData(param map[string]interface{}, urlStr string) (resp *models.BaseResponse, err error) {
  482. postUrl := utils.EDB_LIB_URL + urlStr
  483. postData, err := json.Marshal(param)
  484. if err != nil {
  485. return
  486. }
  487. result, err := HttpPost(postUrl, string(postData), utils.ZhLangVersion, "application/json")
  488. if err != nil {
  489. return
  490. }
  491. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";" + string(postData) + ";result:" + string(result))
  492. err = json.Unmarshal(result, &resp)
  493. if err != nil {
  494. return
  495. }
  496. return resp, nil
  497. }
  498. // postAddEdbData 新增指标数据
  499. func postAddEdbData(paramStr string, urlStr, lang string) (resp *models.BaseResponse, result []byte, err error) {
  500. postUrl := utils.EDB_LIB_URL + urlStr
  501. result, err = HttpPost(postUrl, paramStr, lang, "application/json")
  502. if err != nil {
  503. return
  504. }
  505. err = json.Unmarshal(result, &resp)
  506. if err != nil {
  507. return
  508. }
  509. return
  510. }
  511. func HttpPost(url, postData, lang string, params ...string) ([]byte, error) {
  512. body := ioutil.NopCloser(strings.NewReader(postData))
  513. client := &http.Client{}
  514. req, err := http.NewRequest("POST", url, body)
  515. if err != nil {
  516. return nil, err
  517. }
  518. contentType := "application/x-www-form-urlencoded;charset=utf-8"
  519. if len(params) > 0 && params[0] != "" {
  520. contentType = params[0]
  521. }
  522. req.Header.Set("Content-Type", contentType)
  523. req.Header.Set("Lang", lang)
  524. req.Header.Set("authorization", utils.MD5(utils.APP_EDB_LIB_NAME_EN+utils.EDB_LIB_Md5_KEY))
  525. resp, err := client.Do(req)
  526. if err != nil {
  527. return nil, err
  528. }
  529. defer resp.Body.Close()
  530. b, err := ioutil.ReadAll(resp.Body)
  531. utils.FileLog.Debug("HttpPost:" + string(b))
  532. return b, err
  533. }
  534. // BaseStepCalculate 基础运算-多步骤
  535. func BaseStepCalculate(param, lang string) (resp *BaseCalculateResp, err error) {
  536. urlStr := "calculate/base/step_calculate"
  537. _, resultByte, err := postAddEdbData(param, urlStr, lang)
  538. err = json.Unmarshal(resultByte, &resp)
  539. if err != nil {
  540. return
  541. }
  542. return
  543. }