base-golang/tool/http.go

package tool

import (
	"encoding/json"
	"git.bvbej.com/bvbej/base-golang/pkg/aes"
	"git.bvbej.com/bvbej/base-golang/pkg/hmac"
	"git.bvbej.com/bvbej/base-golang/pkg/httpclient"
	"git.bvbej.com/bvbej/base-golang/pkg/time_parse"
	"github.com/tidwall/gjson"
	netUrl "net/url"
	"strconv"
	"time"
)

func HttpPostByAes(url, productID, key, vi string, params netUrl.Values) (body []byte, err error) {
	var signatureDatetime = time_parse.CSTLayoutString()
	unix, _ := time_parse.CSTLayoutStringToUnix(signatureDatetime)
	params.Set("t", strconv.FormatInt(unix, 10))
	var sign string
	sign, err = aes.New(key, vi).EncryptCBC(params.Encode(), false)
	if err != nil {
		return body, err
	}
	body, err = httpclient.PostForm(
		url,
		params,
		httpclient.WithTTL(time.Second*30),
		httpclient.WithHeader("Signature", sign),
		httpclient.WithHeader("Signature-Datetime", signatureDatetime),
		httpclient.WithHeader("Product-ID", productID),
		httpclient.WithOnFailedRetry(1, time.Millisecond*200,
			func(body []byte) (shouldRetry bool) {
				return len(body) == 0
			},
		),
	)
	return body, err
}

func HttpPostByAesWithTimeout(url, productID, key, vi string, params netUrl.Values, timeout time.Duration) (body []byte, err error) {
	var signatureDatetime = time_parse.CSTLayoutString()
	unix, _ := time_parse.CSTLayoutStringToUnix(signatureDatetime)
	params.Set("t", strconv.FormatInt(unix, 10))
	var sign string
	sign, err = aes.New(key, vi).EncryptCBC(params.Encode(), false)
	if err != nil {
		return body, err
	}
	body, err = httpclient.PostForm(
		url,
		params,
		httpclient.WithTTL(timeout),
		httpclient.WithHeader("Signature", sign),
		httpclient.WithHeader("Signature-Datetime", signatureDatetime),
		httpclient.WithHeader("Product-ID", productID),
		httpclient.WithOnFailedRetry(3, time.Second*1,
			func(body []byte) (shouldRetry bool) {
				return len(body) == 0
			},
		),
	)
	return body, err
}

func DingTalkAlertMarkdown(accessToken, secret, title, text string) bool {
	params := netUrl.Values{}
	timestamp := strconv.FormatInt(time_parse.GetMilliTimestamp(), 10)
	signData := timestamp + "\n" + secret
	sign := netUrl.QueryEscape(hmac.New(secret).Sha256ToBase64String(signData))
	params.Set("access_token", accessToken)
	params.Set("sign", sign)
	params.Set("timestamp", timestamp)
	var content struct {
		MsgType  string `json:"msgtype"`
		Markdown struct {
			Title string `json:"title"`
			Text  string `json:"text"`
		} `json:"markdown"`
	}
	content.MsgType = "markdown"
	content.Markdown.Title = title
	content.Markdown.Text = text
	body, _ := json.Marshal(content)
	response, err := httpclient.PostJSON(
		"https://oapi.dingtalk.com/robot/send?"+params.Encode(),
		body,
		httpclient.WithTTL(time.Second*5),
		httpclient.WithHeader("Content-Type", "application/json;charset=utf-8"),
		httpclient.WithOnFailedRetry(3, time.Second*1,
			func(body []byte) (shouldRetry bool) {
				return len(body) == 0
			},
		),
	)
	if err != nil {
		return false
	}
	return gjson.Get(string(response), "errcode").Int() == 0
}

func DingTalkAlertText(accessToken, secret string, text string) bool {
	params := netUrl.Values{}
	timestamp := strconv.FormatInt(time_parse.GetMilliTimestamp(), 10)
	signData := timestamp + "\n" + secret
	sign := netUrl.QueryEscape(hmac.New(secret).Sha256ToBase64String(signData))
	params.Set("access_token", accessToken)
	params.Set("sign", sign)
	params.Set("timestamp", timestamp)
	var content struct {
		MsgType string `json:"msgtype"`
		Text    struct {
			Content string `json:"content"`
		} `json:"text"`
	}
	content.MsgType = "text"
	content.Text.Content = text
	body, _ := json.Marshal(content)
	response, err := httpclient.PostJSON(
		"https://oapi.dingtalk.com/robot/send?"+params.Encode(),
		body,
		httpclient.WithTTL(time.Second*5),
		httpclient.WithHeader("Content-Type", "application/json;charset=utf-8"),
		httpclient.WithOnFailedRetry(3, time.Second*1,
			func(body []byte) (shouldRetry bool) {
				return len(body) == 0
			},
		),
	)
	if err != nil {
		return false
	}
	return gjson.Get(string(response), "errcode").Int() == 0
}