添加文件: intelligence_finance_v1_ga/crypto.go
This commit is contained in:
parent
56ce54eadc
commit
dc5b1f2561
|
|
@ -0,0 +1,74 @@
|
||||||
|
package intelligence_finance_v1_ga
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/hmac"
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/sha256"
|
||||||
|
"encoding/base64"
|
||||||
|
"fmt"
|
||||||
|
"math/big"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GenerateTimestamp 生成秒级时间戳。
|
||||||
|
func GenerateTimestamp() string {
|
||||||
|
return fmt.Sprintf("%d", time.Now().Unix())
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenerateTimestampMillis 生成毫秒级时间戳。
|
||||||
|
func GenerateTimestampMillis() string {
|
||||||
|
return fmt.Sprintf("%d", time.Now().UnixMilli())
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenerateNonce 生成指定长度的随机字符串(加密安全)。
|
||||||
|
func GenerateNonce(length int) (string, error) {
|
||||||
|
const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||||||
|
b := make([]byte, length)
|
||||||
|
for i := range b {
|
||||||
|
num, err := rand.Int(rand.Reader, big.NewInt(int64(len(charset))))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
b[i] = charset[num.Int64()]
|
||||||
|
}
|
||||||
|
return string(b), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// HmacSHA256Base64 使用 HmacSHA256 算法对 data 进行签名,并返回 Base64 编码结果。
|
||||||
|
// 密钥为 client-secret,签名数据为 timestamp + nonce。
|
||||||
|
func HmacSHA256Base64(secret, data string) string {
|
||||||
|
mac := hmac.New(sha256.New, []byte(secret))
|
||||||
|
mac.Write([]byte(data))
|
||||||
|
return base64.StdEncoding.EncodeToString(mac.Sum(nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildSignString 方式1:字典序排序拼接。
|
||||||
|
// 空值会被排除,且自动排除 sign / signature 字段本身。
|
||||||
|
func BuildSignString(params map[string]string) string {
|
||||||
|
keys := make([]string, 0, len(params))
|
||||||
|
for k, v := range params {
|
||||||
|
if v != "" && k != "sign" && k != "signature" {
|
||||||
|
keys = append(keys, k)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sort.Strings(keys)
|
||||||
|
|
||||||
|
var parts []string
|
||||||
|
for _, k := range keys {
|
||||||
|
parts = append(parts, fmt.Sprintf("%s=%s", k, params[k]))
|
||||||
|
}
|
||||||
|
return strings.Join(parts, "&")
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildSignStringOrdered 方式2:固定顺序拼接。
|
||||||
|
func BuildSignStringOrdered(params map[string]string, orderedKeys []string) string {
|
||||||
|
var parts []string
|
||||||
|
for _, k := range orderedKeys {
|
||||||
|
if v, ok := params[k]; ok && v != "" {
|
||||||
|
parts = append(parts, fmt.Sprintf("%s=%s", k, v))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return strings.Join(parts, "&")
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue