|
@@ -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
|
|
|
|
+}
|