first commit

2024-07-23 10:23:43 +08:00
commit 7b4c2521a3
126 changed files with 15931 additions and 0 deletions
--- a/tool/http.go
+++ b/tool/http.go
@@ -0,0 +1,133 @@
+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
+}