添加中间件

internal/middleware/auth.go

@@ -0,0 +1,48 @@
+package middleware
+
+import (
+	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/gogf/gf/v2/os/glog"
+	"server/utility/ecode"
+	"server/utility/jwt"
+	"strings"
+)
+
+// Auth 是用于权限验证的中间件。
+//
+// 它会从请求头中读取 Authorization 字段，验证 JWT Token 的合法性。
+// 如果请求未携带 Token，则默认赋予 "guest" 权限。
+// 如果 Token 存在且合法，则将用户信息（userId、permission、jti）注入到请求上下文中。
+//
+// Token 应采用 Bearer 方案，例如：
+//
+//	Authorization: Bearer <token>
+//
+// 参数:
+//
+//	r *ghttp.Request - 当前的请求对象，由 ghttp 框架自动注入。
+//
+// 行为:
+//   - 若无 Token：设定 permission 为 guest，放行请求。
+//   - 若 Token 格式非法或解析失败：终止请求并返回错误。
+//   - 若 Token 合法：将用户信息写入上下文，继续执行下一个中间件或处理函数。
+func Auth(r *ghttp.Request) {
+	token := r.Header.Get("Authorization")
+	ctx := r.GetCtx()
+	if token == "" {
+		glog.Infof(ctx, "未登录用户访问: %s %s", r.URL.Path, r.Method)
+		r.SetCtxVar("permission", "guest")
+	} else {
+		if !strings.HasPrefix(token, "Bearer ") {
+			Exit(r, ecode.InvalidOperation.Sub("无效的token格式"))
+		}
+		tokenOut, err := jwt.ParseToken(token)
+		if err != nil {
+			Exit(r, err)
+		}
+		r.SetCtxVar("userId", tokenOut.UserId)
+		r.SetCtxVar("permission", tokenOut.Permission)
+		r.SetCtxVar("jti", tokenOut.JTI)
+	}
+	r.Middleware.Next()
+}