Files
servicebase/pkg/helper/Helper.go
2025-11-18 18:33:22 +08:00

156 lines
2.8 KiB
Go

package helper
import (
"crypto/md5"
"crypto/sha1"
"encoding/base64"
"encoding/hex"
"encoding/json"
"io"
"io/ioutil"
"net/http"
"sort"
"strconv"
"strings"
"time"
"github.com/spf13/viper"
)
// 生成header
func GenerateApiHeader(jsonBody string) map[string]string {
sign := GenerateSign(jsonBody)
headerMap := getCommonHeaderMap()
headerMap["Signature"] = sign
return headerMap
}
// 根据body生成签名
func GenerateSign(jsonBody string) string {
var mapRequest map[string]string
err := json.Unmarshal([]byte(jsonBody), &mapRequest)
if err != nil {
return ""
}
commonHeaderMap := getCommonHeaderMap()
for k, v := range commonHeaderMap {
mapRequest[k] = v
}
sign := createSign(mapRequest)
return sign
}
// 获取公共的header字段
func getCommonHeaderMap() map[string]string {
headerMap := make(map[string]string, 0)
headerMap["ClientVersion"] = "1.0.1"
headerMap["DeviceId"] = "MWEB"
headerMap["Platform"] = "H5"
headerMap["MarketChannel"] = ""
headerMap["DeviceModel"] = "MWEB"
headerMap["ApiVersionNum"] = viper.GetString("api-num")
headerMap["TimeStamp"] = strconv.Itoa(time.Now().Second())
headerMap["BundleId"] = "meetalkH5"
headerMap["OsVersion"] = "10.01"
return headerMap
}
func createSign(params map[string]string) string {
keys := make([]string, 20)
for key, _ := range params {
if key == "Signature" {
continue
}
keys = append(keys, key)
}
//按key升序
sort.Strings(keys)
//把KEY值合并为字符串
waitSignString := ""
for _, value := range keys {
waitSignString += base64.StdEncoding.EncodeToString([]byte(params[value]))
}
//加上密钥
appSecret := "x63363eacf804b4394a120aea240fd9a"
waitSignString += appSecret
//sha1加密
t := sha1.New()
io.WriteString(t, waitSignString)
sign := t.Sum(nil)
return hex.EncodeToString(sign)
}
// 复杂http请求
func HttpDo(httpMethod string, url string, headerMap map[string]string, rawBody string) (remoteResponse string, err error) {
client := &http.Client{}
req, err0 := http.NewRequest(httpMethod, url, strings.NewReader(rawBody))
if err0 != nil {
err = err0
return
}
if len(headerMap) > 0 {
for k, v := range headerMap {
req.Header.Set(k, v)
}
}
resp, err1 := client.Do(req)
if err1 != nil {
err = err1
return
}
defer resp.Body.Close()
body, err2 := ioutil.ReadAll(resp.Body)
if err2 != nil {
err = err2
return
}
remoteResponse = string(body)
return
}
func StringToMD5(waitMD5string string) string {
h := md5.New()
h.Write([]byte(waitMD5string))
cipherStr := h.Sum(nil)
result := hex.EncodeToString(cipherStr)
return result
}
// SHA1加密
func StringToSHA1(waitMD5string string) string {
h := sha1.New()
h.Write([]byte(waitMD5string))
cipherStr := h.Sum(nil)
result := hex.EncodeToString(cipherStr)
return result
}