May1145 1 year ago
parent
commit
cca6e48b4a
4 changed files with 51 additions and 1 deletions
  1. 7 1
      trading-go/controller/usercontroller.go
  2. 1 0
      trading-go/go.mod
  3. 2 0
      trading-go/go.sum
  4. 41 0
      trading-go/util/jwt-go.go

+ 7 - 1
trading-go/controller/usercontroller.go

@@ -37,8 +37,14 @@ func Login(c *gin.Context) {
 		response.Fail(c.Writer, msg, 500)
 		return
 	}
+	token, err := util.CreatToken(nu.Uid)
+	if err != nil {
+		msg := err.Error()
+		response.Fail(c.Writer, msg, 500)
+		return
+	}
 	data := gin.H{
-		"token":     "test",
+		"token":     token,
 		"user_info": nu,
 	}
 	response.Success(c.Writer, "success", data)

+ 1 - 0
trading-go/go.mod

@@ -18,6 +18,7 @@ require (
 	github.com/go-redis/redis/v8 v8.11.5 // indirect
 	github.com/go-sql-driver/mysql v1.7.1 // indirect
 	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/jmoiron/sqlx v1.3.5 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect

+ 2 - 0
trading-go/go.sum

@@ -92,6 +92,8 @@ github.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrt
 github.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
 github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
 github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
+github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
+github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
 github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
 github.com/golang/groupcache v0.0.0-20190702054246-869f871628b6/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
 github.com/golang/groupcache v0.0.0-20191227052852-215e87163ea7/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=

+ 41 - 0
trading-go/util/jwt-go.go

@@ -0,0 +1,41 @@
+package util
+
+import (
+	"github.com/golang-jwt/jwt"
+	"time"
+)
+
+var jwtKey = []byte("hello223rdwvwdfforl-'dandfdsafdafdsfsmay")
+
+type Claims struct {
+	UserId string
+	jwt.StandardClaims
+}
+
+func CreatToken(uid string) (string, error) {
+	expirationTime := time.Now().Add(31 * 24 * time.Hour)
+	claims := &Claims{
+		UserId: uid,
+		StandardClaims: jwt.StandardClaims{
+			ExpiresAt: expirationTime.Unix(),
+			IssuedAt:  time.Now().Unix(),
+			Issuer:    "may1145",
+			Subject:   "user token",
+		},
+	}
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
+	tokenString, err := token.SignedString(jwtKey)
+	if err != nil {
+		return "", err
+	}
+	return tokenString, nil
+}
+
+func ParseToken(tokenString string) (*jwt.Token, *Claims, error) {
+	claims := &Claims{}
+
+	token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (i any, err error) {
+		return jwtKey, nil
+	})
+	return token, claims, err
+}