base_edb_lib.go 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292
  1. package data
  2. import (
  3. "encoding/json"
  4. "eta/eta_mobile/models"
  5. "eta/eta_mobile/models/data_manage"
  6. "eta/eta_mobile/utils"
  7. "fmt"
  8. "io/ioutil"
  9. "net/http"
  10. "strings"
  11. )
  12. // AddEdbData 新增指标数据
  13. func AddEdbData(source int, edbCode string) (resp *models.BaseResponse, err error) {
  14. param := make(map[string]interface{})
  15. param["EdbCode"] = edbCode
  16. urlStr := ``
  17. switch source {
  18. case utils.DATA_SOURCE_THS:
  19. urlStr = "ths/add"
  20. case utils.DATA_SOURCE_WIND:
  21. urlStr = "wind/add"
  22. case utils.DATA_SOURCE_PB:
  23. urlStr = "pb/add"
  24. case utils.DATA_SOURCE_PB_FINANCE:
  25. urlStr = "pb_finance/add"
  26. case utils.DATA_SOURCE_MANUAL:
  27. urlStr = "manual/add"
  28. case utils.DATA_SOURCE_LZ:
  29. urlStr = "lz/add"
  30. case utils.DATA_SOURCE_YS:
  31. urlStr = "smm/add"
  32. case utils.DATA_SOURCE_GL:
  33. urlStr = "mysteel/add"
  34. case utils.DATA_SOURCE_ZZ:
  35. urlStr = "zz/add"
  36. case utils.DATA_SOURCE_DL:
  37. urlStr = "dl/add"
  38. case utils.DATA_SOURCE_SH:
  39. urlStr = "sh/add"
  40. case utils.DATA_SOURCE_CFFEX:
  41. urlStr = "cffex/add"
  42. case utils.DATA_SOURCE_SHFE:
  43. urlStr = "shfe/add"
  44. case utils.DATA_SOURCE_GIE:
  45. urlStr = "gie/add"
  46. case utils.DATA_SOURCE_LT:
  47. urlStr = "lt/add"
  48. case utils.DATA_SOURCE_COAL:
  49. urlStr = "coal/add"
  50. case utils.DATA_SOURCE_GOOGLE_TRAVEL:
  51. urlStr = "google_travel/add"
  52. case utils.DATA_SOURCE_MYSTEEL_CHEMICAL:
  53. urlStr = "mysteel_chemical/add"
  54. case utils.DATA_SOURCE_EIA_STEO:
  55. urlStr = "eia_steo/add"
  56. case utils.DATA_SOURCE_COM_TRADE:
  57. urlStr = "com_trade/add"
  58. case utils.DATA_SOURCE_SCI:
  59. urlStr = "sci/add"
  60. case utils.DATA_SOURCE_BAIINFO:
  61. urlStr = "baiinfo/add"
  62. case utils.DATA_SOURCE_NATIONAL_STATISTICS:
  63. urlStr = "national_statistics/add"
  64. case utils.DATA_SOURCE_FUBAO:
  65. urlStr = "fubao/add"
  66. default:
  67. }
  68. if urlStr == "" {
  69. err = fmt.Errorf("未实现该指标的刷新接口,请联系管理员")
  70. return
  71. }
  72. resp, err = postRefreshEdbData(param, urlStr)
  73. return
  74. }
  75. type AddPredictEdbDataResponse struct {
  76. Ret int
  77. Msg string
  78. ErrMsg string
  79. ErrCode string
  80. Data data_manage.AddEdbInfoResp
  81. Success bool `description:"true 执行成功,false 执行失败"`
  82. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  83. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  84. }
  85. // RefreshEdbData 刷新指标数据
  86. func RefreshEdbData(edbInfoId, source int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  87. param := make(map[string]interface{})
  88. param["EdbCode"] = edbCode
  89. param["EdbInfoId"] = edbInfoId
  90. param["StartDate"] = startDate
  91. urlStr := ``
  92. switch source {
  93. case utils.DATA_SOURCE_THS:
  94. urlStr = "ths/refresh"
  95. case utils.DATA_SOURCE_WIND:
  96. urlStr = "wind/refresh"
  97. case utils.DATA_SOURCE_PB:
  98. urlStr = "pb/refresh"
  99. case utils.DATA_SOURCE_PB_FINANCE:
  100. urlStr = "pb_finance/refresh"
  101. case utils.DATA_SOURCE_MANUAL:
  102. urlStr = "manual/refresh"
  103. case utils.DATA_SOURCE_LZ:
  104. urlStr = "lz/refresh"
  105. case utils.DATA_SOURCE_YS:
  106. urlStr = "smm/refresh"
  107. case utils.DATA_SOURCE_GL:
  108. urlStr = "mysteel/refresh"
  109. case utils.DATA_SOURCE_ZZ:
  110. urlStr = "zz/refresh"
  111. case utils.DATA_SOURCE_DL:
  112. urlStr = "dl/refresh"
  113. case utils.DATA_SOURCE_SH:
  114. urlStr = "sh/refresh"
  115. case utils.DATA_SOURCE_CFFEX:
  116. urlStr = "cffex/refresh"
  117. case utils.DATA_SOURCE_SHFE:
  118. urlStr = "shfe/refresh"
  119. case utils.DATA_SOURCE_GIE:
  120. urlStr = "gie/refresh"
  121. case utils.DATA_SOURCE_LT:
  122. urlStr = "lt/refresh"
  123. case utils.DATA_SOURCE_COAL:
  124. urlStr = "coal/refresh"
  125. case utils.DATA_SOURCE_GOOGLE_TRAVEL:
  126. urlStr = "google_travel/refresh"
  127. case utils.DATA_SOURCE_MYSTEEL_CHEMICAL:
  128. urlStr = "mysteel_chemical/refresh"
  129. case utils.DATA_SOURCE_EIA_STEO:
  130. urlStr = "eia_steo/refresh"
  131. case utils.DATA_SOURCE_PREDICT:
  132. urlStr = "predict/refresh"
  133. case utils.DATA_SOURCE_COM_TRADE:
  134. urlStr = "com_trade/refresh"
  135. case utils.DATA_SOURCE_SCI:
  136. urlStr = "sci/refresh"
  137. case utils.DATA_SOURCE_BAIINFO:
  138. urlStr = "baiinfo/refresh"
  139. case utils.DATA_SOURCE_STOCK_PLANT:
  140. urlStr = "stock_plant/refresh"
  141. case utils.DATA_SOURCE_NATIONAL_STATISTICS:
  142. urlStr = "national_statistics/refresh"
  143. case utils.DATA_SOURCE_FUBAO:
  144. urlStr = "fubao/refresh"
  145. }
  146. if urlStr == "" {
  147. err = fmt.Errorf(fmt.Sprint("source:", source, ";未实现该指标的刷新接口,请联系管理员"))
  148. return
  149. }
  150. resp, err = postRefreshEdbData(param, urlStr)
  151. return
  152. }
  153. // RefreshEdbCalculateData 刷新普通计算指标数据请求
  154. func RefreshEdbCalculateData(edbInfoId int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  155. param := make(map[string]interface{})
  156. param["EdbCode"] = edbCode
  157. param["EdbInfoId"] = edbInfoId
  158. param["StartDate"] = startDate
  159. resp, err = postRefreshEdbData(param, "calculate/refresh")
  160. return
  161. }
  162. // RefreshPredictEdbCalculateData 刷新 预测计算指标 数据请求
  163. func RefreshPredictEdbCalculateData(edbInfoId int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  164. param := make(map[string]interface{})
  165. param["EdbCode"] = edbCode
  166. param["EdbInfoId"] = edbInfoId
  167. param["StartDate"] = startDate
  168. resp, err = postRefreshEdbData(param, "predict_calculate/refresh")
  169. return
  170. }
  171. // AddEdbCalculateData 新增 累计值转月-同比值-同差等计算新增
  172. func AddEdbCalculateData(edbInfoCalculateBatchSaveReqStr string) (resp *AddPredictEdbDataResponse, err error) {
  173. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/add")
  174. err = json.Unmarshal(resultByte, &resp)
  175. if err != nil {
  176. return
  177. }
  178. return
  179. }
  180. // EditEdbCalculateData 编辑 累计值转月-同比值-同差等计算新增
  181. func EditEdbCalculateData(edbInfoCalculateBatchSaveReqStr string) (resp *AddPredictEdbDataResponse, err error) {
  182. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/edit")
  183. err = json.Unmarshal(resultByte, &resp)
  184. if err != nil {
  185. return
  186. }
  187. return
  188. }
  189. // BatchSaveEdbCalculateData 新增 累计值转月-同比值-同差等计算新增
  190. func BatchSaveEdbCalculateData(edbInfoCalculateBatchSaveReqStr string) (resp *AddPredictEdbDataResponse, err error) {
  191. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/batch/save")
  192. err = json.Unmarshal(resultByte, &resp)
  193. if err != nil {
  194. return
  195. }
  196. return
  197. }
  198. // BatchEditEdbCalculateData 编辑 累计值转月-同比值-同差等计算新增
  199. func BatchEditEdbCalculateData(edbInfoCalculateBatchSaveReqStr string) (resp *AddPredictEdbDataResponse, err error) {
  200. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/batch/edit")
  201. err = json.Unmarshal(resultByte, &resp)
  202. if err != nil {
  203. return
  204. }
  205. return
  206. }
  207. // CalculateComputeCorrelationResp 拟合残差计算相关性的值返回
  208. type CalculateComputeCorrelationResp struct {
  209. Ret int
  210. Msg string
  211. ErrMsg string
  212. ErrCode string
  213. Data float64
  214. Success bool `description:"true 执行成功,false 执行失败"`
  215. IsSendEmail bool `json:"-" description:"true 发送邮件,false 不发送邮件"`
  216. IsAddLog bool `json:"-" description:"true 新增操作日志,false 不新增操作日志" `
  217. }
  218. // CalculateComputeCorrelation 拟合残差计算相关性的值
  219. func CalculateComputeCorrelation(edbInfoCalculateBatchSaveReqStr string) (resp *CalculateComputeCorrelationResp, err error) {
  220. _, resultByte, err := postAddEdbData(edbInfoCalculateBatchSaveReqStr, "calculate/compute_correlation")
  221. err = json.Unmarshal(resultByte, &resp)
  222. if err != nil {
  223. return
  224. }
  225. return
  226. }
  227. // postRefreshEdbData 刷新指标数据
  228. func postRefreshEdbData(param map[string]interface{}, urlStr string) (resp *models.BaseResponse, err error) {
  229. postUrl := utils.EDB_LIB_URL + urlStr
  230. postData, err := json.Marshal(param)
  231. if err != nil {
  232. return
  233. }
  234. result, err := HttpPost(postUrl, string(postData), "application/json")
  235. if err != nil {
  236. return
  237. }
  238. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";" + string(postData) + ";result:" + string(result))
  239. err = json.Unmarshal(result, &resp)
  240. if err != nil {
  241. return
  242. }
  243. return resp, nil
  244. }
  245. // postAddEdbData 新增指标数据
  246. func postAddEdbData(paramStr string, urlStr string) (resp *models.BaseResponse, result []byte, err error) {
  247. postUrl := utils.EDB_LIB_URL + urlStr
  248. result, err = HttpPost(postUrl, paramStr, "application/json")
  249. if err != nil {
  250. return
  251. }
  252. err = json.Unmarshal(result, &resp)
  253. if err != nil {
  254. return
  255. }
  256. return
  257. }
  258. func HttpPost(url, postData string, params ...string) ([]byte, error) {
  259. body := ioutil.NopCloser(strings.NewReader(postData))
  260. client := &http.Client{}
  261. req, err := http.NewRequest("POST", url, body)
  262. if err != nil {
  263. return nil, err
  264. }
  265. contentType := "application/x-www-form-urlencoded;charset=utf-8"
  266. if len(params) > 0 && params[0] != "" {
  267. contentType = params[0]
  268. }
  269. req.Header.Set("Content-Type", contentType)
  270. req.Header.Set("authorization", utils.MD5(utils.APP_EDB_LIB_NAME_EN+utils.EDB_LIB_Md5_KEY))
  271. resp, err := client.Do(req)
  272. defer resp.Body.Close()
  273. b, err := ioutil.ReadAll(resp.Body)
  274. fmt.Println("HttpPost:" + string(b))
  275. return b, err
  276. }