base_edb_lib.go 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190
  1. package data
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "hongze/hongze_chart_lib/models"
  6. "hongze/hongze_chart_lib/utils"
  7. "io/ioutil"
  8. "net/http"
  9. "strings"
  10. )
  11. // 新增指标数据
  12. func AddEdbData(source int, edbCode string) (resp *models.BaseResponse, err error) {
  13. param := make(map[string]interface{})
  14. param["EdbCode"] = edbCode
  15. urlStr := ``
  16. switch source {
  17. case utils.DATA_SOURCE_THS:
  18. urlStr = "/ths/add"
  19. case utils.DATA_SOURCE_WIND:
  20. urlStr = "/wind/add"
  21. case utils.DATA_SOURCE_PB:
  22. urlStr = "/pb/add"
  23. case utils.DATA_SOURCE_PB_FINANCE:
  24. urlStr = "/pb_finance/add"
  25. case utils.DATA_SOURCE_MANUAL:
  26. urlStr = "/manual/add"
  27. case utils.DATA_SOURCE_LZ:
  28. urlStr = "/lz/add"
  29. case utils.DATA_SOURCE_YS:
  30. urlStr = "/smm/add"
  31. case utils.DATA_SOURCE_GL:
  32. urlStr = "/mysteel/add"
  33. case utils.DATA_SOURCE_ZZ:
  34. urlStr = "/zz/add"
  35. case utils.DATA_SOURCE_DL:
  36. urlStr = "/dl/add"
  37. case utils.DATA_SOURCE_SH:
  38. urlStr = "/sh/add"
  39. case utils.DATA_SOURCE_CFFEX:
  40. urlStr = "/cffex/add"
  41. case utils.DATA_SOURCE_SHFE:
  42. urlStr = "/shfe/add"
  43. case utils.DATA_SOURCE_GIE:
  44. urlStr = "/gie/add"
  45. case utils.DATA_SOURCE_LT:
  46. urlStr = "/lt/add"
  47. case utils.DATA_SOURCE_COAL:
  48. urlStr = "/coal/add"
  49. case utils.DATA_SOURCE_SCI:
  50. urlStr = "/sci/add"
  51. case utils.DATA_SOURCE_BAIINFO:
  52. urlStr = "/baiinfo/add"
  53. case utils.DATA_SOURCE_NATIONAL_STATISTICS:
  54. urlStr = "/national_statistics/add"
  55. default:
  56. }
  57. if urlStr == "" {
  58. err = fmt.Errorf("未实现该指标的刷新接口,请联系管理员")
  59. return
  60. }
  61. resp, err = postRefreshEdbData(param, urlStr)
  62. return
  63. }
  64. // RefreshEdbData 刷新指标数据
  65. func RefreshEdbData(edbInfoId, source int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  66. param := make(map[string]interface{})
  67. param["EdbCode"] = edbCode
  68. param["EdbInfoId"] = edbInfoId
  69. param["StartDate"] = startDate
  70. urlStr := ``
  71. switch source {
  72. case utils.DATA_SOURCE_THS:
  73. urlStr = "/ths/refresh"
  74. case utils.DATA_SOURCE_WIND:
  75. urlStr = "/wind/refresh"
  76. case utils.DATA_SOURCE_PB:
  77. urlStr = "/pb/refresh"
  78. case utils.DATA_SOURCE_PB_FINANCE:
  79. urlStr = "/pb_finance/refresh"
  80. case utils.DATA_SOURCE_MANUAL:
  81. urlStr = "/manual/refresh"
  82. case utils.DATA_SOURCE_LZ:
  83. urlStr = "/lz/refresh"
  84. case utils.DATA_SOURCE_YS:
  85. urlStr = "/smm/refresh"
  86. case utils.DATA_SOURCE_GL:
  87. urlStr = "/mysteel/refresh"
  88. case utils.DATA_SOURCE_ZZ:
  89. urlStr = "/zz/refresh"
  90. case utils.DATA_SOURCE_DL:
  91. urlStr = "/dl/refresh"
  92. case utils.DATA_SOURCE_SH:
  93. urlStr = "/sh/refresh"
  94. case utils.DATA_SOURCE_CFFEX:
  95. urlStr = "/cffex/refresh"
  96. case utils.DATA_SOURCE_SHFE:
  97. urlStr = "/shfe/refresh"
  98. case utils.DATA_SOURCE_GIE:
  99. urlStr = "/gie/refresh"
  100. case utils.DATA_SOURCE_LT:
  101. urlStr = "/lt/refresh"
  102. case utils.DATA_SOURCE_COAL:
  103. urlStr = "/coal/refresh"
  104. case utils.DATA_SOURCE_GOOGLE_TRAVEL:
  105. urlStr = "google_travel/refresh"
  106. case utils.DATA_SOURCE_MYSTEEL_CHEMICAL:
  107. urlStr = "mysteel_chemical/refresh"
  108. case utils.DATA_SOURCE_EIA_STEO:
  109. urlStr = "eia_steo/refresh"
  110. case utils.DATA_SOURCE_PREDICT:
  111. urlStr = "predict/refresh"
  112. case utils.DATA_SOURCE_COM_TRADE:
  113. urlStr = "com_trade/refresh"
  114. case utils.DATA_SOURCE_SCI:
  115. urlStr = "/sci/refresh"
  116. case utils.DATA_SOURCE_BAIINFO:
  117. urlStr = "/baiinfo/refresh"
  118. case utils.DATA_SOURCE_NATIONAL_STATISTICS:
  119. urlStr = "/national_statistics/refresh"
  120. case utils.DATA_SOURCE_FUBAO:
  121. urlStr = "fubao/refresh"
  122. }
  123. if urlStr == "" {
  124. err = fmt.Errorf(fmt.Sprint("source:", source, ";未实现该指标的刷新接口,请联系管理员"))
  125. return
  126. }
  127. resp, err = postRefreshEdbData(param, urlStr)
  128. return
  129. }
  130. // RefreshEdbCalculateData 刷新计算指标数据请求
  131. func RefreshEdbCalculateData(edbInfoId int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  132. param := make(map[string]interface{})
  133. param["EdbCode"] = edbCode
  134. param["EdbInfoId"] = edbInfoId
  135. param["StartDate"] = startDate
  136. resp, err = postRefreshEdbData(param, "/calculate/refresh")
  137. return
  138. }
  139. // RefreshPredictEdbCalculateData 刷新 预测计算指标 数据请求
  140. func RefreshPredictEdbCalculateData(edbInfoId int, edbCode, startDate string) (resp *models.BaseResponse, err error) {
  141. param := make(map[string]interface{})
  142. param["EdbCode"] = edbCode
  143. param["EdbInfoId"] = edbInfoId
  144. param["StartDate"] = startDate
  145. resp, err = postRefreshEdbData(param, "predict_calculate/refresh")
  146. return
  147. }
  148. // postRefreshEdbData 刷新指标数据
  149. func postRefreshEdbData(param map[string]interface{}, urlStr string) (resp *models.BaseResponse, err error) {
  150. postUrl := utils.EDB_LIB_URL + urlStr
  151. postData, err := json.Marshal(param)
  152. if err != nil {
  153. return
  154. }
  155. result, err := HttpPost(postUrl, string(postData), "application/json")
  156. if err != nil {
  157. return
  158. }
  159. err = json.Unmarshal(result, &resp)
  160. if err != nil {
  161. return
  162. }
  163. return resp, nil
  164. }
  165. func HttpPost(url, postData string, params ...string) ([]byte, error) {
  166. body := ioutil.NopCloser(strings.NewReader(postData))
  167. client := &http.Client{}
  168. req, err := http.NewRequest("POST", url, body)
  169. if err != nil {
  170. return nil, err
  171. }
  172. contentType := "application/x-www-form-urlencoded;charset=utf-8"
  173. if len(params) > 0 && params[0] != "" {
  174. contentType = params[0]
  175. }
  176. req.Header.Set("Content-Type", contentType)
  177. req.Header.Set("authorization", utils.MD5(utils.APP_EDB_LIB_NAME_EN+utils.EDB_LIB_Md5_KEY))
  178. resp, err := client.Do(req)
  179. defer resp.Body.Close()
  180. b, err := ioutil.ReadAll(resp.Body)
  181. fmt.Println("HttpPost:" + string(b))
  182. return b, err
  183. }