// 設置token過期時間
const TokenExpireDuration = time.Hour * 2
var mySecret = []byte("努力學習go語言")
// 這里可以根據(jù)自己的業(yè)務做出更改
type MyClaims struct {
	UserID   int64  `json:"user_id"`
	UserName string `json:"user_name"`
	jwt.StandardClaims
}
// GenToken 生成JWT
func GenToken(userId int64, username string) (string, error) {
	// 創(chuàng)建我們自己聲明的數(shù)據(jù)
	c := MyClaims{
		userId,
		username, // 定義字段
		jwt.StandardClaims{
			IssuedAt: time.Now().Add(TokenExpireDuration).Unix(), // 過期時間
			Issuer:   "三杯溫開水",                                 // 簽發(fā)人
		},
	}
	// 使用指定的簽名方法創(chuàng)建簽名對象
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, c)
	// 返回token
	return token.SignedString(mySecret)
}
// ParseToken 解析 JWT
func ParseToken(tokenString string) (*MyClaims, error) {
	mc := new(MyClaims)
	// 解析token
	token, err := jwt.ParseWithClaims(tokenString, mc, func(token *jwt.Token) (i interface{}, err error) {
		return mySecret, nil
	})
	if err != nil {
		return nil, err
	}
	if token.Valid { // 校驗token
		return mc, nil
	}
	return nil, errors.New("invalid token")
}

// JWTAuthMiddleware 基于JWT的認證中間件
func JWTAuthMiddleware() func(c *gin.Context) {
	return func(c *gin.Context) {
		// 這里根據(jù)自己的業(yè)務進行更改
		authHeader := c.Request.Header.Get("Authorization")
		if authHeader == "" {
			controller.ResponseError(c, controller.CodeNeedLogin)
			c.Abort()
			return
		}
		// 按空格分割(基于Bearer開頭的)，如果不是Bearer開頭的下面代碼可以進行省略
		// ------ start ------
		parts := strings.SplitN(authHeader, " ", 2)
		if !(len(parts) == 2 && parts[0] == "Bearer") {
			controller.ResponseError(c, controller.CodeInvalidToken)
			c.Abort()
			return
		}
		// ------ end -------
		// parts[1]是獲取到的tokenString，我們使用之前定義好的解析JWT的函數(shù)來解析它
		mc, err := jwt.ParseToken(parts[1])
		if err != nil {
			controller.ResponseError(c, controller.CodeInvalidToken)
			c.Abort()
			return
		}
		// 將當前請求的username信息保存到請求的上下文c上
		c.Set(controller.CtxUserIdKey, mc.UserID)
		c.Next() // 后續(xù)的處理函數(shù)可以用過c.Get(CtxUserIdKey)來獲取當前請求的用戶信息
	}
}

r.POST("/ping", JWTAuthMiddleware(), func(c *gin.Context) {
		c.String(http.StatusOK, "ok!")
	})