mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-08-26 16:46:14 +08:00
50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
// Package tcp
|
|
// @Link https://github.com/bufanyun/hotgo
|
|
// @Copyright Copyright (c) 2023 HotGo CLI
|
|
// @Author Ms <133814250@qq.com>
|
|
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
|
package tcp
|
|
|
|
import (
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
"hotgo/internal/model/input/msgin"
|
|
)
|
|
|
|
type Sign interface {
|
|
SetSign(appId, secretKey string) *msgin.RpcMsg
|
|
SetTraceID(traceID string)
|
|
}
|
|
|
|
// PkgSign 打包签名
|
|
func PkgSign(data interface{}, appId, secretKey, traceID string) *msgin.RpcMsg {
|
|
if c, ok := data.(Sign); ok {
|
|
c.SetTraceID(traceID)
|
|
return c.SetSign(appId, secretKey)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// VerifySign 验证签名
|
|
func VerifySign(data interface{}, appId, secretKey string) (in *msgin.RpcMsg, err error) {
|
|
// 无密钥，无需签名
|
|
if secretKey == "" {
|
|
return
|
|
}
|
|
|
|
if err = gconv.Scan(data, &in); err != nil {
|
|
return
|
|
}
|
|
|
|
if appId != in.AppId {
|
|
err = gerror.New("appId invalid")
|
|
return
|
|
}
|
|
|
|
if in.Sign != in.GetSign(secretKey) {
|
|
err = gerror.New("sign invalid")
|
|
return
|
|
}
|
|
return
|
|
}
|