light-apiengine/middleware/login_ware.go

package middleware

import (
	sysmodel "git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/utils"
	"github.com/gin-gonic/gin"
)

func LoginWare() gin.HandlerFunc {
	return func(c *gin.Context) {
		//验证登陆
		token := c.GetHeader("token")

		if token == "" {
			c.AbortWithStatusJSON(401, sysmodel.SysReturn{401, "token is empty", ""})
			c.Redirect(302, "/login")
			return
		}

		tk := utils.GetGlobalTokenStore().Get(token)
		if tk == nil {
			c.AbortWithStatusJSON(401, sysmodel.SysReturn{401, "token Invalid", ""})
			c.Redirect(302, "/login")
			return
		}

		c.Set("token", tk)
		c.Set("user_id", tk.UserId)
		c.Set("domain", tk.Domain)
		c.Next()
	}
}