refactor: 包名变更ctx -> req

server/pkg/req/token.go

@@ -0,0 +1,61 @@
+package req
+
+import (
+	"errors"
+
+	"mayfly-go/pkg/biz"
+	"mayfly-go/pkg/config"
+	"mayfly-go/pkg/global"
+	"mayfly-go/pkg/model"
+	"mayfly-go/pkg/utils"
+	"time"
+
+	"github.com/golang-jwt/jwt/v4"
+)
+
+func InitTokenConfig() {
+	JwtKey = config.Conf.Jwt.Key
+	ExpTime = config.Conf.Jwt.ExpireTime
+}
+
+var (
+	JwtKey  string
+	ExpTime uint64
+)
+
+// 创建用户token
+func CreateToken(userId uint64, username string) string {
+	// 带权限创建令牌
+	// 设置有效期，过期需要重新登录获取token
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
+		"id":       userId,
+		"username": username,
+		"exp":      time.Now().Add(time.Minute * time.Duration(ExpTime)).Unix(),
+	})
+
+	// 如果配置文件中的jwt key为空，则随机生成字符串
+	if JwtKey == "" {
+		JwtKey = utils.RandString(32)
+		global.Log.Infof("config.yml未配置jwt.key, 随机生成key为: %s", JwtKey)
+	}
+	// 使用自定义字符串加密 and get the complete encoded token as a string
+	tokenString, err := token.SignedString([]byte(JwtKey))
+	biz.ErrIsNil(err, "token创建失败")
+	return tokenString
+}
+
+// 解析token，并返回登录者账号信息
+func ParseToken(tokenStr string) (*model.LoginAccount, error) {
+	if tokenStr == "" {
+		return nil, errors.New("token error")
+	}
+	// Parse token
+	token, err := jwt.Parse(tokenStr, func(token *jwt.Token) (interface{}, error) {
+		return []byte(JwtKey), nil
+	})
+	if err != nil || token == nil {
+		return nil, err
+	}
+	i := token.Claims.(jwt.MapClaims)
+	return &model.LoginAccount{Id: uint64(i["id"].(float64)), Username: i["username"].(string)}, nil
+}