Add files via upload

internal/handler/monitor.go

@@ -2,6 +2,7 @@ package handler
 
 import (
 	"net/http"
+	"strconv"
 	"time"
 
 	"cyberstrike-ai/internal/database"
@@ -34,31 +35,96 @@ type MonitorResponse struct {
 	Executions []*mcp.ToolExecution      `json:"executions"`
 	Stats      map[string]*mcp.ToolStats `json:"stats"`
 	Timestamp  time.Time                  `json:"timestamp"`
+	Total      int                        `json:"total,omitempty"`
+	Page       int                        `json:"page,omitempty"`
+	PageSize   int                        `json:"page_size,omitempty"`
+	TotalPages int                        `json:"total_pages,omitempty"`
 }
 
 // Monitor 获取监控信息
 func (h *MonitorHandler) Monitor(c *gin.Context) {
-	executions := h.loadExecutions()
+	// 解析分页参数
+	page := 1
+	pageSize := 20
+	if pageStr := c.Query("page"); pageStr != "" {
+		if p, err := strconv.Atoi(pageStr); err == nil && p > 0 {
+			page = p
+		}
+	}
+	if pageSizeStr := c.Query("page_size"); pageSizeStr != "" {
+		if ps, err := strconv.Atoi(pageSizeStr); err == nil && ps > 0 && ps <= 100 {
+			pageSize = ps
+		}
+	}
+
+	executions, total := h.loadExecutionsWithPagination(page, pageSize)
 	stats := h.loadStats()
 
+	totalPages := (total + pageSize - 1) / pageSize
+	if totalPages == 0 {
+		totalPages = 1
+	}
+
 	c.JSON(http.StatusOK, MonitorResponse{
 		Executions: executions,
 		Stats:      stats,
 		Timestamp:  time.Now(),
+		Total:      total,
+		Page:       page,
+		PageSize:   pageSize,
+		TotalPages: totalPages,
 	})
 }
 
 func (h *MonitorHandler) loadExecutions() []*mcp.ToolExecution {
+	executions, _ := h.loadExecutionsWithPagination(1, 1000)
+	return executions
+}
+
+func (h *MonitorHandler) loadExecutionsWithPagination(page, pageSize int) ([]*mcp.ToolExecution, int) {
 	if h.db == nil {
-		return h.mcpServer.GetAllExecutions()
+		allExecutions := h.mcpServer.GetAllExecutions()
+		total := len(allExecutions)
+		offset := (page - 1) * pageSize
+		end := offset + pageSize
+		if end > total {
+			end = total
+		}
+		if offset >= total {
+			return []*mcp.ToolExecution{}, total
+		}
+		return allExecutions[offset:end], total
 	}
 
-	executions, err := h.db.LoadToolExecutions()
+	offset := (page - 1) * pageSize
+	executions, err := h.db.LoadToolExecutionsWithPagination(offset, pageSize)
 	if err != nil {
 		h.logger.Warn("从数据库加载执行记录失败，回退到内存数据", zap.Error(err))
-		return h.mcpServer.GetAllExecutions()
+		allExecutions := h.mcpServer.GetAllExecutions()
+		total := len(allExecutions)
+		offset := (page - 1) * pageSize
+		end := offset + pageSize
+		if end > total {
+			end = total
+		}
+		if offset >= total {
+			return []*mcp.ToolExecution{}, total
+		}
+		return allExecutions[offset:end], total
 	}
-	return executions
+
+	// 获取总数
+	total, err := h.db.CountToolExecutions()
+	if err != nil {
+		h.logger.Warn("获取执行记录总数失败", zap.Error(err))
+		// 回退：使用已加载的记录数估算
+		total = offset + len(executions)
+		if len(executions) == pageSize {
+			total = offset + len(executions) + 1
+		}
+	}
+
+	return executions, total
 }
 
 func (h *MonitorHandler) loadStats() map[string]*mcp.ToolStats {