package middleware import ( "net/http" "strings" "NetworkAuth/services" "github.com/gin-gonic/gin" ) // MaintenanceMiddleware 维护模式中间件 // 当开启维护模式时，拦截非白名单请求 func MaintenanceMiddleware() gin.HandlerFunc { return func(c *gin.Context) { // 检查是否开启维护模式 if !services.GetSettingsService().IsMaintenanceMode() { c.Next() return } // 白名单检查（路径前缀匹配） path := c.Request.URL.Path // 1. 允许静态资源 if strings.HasPrefix(path, "/static/") || strings.HasPrefix(path, "/assets/") || path == "/favicon.ico" { c.Next() return } // 2. 允许管理员后台相关接口（以便管理员登录关闭维护模式） // 包括登录页、登录接口、API接口、CSRF Token等 if strings.HasPrefix(path, "/admin") { c.Next() return } // 3. 检查请求类型 // AJAX/JSON 请求返回 503 JSON accept := c.GetHeader("Accept") xrw := strings.ToLower(strings.TrimSpace(c.GetHeader("X-Requested-With"))) if strings.Contains(accept, "application/json") || xrw == "xmlhttprequest" || strings.HasPrefix(path, "/api/") { c.JSON(http.StatusServiceUnavailable, gin.H{ "code": 503, "success": false, "msg": "系统正在维护中，请稍后再试", }) c.Abort() return } // 4. 普通页面请求渲染维护页面 c.Header("Content-Type", "text/html; charset=utf-8") c.Status(http.StatusServiceUnavailable) c.Writer.WriteString(maintenanceHTML) c.Abort() } } // 简单的维护页面 HTML const maintenanceHTML = ` 系统维护中

⚠️

系统维护中

为了提供更好的服务，系统正在进行升级维护。
请稍后访问，给您带来的不便敬请谅解。