【Go】十三、TOKEN机制与跨域处理方式

身份校验

对于微服务场景来说，使用 cookie + session 来进行身份校验是一种不合适的做法，因为 生成的sessionid 是不存储任何信息的，这样我们如果要在其他模块中进行身份校验就是做不到的，例如：我们无法在商品模块中筛选出对应登录用户上传的商品，因为我们无法通过sessionid 知道当前登录的是哪个用户。

由此，登录逻辑中产生了 TOKEN 的做法，TOKEN 中会携带一定的用户信息，这样即使不处于用户模块，也可以获取用户的相关信息 - 利用 TOKEN 中的用户信息去另外的模块进行查找。

JSON WEB TOKEN（JWT）

这里使用 jwt 来对 TOKEN进行构建

jwt 的基本结构：

xxxxx.yyyyy.zzzzz

其中：xxxxx为加密算法

yyyyy为TOKEN中保存的信息

zzzzz为签名信息

例如：

这是一串典型的JWT-TOKEN
eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c下面是他的内容
{"alg": "HS256","typ": "JWT"
}{"sub": "1234567890","name": "John Doe","iat": 1516239022
}HMACSHA256(base64UrlEncode(header) + "." +base64UrlEncode(payload),your-256-bit-secret
) 

JWT 在GIN 中的实际使用

构建新目录 models

user-web

models

request.go

middlewares

jwt.go

config

config.go

request.go：

package modelsimport "github.com/dgrijalva/jwt-go"// 这里是 给 jwt 构建的信息
type CustomClaims struct {ID                 uint   // 用户IDNickName           string // 用户昵称AuthorityID        uint   // 用户角色 rolejwt.StandardClaims        // JWT 基本元素
}

jwt.go

注意此时的 jwt.go 可以直接粘贴，但需要对应进行修改其他引用文件

