浏览代码

Initial commit

duxin 1 年之前
父节点
当前提交
6b9cb5a7fa
共有 1 个文件被更改,包括 17 次插入0 次删除
  1. 17 0
      gtoken/gtoken_group.go

+ 17 - 0
gtoken/gtoken_group.go

@@ -3,6 +3,7 @@ package gtoken
 import (
 import (
 	"context"
 	"context"
 	"errors"
 	"errors"
+
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
 	"github.com/gogf/gf/v2/net/ghttp"
 	"github.com/gogf/gf/v2/text/gstr"
 	"github.com/gogf/gf/v2/text/gstr"
@@ -52,3 +53,19 @@ func registerFunc(ctx context.Context, group *ghttp.RouterGroup, pattern string,
 		group.ALL(pattern, object)
 		group.ALL(pattern, object)
 	}
 	}
 }
 }
+
+// AuthMiddleware 绑定登录状态校验
+func (m *GfToken) AuthMiddleware(group *ghttp.RouterGroup) bool {
+	if !m.InitConfig() {
+		return false
+	}
+	// 设置为Group模式
+	m.MiddlewareType = MiddlewareTypeGroup
+	// 缓存模式
+	if m.CacheMode > CacheModeRedis {
+		g.Log().Error(context.Background(), "[GToken]CacheMode set error")
+		return false
+	}
+	group.Middleware(m.authMiddleware)
+	return true
+}