base_future_good_lib.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. package future_good
  2. import (
  3. "encoding/json"
  4. "eta/eta_task/models"
  5. "eta/eta_task/utils"
  6. "io/ioutil"
  7. "net/http"
  8. "strings"
  9. )
  10. // AddEdbData 添加指标数据
  11. func AddEdbData(futureGoodEdbCode string) (resp *models.BaseResponse, err error) {
  12. param := make(map[string]interface{})
  13. param["EdbCode"] = futureGoodEdbCode
  14. urlStr := `future_good/add`
  15. resp, err = postRefreshEdbData(param, urlStr)
  16. return
  17. }
  18. // RefreshEdbData 刷新指标数据
  19. func RefreshEdbData(futureGoodEdbInfoId int, futureGoodEdbCode, startDate string) (resp *models.BaseResponse, err error) {
  20. param := make(map[string]interface{})
  21. param["FutureGoodEdbCode"] = futureGoodEdbCode
  22. param["FutureGoodEdbInfoId"] = futureGoodEdbInfoId
  23. param["StartDate"] = startDate
  24. urlStr := `future_good/refresh`
  25. resp, err = postRefreshEdbData(param, urlStr)
  26. return
  27. }
  28. var step int //计数器
  29. // postRefreshEdbData 刷新指标数据
  30. func postRefreshEdbData(param map[string]interface{}, urlStr string) (resp *models.BaseResponse, err error) {
  31. var postUrl string
  32. edbLibUrlListLen := len(utils.EDB_LIB_URL_LIST)
  33. if edbLibUrlListLen <= 0 {
  34. postUrl = utils.EDB_LIB_URL
  35. } else {
  36. rn := step % edbLibUrlListLen
  37. postUrl = utils.EDB_LIB_URL_LIST[rn].Url
  38. }
  39. postUrl = postUrl + urlStr
  40. step++
  41. postData, err := json.Marshal(param)
  42. if err != nil {
  43. return
  44. }
  45. result, err := HttpPost(postUrl, string(postData), "application/json")
  46. if err != nil {
  47. return
  48. }
  49. utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";" + string(postData) + ";result:" + string(result))
  50. err = json.Unmarshal(result, &resp)
  51. if err != nil {
  52. return
  53. }
  54. return resp, nil
  55. }
  56. func HttpPost(url, postData string, params ...string) ([]byte, error) {
  57. body := ioutil.NopCloser(strings.NewReader(postData))
  58. client := &http.Client{}
  59. req, err := http.NewRequest("POST", url, body)
  60. if err != nil {
  61. return nil, err
  62. }
  63. contentType := "application/x-www-form-urlencoded;charset=utf-8"
  64. if len(params) > 0 && params[0] != "" {
  65. contentType = params[0]
  66. }
  67. req.Header.Set("Content-Type", contentType)
  68. req.Header.Set("authorization", utils.MD5(utils.APP_EDB_LIB_NAME_EN+utils.EDB_LIB_Md5_KEY))
  69. resp, err := client.Do(req)
  70. defer resp.Body.Close()
  71. b, err := ioutil.ReadAll(resp.Body)
  72. //fmt.Println("HttpPost:" + string(b))
  73. return b, err
  74. }