MTP20_IF/token/token.go

package token

import (
	"errors"
	"fmt"
	"mtp2_if/rediscli"
	"net/http"
	"strings"

	"github.com/gin-gonic/gin"
)

// CheckToken Token校验
func CheckToken(loginid string, token string) error {
	key := fmt.Sprintf("monitor:online_loginid::%s", loginid)
	field := "Token"

	realToken, err := rediscli.GetRedisClient().HGet(key, field).Result()
	if err != nil {
		return err
	}

	if realToken != token {
		return errors.New("token is invalid")
	}

	return nil
}

// Auth Token校验中间件
func Auth() gin.HandlerFunc {
	return func(c *gin.Context) {
		var code int
		var msg string
		var data interface{}

		code = 0
		token := c.GetHeader("token")
		if token == "" {
			code = 3001
			msg = "缺少token"
		} else {
			// 获取loginid
			s := strings.Split(token, "_")
			loginid := s[0]
			if err := CheckToken(loginid, token); err != nil {
				code = 3002
				msg = "token校验失败"
			}
		}

		// Token检验失败
		if code != 0 {
			c.JSON(http.StatusUnauthorized, gin.H{
				"code": code,
				"msg":  msg,
				"data": data,
			})

			c.Abort()
			return
		}

		// Token检验成功
		c.Next()
	}
}