-
Notifications
You must be signed in to change notification settings - Fork 0
/
sign.go
80 lines (68 loc) · 1.82 KB
/
sign.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
// Copyright (c) 2022. Homeland Interactive Technology Ltd. All rights reserved.
package ruixuego
import (
"crypto/sha1"
"encoding/hex"
"hash"
"net/url"
"strconv"
"sync"
)
var (
sha1Pool = &sync.Pool{
New: func() interface{} {
return sha1.New() // nolint:gosec
},
}
)
// GetSign 获取请求签名,该签名通过 sha1(TraceID+Timestamp+CPKey) 得来
func GetSign(traceID, ts string) string {
h := sha1Pool.Get().(hash.Hash)
_, _ = h.Write([]byte(traceID + ts + config.CPKey))
ret := hex.EncodeToString(h.Sum(nil))
h.Reset()
sha1Pool.Put(h)
return ret
}
func GetLoginResultSign(result *LoginResult, signFields []string) string {
params := url.Values{}
for _, field := range signFields {
switch field {
case SignFieldNickname:
params.Set(field, result.Nickname)
case SignFieldAvatar:
params.Set(field, result.Avatar)
case SignFieldOpenID:
params.Set(field, result.OpenID)
case SignFieldRegion:
params.Set(field, result.Region)
case SignFieldSex:
params.Set(field, strconv.FormatInt(int64(result.Sex), 10))
case SignFieldAge:
params.Set(field, strconv.FormatInt(int64(result.Age), 10))
}
}
signSource := params.Encode() + config.CPKey
h := sha1Pool.Get().(hash.Hash)
h.Write([]byte(signSource))
sign := hex.EncodeToString(h.Sum(nil))
h.Reset()
sha1Pool.Put(h)
return sign
}
type LoginResult struct {
Nickname string `json:"nickname"` // 昵称
Avatar string `json:"avatar"` // 头像地址
OpenID string `json:"openid"` // 加密后的瑞雪openid
Region string `json:"region"` // 地区码
Sex int8 `json:"sex"` // 性别
Age uint8 `json:"age"` // 年龄
}
const (
SignFieldNickname = "nickname"
SignFieldAvatar = "avatar"
SignFieldOpenID = "openid"
SignFieldRegion = "region"
SignFieldSex = "sex"
SignFieldAge = "age"
)