-
Notifications
You must be signed in to change notification settings - Fork 0
/
signersdk.go
104 lines (93 loc) · 2.46 KB
/
signersdk.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package sdk
import (
"encoding/json"
"errors"
"github.com/Golang-Tools/grpcsdk"
"github.com/Golang-Tools/jwthelper/v2/jwt_pb"
"github.com/Golang-Tools/jwthelper/v2/jwtsigner_pb"
"github.com/Golang-Tools/jwthelper/v2/signoptions"
"github.com/Golang-Tools/loggerhelper/v2"
"github.com/Golang-Tools/optparams"
)
type SignerSDK struct {
client *grpcsdk.SDK[jwtsigner_pb.JwtsignerClient]
}
func NewSignerSDK() *SignerSDK {
s := new(SignerSDK)
s.client = grpcsdk.New(jwtsigner_pb.NewJwtsignerClient, &jwtsigner_pb.Jwtsigner_ServiceDesc)
return s
}
func (s *SignerSDK) Init(opts ...optparams.Option[grpcsdk.SDKConfig]) {
s.client.Init(opts...)
}
func (s *SignerSDK) GetLogger() *loggerhelper.Log {
return s.client.Logger
}
//Close 断开连接
func (c *SignerSDK) Close() error {
return c.client.Close()
}
//Meta 查看远端签名器的元信息
func (c *SignerSDK) Meta() (*jwt_pb.SignerMeta, error) {
ctx, cancel := c.client.NewCtx()
defer cancel()
Conn, release := c.client.GetClient()
defer release()
res, err := Conn.Meta(ctx, &jwtsigner_pb.MetaRequest{})
if err != nil {
return nil, err
}
if res.Status == nil || res.Status.Status == jwt_pb.ResponseStatus_FAILED {
var err error
if res.Status.Message != "" {
err = errors.New(res.Status.Message)
} else {
err = ErrRpcResponseError
}
return nil, err
}
return res.Data, nil
}
// Sign 签名一个token
func (c *SignerSDK) Sign(payload interface{}, opts ...optparams.Option[signoptions.SignOptions]) (*jwt_pb.Token, error) {
if payload == nil {
payload = map[string]interface{}{}
}
payloadb, err := json.Marshal(payload)
if err != nil {
return nil, err
}
defaultopt := signoptions.DefaultSignOptions
optparams.GetOption(&defaultopt, opts...)
query := jwtsigner_pb.SignRequest{
Sub: defaultopt.Sub,
Exp: defaultopt.Exp,
Nbf: defaultopt.Nbf,
Refreshexp: defaultopt.RefreshExp,
Payload: payloadb,
Jti: defaultopt.Jti,
Aud: defaultopt.Aud,
}
ctx, cancel := c.client.NewCtx()
defer cancel()
Conn, release := c.client.GetClient()
defer release()
res, err := Conn.Sign(ctx, &query)
if err != nil {
return nil, err
}
if res.Status == nil || res.Status.Status == jwt_pb.ResponseStatus_FAILED {
var err error
if res.Status.Message != "" {
err = errors.New(res.Status.Message)
} else {
err = ErrRpcResponseError
}
return nil, err
}
return res.Token, nil
}
var DefaultSigner *SignerSDK
func init() {
DefaultSigner = NewSignerSDK()
}