Add files via upload

internal/app/app.go

@@ -233,6 +233,7 @@ func setupRoutes(
 		// 监控
 		protected.GET("/monitor", monitorHandler.Monitor)
 		protected.GET("/monitor/execution/:id", monitorHandler.GetExecution)
+		protected.DELETE("/monitor/execution/:id", monitorHandler.DeleteExecution)
 		protected.GET("/monitor/stats", monitorHandler.GetStats)
 
 		// 配置管理

internal/database/monitor.go

@@ -232,6 +232,17 @@ func (db *DB) GetToolExecution(id string) (*mcp.ToolExecution, error) {
 	return &exec, nil
 }
 
+// DeleteToolExecution 删除工具执行记录
+func (db *DB) DeleteToolExecution(id string) error {
+	query := `DELETE FROM tool_executions WHERE id = ?`
+	_, err := db.Exec(query, id)
+	if err != nil {
+		db.logger.Error("删除工具执行记录失败", zap.Error(err), zap.String("executionId", id))
+		return err
+	}
+	return nil
+}
+
 // SaveToolStats 保存工具统计信息
 func (db *DB) SaveToolStats(toolName string, stats *mcp.ToolStats) error {
 	var lastCallTime sql.NullTime
@@ -332,3 +343,46 @@ func (db *DB) UpdateToolStats(toolName string, totalCalls, successCalls, failedC
 
 	return nil
 }
+
+// DecreaseToolStats 减少工具统计信息（用于删除执行记录时）
+// 如果统计信息变为0，则删除该统计记录
+func (db *DB) DecreaseToolStats(toolName string, totalCalls, successCalls, failedCalls int) error {
+	// 先更新统计信息
+	query := `
+		UPDATE tool_stats SET
+			total_calls = CASE WHEN total_calls - ? < 0 THEN 0 ELSE total_calls - ? END,
+			success_calls = CASE WHEN success_calls - ? < 0 THEN 0 ELSE success_calls - ? END,
+			failed_calls = CASE WHEN failed_calls - ? < 0 THEN 0 ELSE failed_calls - ? END,
+			updated_at = ?
+		WHERE tool_name = ?
+	`
+
+	_, err := db.Exec(query, totalCalls, totalCalls, successCalls, successCalls, failedCalls, failedCalls, time.Now(), toolName)
+	if err != nil {
+		db.logger.Error("减少工具统计信息失败", zap.Error(err), zap.String("toolName", toolName))
+		return err
+	}
+
+	// 检查更新后的 total_calls 是否为 0，如果是则删除该统计记录
+	checkQuery := `SELECT total_calls FROM tool_stats WHERE tool_name = ?`
+	var newTotalCalls int
+	err = db.QueryRow(checkQuery, toolName).Scan(&newTotalCalls)
+	if err != nil {
+		// 如果查询失败（记录不存在），直接返回
+		return nil
+	}
+
+	// 如果 total_calls 为 0，删除该统计记录
+	if newTotalCalls == 0 {
+		deleteQuery := `DELETE FROM tool_stats WHERE tool_name = ?`
+		_, err = db.Exec(deleteQuery, toolName)
+		if err != nil {
+			db.logger.Warn("删除零统计记录失败", zap.Error(err), zap.String("toolName", toolName))
+			// 不返回错误，因为主要操作（更新统计）已成功
+		} else {
+			db.logger.Info("已删除零统计记录", zap.String("toolName", toolName))
+		}
+	}
+
+	return nil
+}

internal/handler/monitor.go

@@ -220,4 +220,59 @@ func (h *MonitorHandler) GetStats(c *gin.Context) {
 	c.JSON(http.StatusOK, stats)
 }
 
+// DeleteExecution 删除执行记录
+func (h *MonitorHandler) DeleteExecution(c *gin.Context) {
+	id := c.Param("id")
+	if id == "" {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "执行记录ID不能为空"})
+		return
+	}
+
+	// 如果使用数据库，先获取执行记录信息，然后删除并更新统计
+	if h.db != nil {
+		// 先获取执行记录信息（用于更新统计）
+		exec, err := h.db.GetToolExecution(id)
+		if err != nil {
+			// 如果找不到记录，可能已经被删除，直接返回成功
+			h.logger.Warn("执行记录不存在，可能已被删除", zap.String("executionId", id), zap.Error(err))
+			c.JSON(http.StatusOK, gin.H{"message": "执行记录不存在或已被删除"})
+			return
+		}
+
+		// 删除执行记录
+		err = h.db.DeleteToolExecution(id)
+		if err != nil {
+			h.logger.Error("删除执行记录失败", zap.Error(err), zap.String("executionId", id))
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "删除执行记录失败: " + err.Error()})
+			return
+		}
+
+		// 更新统计信息（减少相应的计数）
+		totalCalls := 1
+		successCalls := 0
+		failedCalls := 0
+		if exec.Status == "failed" {
+			failedCalls = 1
+		} else if exec.Status == "completed" {
+			successCalls = 1
+		}
+
+		if exec.ToolName != "" {
+			if err := h.db.DecreaseToolStats(exec.ToolName, totalCalls, successCalls, failedCalls); err != nil {
+				h.logger.Warn("更新统计信息失败", zap.Error(err), zap.String("toolName", exec.ToolName))
+				// 不返回错误，因为记录已经删除成功
+			}
+		}
+
+		h.logger.Info("执行记录已从数据库删除", zap.String("executionId", id), zap.String("toolName", exec.ToolName))
+		c.JSON(http.StatusOK, gin.H{"message": "执行记录已删除"})
+		return
+	}
+
+	// 如果不使用数据库，尝试从内存中删除（内部MCP服务器）
+	// 注意：内存中的记录可能已经被清理，所以这里只记录日志
+	h.logger.Info("尝试删除内存中的执行记录", zap.String("executionId", id))
+	c.JSON(http.StatusOK, gin.H{"message": "执行记录已删除（如果存在）"})
+}
+