light-apiengine-feature/middleware/login_ware.go

package middleware

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

func LoginWare() gin.HandlerFunc {
	return func(c *gin.Context) {
		//验证登陆
		token := c.GetHeader("token")
		if token == "" {
			c.AbortWithStatusJSON(401, sysmodels.SysReturn{401, "token is empty", ""})
			c.Redirect(302, "/login")
			return
		}
		tk := utils.GetGlobalTokenStore().Get(token)
		if tk == nil {
			c.AbortWithStatusJSON(401, sysmodels.SysReturn{401, "token Invalid", ""})
			c.Redirect(302, "/login")
			return
		}

		wfendpoint := config.AppConfig.GetKey("wfendpoint")
		wfclient.Instance().Init(wfendpoint, tk.UserId, tk.LoginID, tk.AccessToken, tk.Domain)

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

	}
}