package middlewaresimport ("errors""mxshop-api/user-web/global""mxshop-api/user-web/models""net/http""time""github.com/dgrijalva/jwt-go""github.com/gin-gonic/gin"
)// TOKEN 验证方法
func JWTAuth() gin.HandlerFunc {return func(c *gin.Context) {// 我们这里jwt鉴权取头部信息 x-token 登录时回返回token信息 这里前端需要把token存储到cookie或者本地localSstorage中 不过需要跟后端协商过期时间 可以约定刷新令牌或者重新登录token := c.Request.Header.Get("x-token")if token == "" {c.JSON(http.StatusUnauthorized, map[string]string{"msg": "请登录",})c.Abort()return}j := NewJWT()// parseToken 解析token包含的信息claims, err := j.ParseToken(token)if err != nil {if err == TokenExpired {if err == TokenExpired {c.JSON(http.StatusUnauthorized, map[string]string{"msg": "授权已过期",})c.Abort()return}}c.JSON(http.StatusUnauthorized, "未登陆")c.Abort()return}c.Set("claims", claims)c.Set("userId", claims.ID)c.Next()}
}type JWT struct {SigningKey []byte
}// 错误信息
var (TokenExpired     = errors.New("Token is expired")TokenNotValidYet = errors.New("Token not active yet")TokenMalformed   = errors.New("That's not even a token")TokenInvalid     = errors.New("Couldn't handle this token:")
)// 在这里进行细节的配置，需要配合 global 中的内容配合进行操作
// 取出秘钥
func NewJWT() *JWT {return &JWT{[]byte(global.ServerConfig.JWTInfo.SigningKey), //可以设置过期时间}
}// 创建一个token
func (j *JWT) CreateToken(claims models.CustomClaims) (string, error) {token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)return token.SignedString(j.SigningKey)
}// 解析 token
func (j *JWT) ParseToken(tokenString string) (*models.CustomClaims, error) {token, err := jwt.ParseWithClaims(tokenString, &models.CustomClaims{}, func(token *jwt.Token) (i interface{}, e error) {return j.SigningKey, nil})if err != nil {if ve, ok := err.(*jwt.ValidationError); ok {if ve.Errors&jwt.ValidationErrorMalformed != 0 {return nil, TokenMalformed} else if ve.Errors&jwt.ValidationErrorExpired != 0 {// Token is expiredreturn nil, TokenExpired} else if ve.Errors&jwt.ValidationErrorNotValidYet != 0 {return nil, TokenNotValidYet} else {return nil, TokenInvalid}}}if token != nil {if claims, ok := token.Claims.(*models.CustomClaims); ok && token.Valid {return claims, nil}return nil, TokenInvalid} else {return nil, TokenInvalid}}// 更新token
func (j *JWT) RefreshToken(tokenString string) (string, error) {jwt.TimeFunc = func() time.Time {return time.Unix(0, 0)}token, err := jwt.ParseWithClaims(tokenString, &models.CustomClaims{}, func(token *jwt.Token) (interface{}, error) {return j.SigningKey, nil})if err != nil {return "", err}if claims, ok := token.Claims.(*models.CustomClaims); ok && token.Valid {jwt.TimeFunc = time.Nowclaims.StandardClaims.ExpiresAt = time.Now().Add(1 * time.Hour).Unix()return j.CreateToken(*claims)}return "", TokenInvalid
}

注意引用文件，将引用文件进行配置：配置 config/config.go

在 config.go 中添加如下内容：

type JWTConfig struct {SigningKey string `mapstructure:"key"`
}

在配置文件中添加配置 config-debug.yaml

这个 key 是随机生成的

jwt:key: "$Wv?[cf@v0]i.A5v:08V"

继续修改config.go，令其可以取出key：

type ServerConfig struct {Name        string        `mapstructure:"name"`Port        int32         `mapstructure:"port"`UserSrvInfo UserSrvConfig `mapstructure:"user_srv"`JWTInfo     JWTConfig     `mapstructure:"jwt"`
}type JWTConfig struct {SigningKey string `mapstructure:"key"`
}

改到这种程度，jwt.go 应该就不会报错了，

下一步是生成TOKEN，进业务逻辑：api/user.go

修改登陆成功后的代码，这里直接提供登录的所有代码，以便翻看：

// 用户登录模块
func PassWordLogin(c *gin.Context) {// 绑定请求参数passwordLoginForm := forms.PassWordLoginForm{}if err := c.ShouldBind(&passwordLoginForm); err != nil {HandleValidatorError(c, err)}// 将请求参数转发给 rpc 服务器//ip := "127.0.0.1"//port := 50051// 拨号连接用户 GRPC 服务userConn, err := grpc.Dial(fmt.Sprintf("%s:%d", global.ServerConfig.UserSrvInfo.Host, global.ServerConfig.UserSrvInfo.Port), grpc.WithInsecure())if err != nil {zap.L().Error("[PassWordLogin] 连接 【用户服务失败】",zap.String("msg", err.Error()))}// 生成 grpc 的 client 并调用接口userSrvClient := proto.NewUserClient(userConn)if rsp, err := userSrvClient.GetUserByMobile(context.Background(), &proto.MobileRequest{Mobile: passwordLoginForm.Mobile,}); err != nil {if e, ok := status.FromError(err); ok {switch e.Code() {case codes.NotFound:c.JSON(http.StatusBadRequest, map[string]string{"mobile": "用户不存在",})default:c.JSON(http.StatusInternalServerError, map[string]string{"mobile": "登录失败",})}return}} else {// TODO 这里只能验证找到了用户，后面继续验证密码是否正确的功能// 这里是调用的 Srv 中的服务进行登录逻辑的判断，传入的 第二个参数是需要的参数，这个参数的对应取值在之前已经取到// 这里仅仅演示简单的登录密码验证机制，完全的登录逻辑在之后展示if passRsp, passErr := userSrvClient.CheckPassWord(context.Background(), &proto.PassWordCheckInfo{Password:          passwordLoginForm.PassWord,EncryptedPassword: rsp.Password,}); passErr != nil {c.JSON(http.StatusInternalServerError, map[string]string{"msg": "登录失败",})} else {// 这里代表未发生其他错误，继续验证密码if passRsp.Success {// 登录成功，生成TOKENj := middlewares.NewJWT() // 获得签名// 构建传递的信息以及签名信息claims := models.CustomClaims{ID:          uint(rsp.Id),NickName:    rsp.NickName,AuthorityID: uint(rsp.Role),StandardClaims: jwt.StandardClaims{ // 签名相关信息NotBefore: time.Now().Unix(),               // 签名生效时间：现在ExpiresAt: time.Now().Unix() + 60*60*24*30, // 签名过期时间，从现在开始一个月Issuer:    "BaiLu",                         // 签名 对象 （公司）},}// 真正创建 TOKEN：token, err := j.CreateToken(claims)if err != nil {c.JSON(http.StatusInternalServerError, gin.H{"msg": "TOKEN生成失败",})return}c.JSON(http.StatusOK, gin.H{"id":         rsp.Id,"nick_name":  rsp.NickName,"token":      token,"expired_at": (time.Now().Unix() + 60*60*24*30) * 1000,})} else {// 这里代表登录密码验证错误c.JSON(http.StatusBadRequest, map[string]string{"msg": "密码错误",})}}}}

修改完成之后，模拟请求就可以生成 TOKEN 了，下面是一个 TOKEN 实例：

eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJJRCI6MSwiTmlja05hbWUiOiJIdVRhbyIsIkF1dGhvcml0eUlEIjoxLCJleHAiOjE3MjA2ODgzNDksImlzcyI6IkJhaUx1IiwibmJmIjoxNzE4MDk2MzQ5fQ.xVs31ToQJ5TjgjXn2omBXM2i-bw5sVC5Mf6NtN8bw7Y

其会被解析：

{"alg": "HS256","typ": "JWT"
}{"ID": 1,"NickName": "HuTao","AuthorityID": 1,"exp": 1720688349,"iss": "BaiLu","nbf": 1718096349
}

但是其签名校验无法通过，因为其没有对应的签名信息，也就是之前生成的这个：

$Wv?[cf@v0]i.A5v:08V

配置 GetUserList 模块登陆才能访问

在 JWT.go 中 已经配置好了 JWTAuth 的拦截器，只需要在 Router 的位置进行配置即可

将路由中添加拦截器：

	// 这样就需要 /user/list 才可以进行访问了UserRouter := Router.Group("user"){// 在这里添加拦截器的作用响应位置UserRouter.GET("list", middlewares.JWTAuth(), api.GetUserList)UserRouter.POST("pwd_login", api.PassWordLogin)}

注意拦截器中用到了两个方法c.Set(“userID”, userID)，举例这样就可以在 c 中的后续使用 c.Get 再获取刚刚 Set 的信息，对于传递信息十分有效第二个方法是 c.Next() 这个方法会令 gin 调用执行链中的下一个拦截器或者下一个方法等

此时再发送 GetUserList 的请求就会直接返回请登录

之后只要将合法的 token 写入 header 的 x-token 参数中，就可以正确匹配

之后由于我们在 jwt token写入的末尾 有 c.Set(“claims”, claims) 将对应的对象存入上下文，故而我们可以：

采用如下方式调用到我们需要的数据

	// 测试 ID 是否可以取到claims, _ := ctx.Get("claims")currentUser := claims.(*models.CustomClaims)zap.S().Infof("访问用户: %d", currentUser.ID)

鉴权配置，配置 GetUserList 接口只有管理员才能访问

在 middlewares 目录下创建 admin.go 文件 用于鉴权：

package middlewaresimport ("github.com/gin-gonic/gin""mxshop-api/user-web/models""net/http"
)// 手写一个拦截器
func IsAdminAuth() gin.HandlerFunc {return func(ctx *gin.Context) {claims, _ := ctx.Get("claims")// 拿到对应的角色currentUser := claims.(*models.CustomClaims)// 若用户权限为 2 ，视为管理员if currentUser.AuthorityID == 2 {ctx.Next()} else {ctx.JSON(http.StatusForbidden, gin.H{"msg": "权限不足，禁止访问",})ctx.Abort()return}}
}

在router.go 中进行修改：

UserRouter.GET("list", middlewares.JWTAuth(), middlewares.IsAdminAuth(), api.GetUserList)

跨域的处理

假设我们生成一个前端页面，这个页面可以发送 GetUserList 请求：

<!DOCTYPE html>
<html lang="en">
<head><meta charset="utf-8"><title>TTTTTTT</title><script src="http://libs.baidu.com/jquery/2.0.0/jquery.min.js"></script>
</head>
<body><button type="button" id="query">请求发送Biu~</button><div id="content" style="background-color: aquamarine;width: 300px;height: 500px"></div>
</body>
<script type="text/javascript">$("#query").click(function() {$.ajax({url: "http://127.0.0.1:8021/u/v1/user/list",dataType: "json",type: "get",beforeSend: function(request) {request.setRequestHeader("x-token", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJJRCI6MSwiTmlja05hbWUiOiJIdVRhbyIsIkF1dGhvcml0eUlEIjoxLCJleHAiOjE3MjA4MDAyNDMsImlzcyI6IkJhaUx1IiwibmJmIjoxNzE4MjA4MjQzfQ.KuzJSLNbZrk37u4BgT6s-9JtbW8JwA23QB2ktGShRfY")},success: function (result) {console.log(result.data)$("#content").text(result.data)},error: function (data) {alert("请求出错")}});});
</script>
</html>

这里我们发送请求就会出现 OPTIONS 预检报错，这个问题一般由发起复杂请求引起，复杂请求一般包括三种情况：

• 请求方法为：PUT / DELETE
• 请求头中带有：application/json
• 有自定义添加的请求头，例如上面这样

并且我们的请求地址与响应地址的 ip、端口 不同时就会引起跨域：

这时，我们就需要进行跨域处理：

在middlewares 目录下创建 cors.go 文件：

package middlewaresimport ("github.com/gin-gonic/gin""net/http"
)func Cors() gin.HandlerFunc {return func(c *gin.Context) {// 获取请求方法method := c.Request.Method// 允许所有来源，也就是传统意义上的跨域c.Header("Access-Control-Allow-Origin", "*")// 允许头中带有这些c.Header("Access-Control-Allow-Headers", "Content-Type, AccessToken, X-CSRF-TOKEN, Authorization, Token, x-token")// 允许这些方法c.Header("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, OPTIONS, PATCH")// 允许客户端访问的哪些请求头部c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")// 允许携带Cookiec.Header("Access-Control-Allow-Credentials", "true")if method == "OPTIONS" {c.AbortWithStatus(http.StatusNoContent)}}
}

将配置的拦截器放在 Initialize/router.go 下，令所有请求都可以穿过跨域

func Routers() *gin.Engine {Router := gin.Default()// 配置跨域的拦截器Router.Use(middlewares.Cors())ApiGroup := Router.Group("/u/v1")router2.InitUserRouter(ApiGroup)return Router
}

这样子跨域就处理完成了