feat 初始化

utils/jwt.go

@@ -37,3 +37,35 @@ func ParseToken(tokenString string, secret string) (*Claims, error) {
 	}
 	return claims, nil
 }
+
+type PhoneClaims struct {
+	Phone string `json:"phone"`
+	jwt.RegisteredClaims
+}
+
+func GenPhoneToken(phone string, secret string, expires time.Duration) (string, error) {
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, PhoneClaims{
+		Phone: phone,
+		RegisteredClaims: jwt.RegisteredClaims{
+			ExpiresAt: jwt.NewNumericDate(time.Now().Add(expires)),
+		},
+	})
+	return token.SignedString([]byte(secret))
+}
+
+func ParsePhoneToken(tokenString string, secret string) (*PhoneClaims, error) {
+	if tokenString == "" {
+		return nil, errors.New("invalid token")
+	}
+	claims := &PhoneClaims{}
+	token, err := jwt.ParseWithClaims(tokenString, claims, func(_ *jwt.Token) (interface{}, error) {
+		return []byte(secret), nil
+	})
+	if err != nil {
+		return nil, err
+	}
+	if !token.Valid {
+		return nil, errors.New("invalid token")
+	}
+	return claims, nil
+}