Update.

Author: xOS

cmd/dashboard/controller/controller.go

@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"runtime"
 	"strconv"
 	"strings"
 	"time"
@@ -63,6 +64,7 @@ func corsMiddleware(c *gin.Context) {
 	c.Next()
 }
 
+// ServeWeb 启动Web服务器并配置路由
 func ServeWeb(port uint) *http.Server {
 	gin.SetMode(gin.ReleaseMode)
 
@@ -97,62 +99,155 @@ func ServeWeb(port uint) *http.Server {
 
 	// 添加Recovery中间件，过滤网络连接错误
 	r.Use(gin.CustomRecovery(func(c *gin.Context, recovered interface{}) {
-		if err, ok := recovered.(string); ok {
-			errLower := strings.ToLower(err)
-			if strings.Contains(errLower, "broken pipe") ||
-				strings.Contains(errLower, "connection reset") ||
-				strings.Contains(errLower, "use of closed network connection") {
-				// 静默处理网络连接错误
-				c.AbortWithStatus(http.StatusInternalServerError)
-				return
-			}
-		}
-		// 其他错误正常处理
+		// 检查是否为broken pipe错误
+		if errStr := fmt.Sprintf("%v", recovered); strings.Contains(errStr, "broken pipe") ||
+			strings.Contains(errStr, "connection reset") ||
+			strings.Contains(errStr, "use of closed network connection") {
+			// 静默处理broken pipe错误，不中断其他请求处理
+			c.AbortWithStatus(499) // 使用499表示客户端关闭连接
+			return
+		}
+		
+		// 其他panic正常处理
+		log.Printf("服务器Panic: %v", recovered)
 		c.AbortWithStatus(http.StatusInternalServerError)
 	}))
-
-	if singleton.Conf.Debug {
-		gin.SetMode(gin.DebugMode)
+	
+	// 添加handleBrokenPipe中间件
+	r.Use(handleBrokenPipe)
+	
+	// 添加CORS中间件
+	r.Use(corsMiddleware)
+	
+	// 添加模板渲染保护中间件 - 这是防止前端页面无法访问的关键
+	r.Use(TemplateRenderingSafeguard())
+	
+	// 添加健康检查中间件
+	r.Use(HealthCheckMiddleware())
+	
+	// 添加pprof性能分析
+	if singleton.Conf != nil && singleton.Conf.Debug {
 		pprof.Register(r)
 	}
-	r.Use(natGateway)
-	r.Use(handleBrokenPipe) // 添加broken pipe错误处理中间件
-	r.Use(corsMiddleware)   // 添加CORS中间件处理OPTIONS请求
-	tmpl := template.New("").Funcs(funcMap)
-	var err error
-	// 直接用本地模板目录
-	tmpl, err = tmpl.ParseGlob("resource/template/**/*.html")
-	if err != nil {
-		panic(err)
-	}
-	tmpl = loadThirdPartyTemplates(tmpl)
-	r.SetHTMLTemplate(tmpl)
-	r.Use(mygin.RecordPath)
-	// 直接用本地静态资源目录
-	r.Static("/static", "resource/static")
-
-	routers(r)
-	page404 := func(c *gin.Context) {
-		mygin.ShowErrorPage(c, mygin.ErrInfo{
-			Code:  http.StatusNotFound,
-			Title: "该页面不存在",
-			Msg:   "该页面内容可能已着陆火星",
-			Link:  "/",
-			Btn:   "返回首页",
-		}, true)
-	}
-	r.NoRoute(page404)
-	r.NoMethod(page404)
-
-	srv := &http.Server{
-		Addr:              fmt.Sprintf(":%d", port),
-		ReadHeaderTimeout: time.Second * 5,
-		Handler:           r,
+	
+	// 设置HTML模板和静态资源
+	r.SetFuncMap(template.FuncMap{
+		"tf": func(t time.Time) string {
+			return t.Format("2006-01-02 15:04:05")
+		},
+		"fisher": func(t time.Time) string {
+			return t.Format("2006/01/02 15:04")
+		},
+		"tf2": func(t time.Time) string {
+			return t.In(singleton.Loc).Format("01/02 15:04")
+		},
+		"markdown": func(text string) template.HTML {
+			// 简单地将markdown转为HTML
+			return template.HTML(text)
+		},
+		"tag": func(text string) template.HTML {
+			return template.HTML(`<` + text + `>`)
+		},
+		"trim": func(text string) string {
+			return strings.TrimSpace(text)
+		},
+		"recent": func(t time.Time) bool {
+			return time.Since(t).Seconds() <= 60
+		},
+		"find": func(slice []string, val string) int {
+			for i, item := range slice {
+				if item == val {
+					return i
+				}
+			}
+			return -1
+		},
+		"next": func(pos, length int) int {
+			pos++
+			if pos >= length {
+				pos = 0
+			}
+			return pos
+		},
+		"prev": func(pos, length int) int {
+			pos--
+			if pos < 0 {
+				pos = length - 1
+			}
+			return pos
+		},
+		"fs": func(bytes uint64) string {
+			return bytefmt.ByteSize(bytes)
+		},
+		"tf_rfc3339": func(t time.Time) string {
+			return t.Format(time.RFC3339)
+		},
+		"pb": func(p float64) string {
+			return fmt.Sprintf("%.2f", p*100)
+		},
+		"isPositive": func(a float64) bool {
+			return a > 0
+		},
+		"isNegative": func(a float64) bool {
+			return a < 0
+		},
+		"abs": func(a float64) float64 {
+			if a < 0 {
+				return -a
+			}
+			return a
+		},
+		"roundTo": func(decimals int, num float64) float64 {
+			f := math.Pow10(decimals)
+			// 四舍五入
+			return math.Round(num*f) / f
+		},
+	})
+	
+	// 静态资源
+	r.StaticFS("/static", http.Dir("resource/static"))
+	
+	// 健康检查路由
+	r.GET("/health", HealthCheckHandler)
+	r.GET("/system-health", SystemHealthHandler)
+	
+	// 创建路由分组和控制器
+	commonController := CreateControllers(r)
+	
+	// 创建HTTP服务器
+	server := &http.Server{
+		Addr:    fmt.Sprintf(":%d", port),
+		Handler: r,
 	}
-	return srv
+	
+	return server
 }
 
 func routers(r *gin.Engine) {
+	// 添加健康检查接口
+	r.GET("/health", func(c *gin.Context) {
+		// 简单版本，直接返回200
+		c.JSON(http.StatusOK, gin.H{
+			"status": "ok",
+			"time": time.Now().Format(time.RFC3339),
+		})
+	})
+	
+	// 系统监控端点 (需要权限才能访问)
+	r.GET("/system-health", func(c *gin.Context) {
+		// 返回系统健康信息
+		memStats := runtime.MemStats{}
+		runtime.ReadMemStats(&memStats)
+		
+		c.JSON(http.StatusOK, gin.H{
+			"status": "ok",
+			"goroutines": runtime.NumGoroutine(),
+			"memoryUsageMB": float64(memStats.Alloc) / (1024 * 1024),
+			"time": time.Now().Format(time.RFC3339),
+			"templateHealthy": GlobalTemplateSafeguard.IsHealthy(),
+		})
+	})
+	
 	// 通用页面
 	cp := commonPage{r: r}
 	cp.serve()
@@ -396,6 +491,50 @@ var funcMap = template.FuncMap{
 	},
 }
 
+// CreateControllers 创建并初始化所有控制器
+func CreateControllers(engine *gin.Engine) *commonPage {
+	// 创建通用页面控制器
+	cp := &commonPage{
+		r: engine,
+	}
+	cp.serve()
+
+	// 创建访客页面控制器
+	gp := &guestPage{
+		r: engine,
+	}
+	gp.serve()
+
+	// 创建OAuth2控制器
+	oa := &oauth2controller{
+		r: engine,
+	}
+	oa.serve()
+
+	// 创建会员页面控制器
+	mp := &memberPage{
+		r: engine,
+	}
+	mp.serve()
+
+	// 创建会员API控制器
+	ma := &memberAPI{
+		r: engine,
+	}
+	ma.serve()
+
+	// 创建APIv1控制器
+	api := &apiV1{
+		r: engine,
+	}
+	api.serve()
+
+	// 添加NAT网关中间件
+	engine.NoRoute(natGateway)
+
+	return cp
+}
+
 func natGateway(c *gin.Context) {
 	natConfig := singleton.GetNATConfigByDomain(c.Request.Host)
 	if natConfig == nil {

cmd/dashboard/main.go

@@ -14,8 +14,10 @@ import (
 	flag "github.com/spf13/pflag"
 	"github.com/xos/serverstatus/cmd/dashboard/controller"
 	"github.com/xos/serverstatus/cmd/dashboard/rpc"
+	"github.com/xos/serverstatus/db"
 	"github.com/xos/serverstatus/model"
 	"github.com/xos/serverstatus/proto"
+	"github.com/xos/serverstatus/service"
 	"github.com/xos/serverstatus/service/singleton"
 )
 
@@ -40,6 +42,18 @@ func init() {
 }
 
 func initSystem() {
+	// 初始化模板保护系统
+	controller.InitTemplateMonitoring()
+	log.Println("模板保护系统已初始化")
+	
+	// 启动数据库操作监控
+	db.GlobalDBManager.StartMonitoring()
+	log.Println("数据库操作监控已启动")
+	
+	// 启动健康监控器
+	service.GlobalHealthMonitor.Start()
+	log.Println("系统健康监控器已启动")
+	
 	// 启动 singleton 包下的所有服务
 	singleton.LoadSingleton()
 
@@ -51,7 +65,12 @@ func initSystem() {
 		singleton.SyncAllServerTrafficFromDB()
 	} else {
 		log.Println("使用BadgerDB，执行BadgerDB监控历史清理...")
+		
+		// 使用数据库操作管理器包装BadgerDB操作，避免与前端请求争用资源
+		done := db.GlobalDBManager.StartOperation("清理监控历史")
 		count, err := singleton.CleanMonitorHistory()
+		done() // 标记操作完成
+		
 		if err != nil {
 			log.Printf("BadgerDB监控历史清理失败: %v", err)
 		} else {
@@ -230,6 +249,18 @@ func main() {
 			// 停止内存监控器，防止它继续调用limitDataSize()造成锁竞争
 			singleton.StopMemoryMonitor()
 			log.Println("内存监控器已停止")
+			
+			// 停止健康监控器
+			service.GlobalHealthMonitor.Stop()
+			log.Println("健康监控器已停止")
+			
+			// 等待数据库操作完成
+			dbOps := db.GlobalDBManager.GetRunningOperations()
+			if len(dbOps) > 0 {
+				log.Printf("等待 %d 个数据库操作完成...", len(dbOps))
+				// 给数据库操作一些时间完成
+				time.Sleep(3 * time.Second)
+			}
 
 			// 停止定时任务
 			if singleton.Cron != nil {