hz_data_api.go 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. package services
  2. import (
  3. "bytes"
  4. "crypto/md5"
  5. "encoding/json"
  6. "errors"
  7. "eta/eta_task/utils"
  8. "fmt"
  9. "github.com/shopspring/decimal"
  10. "io/ioutil"
  11. "net/http"
  12. "reflect"
  13. "sort"
  14. "strings"
  15. "time"
  16. )
  17. func HttpPost(funcName, method string, postDataMap map[string]interface{}) (result string, err error) {
  18. defer func() {
  19. if err != nil {
  20. fmt.Println("Sync hz crawler err:" + err.Error())
  21. utils.FileLog.Info("Sync hz crawler err:" + err.Error())
  22. }
  23. }()
  24. bytesData, err := handlePostData(postDataMap)
  25. if err != nil {
  26. fmt.Println("handlePostData Err:" + err.Error())
  27. return
  28. }
  29. fmt.Println("URL:" + utils.HzDataApi + method)
  30. fmt.Println("Params:" + string(bytesData))
  31. client := &http.Client{}
  32. req, err := http.NewRequest("POST", utils.HzDataApi+method, bytes.NewReader(bytesData))
  33. if err != nil {
  34. fmt.Println("http.NewRequest Err:" + err.Error())
  35. return "", err
  36. }
  37. req.Header.Set("content-type", "application/json")
  38. utils.FileLog.Info(fmt.Sprintf("请求函数:%s ;请求体:%s", funcName, req))
  39. resp, err := client.Do(req)
  40. if err != nil {
  41. return "", err
  42. }
  43. body, err := ioutil.ReadAll(resp.Body)
  44. if err != nil {
  45. return "", err
  46. }
  47. result = string(body)
  48. utils.FileLog.Info(fmt.Sprintf("请求函数:%s ;返回体:%s", funcName, result))
  49. return
  50. }
  51. // handlePostData 处理post请求数据
  52. func handlePostData(postDataMap map[string]interface{}) (bytesData []byte, err error) {
  53. postDataMap["nonce_str"] = utils.GetRandString(16) // 随机字符串
  54. postDataMap["timestamp"] = time.Now().Unix() //当前格林威治时间,int64类型
  55. postDataMap["appid"] = utils.APPID //当前格林威治时间,int64类型
  56. // 待签名数据
  57. signData := convertParamInterface(postDataMap)
  58. sign, err := getSignData(signData)
  59. if err != nil {
  60. return
  61. }
  62. postDataMap["sign"] = sign //签名
  63. bytesData, err = json.Marshal(postDataMap)
  64. return
  65. }
  66. // 将请求传入的数据格式转换成签名需要的格式(目前只能处理简单的类型,数组、对象暂不支持)
  67. func convertParamInterface(params map[string]interface{}) (signData map[string]string) {
  68. signData = make(map[string]string)
  69. for key := range params {
  70. val := ``
  71. //fmt.Println("key", key, ";val:", params[key], ";type:", reflect.TypeOf(params[key]))
  72. //signData[key] = params[key][0]
  73. tmpVal := params[key]
  74. switch reflect.TypeOf(tmpVal).Kind() {
  75. case reflect.String:
  76. val = fmt.Sprint(tmpVal)
  77. case reflect.Int, reflect.Int16, reflect.Int64, reflect.Int32, reflect.Int8:
  78. val = fmt.Sprint(tmpVal)
  79. case reflect.Uint, reflect.Uint32, reflect.Uint16, reflect.Uint8, reflect.Uint64:
  80. val = fmt.Sprint(tmpVal)
  81. case reflect.Bool:
  82. val = fmt.Sprint(tmpVal)
  83. case reflect.Float64:
  84. decimalNum := decimal.NewFromFloat(tmpVal.(float64))
  85. val = decimalNum.String()
  86. //val = strconv.FormatFloat(tmpVal.(float64), 'E', -1, 64) //float64
  87. case reflect.Float32:
  88. decimalNum := decimal.NewFromFloat32(tmpVal.(float32))
  89. val = decimalNum.String()
  90. }
  91. signData[key] = val
  92. }
  93. return signData
  94. }
  95. // getSignData 获取参数签名
  96. func getSignData(postData map[string]string) (sign string, err error) {
  97. appid := utils.APPID
  98. if appid == "" {
  99. err = errors.New("参数异常,缺少appid")
  100. return
  101. }
  102. secret := utils.SECRET
  103. if secret == "" {
  104. err = errors.New("参数异常,缺少secret")
  105. return
  106. }
  107. if postData["nonce_str"] == "" {
  108. err = errors.New("参数异常,缺少随机字符串")
  109. return
  110. }
  111. if postData["timestamp"] == "" {
  112. err = errors.New("参数异常,缺少时间戳")
  113. return
  114. }
  115. //先取出除sign外的所有的提交的参数key
  116. var keys []string
  117. for k := range postData {
  118. if k != "sign" {
  119. keys = append(keys, k)
  120. }
  121. }
  122. //1,根据参数名称的ASCII码表的顺序排序
  123. sort.Strings(keys)
  124. //2 根据排序后的参数名称,取出对应的值,并拼接字符串
  125. var signStr string
  126. for _, v := range keys {
  127. signStr += v + "=" + postData[v] + "&"
  128. }
  129. //3,全转小写(md5(拼装的字符串后+分配给你的app_secret))
  130. //sign := strings.ToLower(fmt.Sprintf("%x", md5.Sum([]byte(strings.Trim(signStr, "&")+key))))
  131. //md5.Sum([]byte(signStr+"key="+key)) 这是md5加密出来后的每个字符的ascall码,需要再转换成对应的字符
  132. //3,全转大写(md5(拼装的字符串后+分配给你的app_secret))
  133. sign = strings.ToUpper(fmt.Sprintf("%x", md5.Sum([]byte(signStr+"secret="+secret))))
  134. return
  135. }