优化项目整合内容

modules/config_manager.py

@@ -366,6 +366,50 @@ def delete_redis_query_history(history_id):
     finally:
         conn.close()
 
+def batch_delete_redis_query_history(history_ids):
+    """批量删除Redis查询历史记录"""
+    if not history_ids:
+        return {'success': True, 'message': '没有要删除的记录', 'deleted_count': 0}
+    
+    if not ensure_database():
+        logger.error("数据库初始化失败")
+        return {'success': False, 'error': '数据库初始化失败', 'deleted_count': 0}
+        
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    try:
+        # 构建IN子句的占位符
+        placeholders = ','.join(['?' for _ in history_ids])
+        sql = f'DELETE FROM redis_query_history WHERE id IN ({placeholders})'
+        
+        cursor.execute(sql, history_ids)
+        conn.commit()
+        deleted_count = cursor.rowcount
+        
+        if deleted_count > 0:
+            logger.info(f"成功批量删除 {deleted_count} 条Redis查询历史记录: {history_ids}")
+            return {
+                'success': True, 
+                'message': f'成功删除 {deleted_count} 条记录',
+                'deleted_count': deleted_count
+            }
+        else:
+            return {
+                'success': False, 
+                'error': '没有找到要删除的记录',
+                'deleted_count': 0
+            }
+    except Exception as e:
+        logger.error(f"批量删除Redis查询历史记录失败: {e}")
+        return {
+            'success': False, 
+            'error': f'删除失败: {str(e)}',
+            'deleted_count': 0
+        }
+    finally:
+        conn.close()
+
 def parse_redis_config_from_yaml(yaml_text):
     """从YAML格式文本解析Redis配置"""
     try:
@@ -698,5 +742,49 @@ def delete_query_history(history_id):
     except Exception as e:
         logger.error(f"删除查询历史记录失败: {e}")
         return False
+    finally:
+        conn.close()
+
+def batch_delete_query_history(history_ids):
+    """批量删除Cassandra查询历史记录"""
+    if not history_ids:
+        return {'success': True, 'message': '没有要删除的记录', 'deleted_count': 0}
+    
+    if not ensure_database():
+        logger.error("数据库初始化失败")
+        return {'success': False, 'error': '数据库初始化失败', 'deleted_count': 0}
+        
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    try:
+        # 构建IN子句的占位符
+        placeholders = ','.join(['?' for _ in history_ids])
+        sql = f'DELETE FROM query_history WHERE id IN ({placeholders})'
+        
+        cursor.execute(sql, history_ids)
+        conn.commit()
+        deleted_count = cursor.rowcount
+        
+        if deleted_count > 0:
+            logger.info(f"成功批量删除 {deleted_count} 条Cassandra查询历史记录: {history_ids}")
+            return {
+                'success': True, 
+                'message': f'成功删除 {deleted_count} 条记录',
+                'deleted_count': deleted_count
+            }
+        else:
+            return {
+                'success': False, 
+                'error': '没有找到要删除的记录',
+                'deleted_count': 0
+            }
+    except Exception as e:
+        logger.error(f"批量删除Cassandra查询历史记录失败: {e}")
+        return {
+            'success': False, 
+            'error': f'删除失败: {str(e)}',
+            'deleted_count': 0
+        }
     finally:
         conn.close()