eta_task/services/eta_hub/edb.go

package eta_hub

import (
	"encoding/json"
	"errors"
	"eta/eta_task/utils"
	"fmt"
	"io"
	"net/http"
	"sort"
	"strings"
	"time"
)

// BaseEtaHubDataResp 基础返回参数
type BaseEtaHubDataResp struct {
	Ret         int
	Msg         string
	ErrMsg      string
	ErrCode     string
	Data        interface{}
	Success     bool `description:"true 执行成功，false 执行失败"`
	IsSendEmail bool `json:"-" description:"true 发送邮件，false 不发送邮件"`
	IsAddLog    bool `json:"-" description:"true 新增操作日志，false 不新增操作日志" `
}

// HttpEtaHubPost
// @Description: eta-bridge的post请求
// @author: Roc
// @datetime 2024-02-28 16:55:02
// @param urlStr string
// @param postData string
// @param param interface{}
// @return bResult []byte
// @return err error
// @return errMsg string
func HttpEtaHubPost(urlStr string, param interface{}) (bResult []byte, err error, errMsg string) {
	if utils.EtaHubUrl == "" {
		errMsg = `未配置第三方的桥接服务地址`
		err = errors.New(errMsg)
		return
	}
	data, e := json.Marshal(param)
	if e != nil {
		err = fmt.Errorf("data json marshal err: %s", e.Error())
		return
	}

	postUrl := utils.EtaHubUrl + "/v1" + urlStr

	body := io.NopCloser(strings.NewReader(string(data)))
	client := &http.Client{}
	req, e := http.NewRequest("POST", postUrl, body)
	if e != nil {
		err = fmt.Errorf("http create request err: %s", e.Error())
		return
	}

	// 随机字符串
	nonce := utils.GetRandStringNoSpecialChar(16)
	timestamp := fmt.Sprint(time.Now().Second())
	sign := GetSign(nonce, timestamp)

	contentType := "application/json;charset=utf-8"
	req.Header.Set("Content-Type", contentType)
	req.Header.Set("nonce", nonce)
	req.Header.Set("timestamp", timestamp)
	req.Header.Set("appid", utils.APPID)
	req.Header.Set("signature", sign)

	resp, e := client.Do(req)
	if e != nil {
		err = fmt.Errorf("http client do err: %s", e.Error())
		return
	}
	defer func() {
		_ = resp.Body.Close()
	}()
	bResult, e = io.ReadAll(resp.Body)
	if e != nil {
		err = fmt.Errorf("resp body read err: %s", e.Error())
		return
	}
	if len(bResult) == 0 {
		err = fmt.Errorf("resp body is empty")
		return
	}

	result := new(BaseEtaHubDataResp)
	if e = json.Unmarshal(bResult, &result); e != nil {
		err = fmt.Errorf("result unmarshal err: %s\nresult: %s", e.Error(), string(bResult))
		utils.FileLog.Info("桥接服务post请求失败:\n" + string(bResult))
		return
	}
	if result.Ret != 200 {
		errMsg = result.Msg
		err = fmt.Errorf("result: %s", string(bResult))
		return
	}

	return
}

// HttpEtaHubGet
// @Description: eta-bridge的get请求
// @author: Roc
// @datetime 2024-02-28 16:55:02
// @param urlStr string
// @param postData string
// @return bResult []byte
// @return err error
// @return errMsg string
func HttpEtaHubGet(urlStr string) (bResult []byte, err error, errMsg string) {
	if utils.EtaHubUrl == "" {
		errMsg = `未配置第三方的桥接服务地址`
		err = errors.New(errMsg)
		return
	}

	postUrl := utils.EtaHubUrl + "/v1" + urlStr

	client := &http.Client{}
	req, e := http.NewRequest("GET", postUrl, nil)
	if e != nil {
		err = fmt.Errorf("http create request err: %s", e.Error())
		return
	}

	contentType := "application/json;charset=utf-8"
	// 随机字符串
	nonce := utils.GetRandStringNoSpecialChar(16)
	timestamp := fmt.Sprint(time.Now().Second())
	sign := GetSign(nonce, timestamp)

	req.Header.Set("Content-Type", contentType)
	req.Header.Set("nonce", nonce)
	req.Header.Set("timestamp", timestamp)
	req.Header.Set("appid", utils.APPID)
	req.Header.Set("signature", sign)

	resp, e := client.Do(req)
	if e != nil {
		err = fmt.Errorf("http client do err: %s", e.Error())
		return
	}
	defer func() {
		_ = resp.Body.Close()
	}()
	bResult, e = io.ReadAll(resp.Body)
	if e != nil {
		err = fmt.Errorf("resp body read err: %s", e.Error())
		return
	}
	if len(bResult) == 0 {
		err = fmt.Errorf("resp body is empty")
		return
	}

	result := new(BaseEtaHubDataResp)
	if e = json.Unmarshal(bResult, &result); e != nil {
		err = fmt.Errorf("result unmarshal err: %s\nresult: %s", e.Error(), string(bResult))
		utils.FileLog.Info("桥接服务get请求失败:\n" + string(bResult))
		return
	}
	if result.Ret != 200 {
		errMsg = result.Msg
		err = fmt.Errorf("result: %s", string(bResult))
		return
	}

	return
}

func GetSign(nonce, timestamp string) (sign string) {
	signStrMap := map[string]string{
		"nonce":     nonce,
		"timestamp": timestamp,
		"appid":     utils.APPID,
	}
	keys := make([]string, 0, len(signStrMap))
	for k := range signStrMap {
		keys = append(keys, k)
	}
	sort.Strings(keys)
	var signStr string
	for _, k := range keys {
		signStr += k + "=" + signStrMap[k] + "&"
	}
	signStr = strings.Trim(signStr, "&")
	//fmt.Println("signStr:" + signStr)
	sign = utils.HmacSha256ToBase64(utils.SECRET, signStr)
	//fmt.Println("sign:" + sign)
	return
}