Add files via upload

internal/database/monitor.go

@@ -70,10 +70,15 @@ func (db *DB) SaveToolExecution(exec *mcp.ToolExecution) error {
 }
 
 // CountToolExecutions 统计工具执行记录总数
-func (db *DB) CountToolExecutions() (int, error) {
+func (db *DB) CountToolExecutions(status string) (int, error) {
 	query := `SELECT COUNT(*) FROM tool_executions`
+	args := []interface{}{}
+	if status != "" {
+		query += ` WHERE status = ?`
+		args = append(args, status)
+	}
 	var count int
-	err := db.QueryRow(query).Scan(&count)
+	err := db.QueryRow(query, args...).Scan(&count)
 	if err != nil {
 		return 0, err
 	}
@@ -82,13 +87,14 @@ func (db *DB) CountToolExecutions() (int, error) {
 
 // LoadToolExecutions 加载所有工具执行记录（支持分页）
 func (db *DB) LoadToolExecutions() ([]*mcp.ToolExecution, error) {
-	return db.LoadToolExecutionsWithPagination(0, 1000)
+	return db.LoadToolExecutionsWithPagination(0, 1000, "")
 }
 
 // LoadToolExecutionsWithPagination 分页加载工具执行记录
 // limit: 最大返回记录数，0 表示使用默认值 1000
 // offset: 跳过的记录数，用于分页
-func (db *DB) LoadToolExecutionsWithPagination(offset, limit int) ([]*mcp.ToolExecution, error) {
+// status: 状态筛选，空字符串表示不过滤
+func (db *DB) LoadToolExecutionsWithPagination(offset, limit int, status string) ([]*mcp.ToolExecution, error) {
 	if limit <= 0 {
 		limit = 1000 // 默认限制
 	}
@@ -99,11 +105,16 @@ func (db *DB) LoadToolExecutionsWithPagination(offset, limit int) ([]*mcp.ToolEx
 	query := `
 		SELECT id, tool_name, arguments, status, result, error, start_time, end_time, duration_ms
 		FROM tool_executions
-		ORDER BY start_time DESC
-		LIMIT ? OFFSET ?
 	`
+	args := []interface{}{}
+	if status != "" {
+		query += ` WHERE status = ?`
+		args = append(args, status)
+	}
+	query += ` ORDER BY start_time DESC LIMIT ? OFFSET ?`
+	args = append(args, limit, offset)
 
-	rows, err := db.Query(query, limit, offset)
+	rows, err := db.Query(query, args...)
 	if err != nil {
 		return nil, err
 	}

internal/handler/monitor.go

@@ -64,7 +64,10 @@ func (h *MonitorHandler) Monitor(c *gin.Context) {
 		}
 	}
 
-	executions, total := h.loadExecutionsWithPagination(page, pageSize)
+	// 解析状态筛选参数
+	status := c.Query("status")
+
+	executions, total := h.loadExecutionsWithPagination(page, pageSize, status)
 	stats := h.loadStats()
 
 	totalPages := (total + pageSize - 1) / pageSize
@@ -84,13 +87,23 @@ func (h *MonitorHandler) Monitor(c *gin.Context) {
 }
 
 func (h *MonitorHandler) loadExecutions() []*mcp.ToolExecution {
-	executions, _ := h.loadExecutionsWithPagination(1, 1000)
+	executions, _ := h.loadExecutionsWithPagination(1, 1000, "")
 	return executions
 }
 
-func (h *MonitorHandler) loadExecutionsWithPagination(page, pageSize int) ([]*mcp.ToolExecution, int) {
+func (h *MonitorHandler) loadExecutionsWithPagination(page, pageSize int, status string) ([]*mcp.ToolExecution, int) {
 	if h.db == nil {
 		allExecutions := h.mcpServer.GetAllExecutions()
+		// 如果指定了状态筛选，先进行筛选
+		if status != "" {
+			filtered := make([]*mcp.ToolExecution, 0)
+			for _, exec := range allExecutions {
+				if exec.Status == status {
+					filtered = append(filtered, exec)
+				}
+			}
+			allExecutions = filtered
+		}
 		total := len(allExecutions)
 		offset := (page - 1) * pageSize
 		end := offset + pageSize
@@ -104,10 +117,20 @@ func (h *MonitorHandler) loadExecutionsWithPagination(page, pageSize int) ([]*mc
 	}
 
 	offset := (page - 1) * pageSize
-	executions, err := h.db.LoadToolExecutionsWithPagination(offset, pageSize)
+	executions, err := h.db.LoadToolExecutionsWithPagination(offset, pageSize, status)
 	if err != nil {
 		h.logger.Warn("从数据库加载执行记录失败，回退到内存数据", zap.Error(err))
 		allExecutions := h.mcpServer.GetAllExecutions()
+		// 如果指定了状态筛选，先进行筛选
+		if status != "" {
+			filtered := make([]*mcp.ToolExecution, 0)
+			for _, exec := range allExecutions {
+				if exec.Status == status {
+					filtered = append(filtered, exec)
+				}
+			}
+			allExecutions = filtered
+		}
 		total := len(allExecutions)
 		offset := (page - 1) * pageSize
 		end := offset + pageSize
@@ -120,8 +143,8 @@ func (h *MonitorHandler) loadExecutionsWithPagination(page, pageSize int) ([]*mc
 		return allExecutions[offset:end], total
 	}
 
-	// 获取总数
-	total, err := h.db.CountToolExecutions()
+	// 获取总数（考虑状态筛选）
+	total, err := h.db.CountToolExecutions(status)
 	if err != nil {
 		h.logger.Warn("获取执行记录总数失败", zap.Error(err))
 		// 回退：使用已加载的记录数估算