mcp-dbutils 0.7.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

mcp_dbutils/stats.py

@@ -2,8 +2,9 @@
 
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Optional
+from typing import Optional, List, Tuple
 import sys
+import statistics
 
 @dataclass
 class ResourceStats:
@@ -24,11 +25,23 @@ class ResourceStats:
 
     # Resource stats
     estimated_memory: int = 0
+    
+    # Performance monitoring
+    query_durations: List[float] = None  # 查询执行时间列表 (秒)
+    query_types: dict[str, int] = None   # 查询类型统计 (SELECT, EXPLAIN等)
+    slow_queries: List[Tuple[str, float]] = None  # 慢查询记录 (SQL, 时间)
+    peak_memory: int = 0  # 峰值内存使用
 
     def __post_init__(self):
         """Initialize mutable defaults"""
         if self.error_types is None:
             self.error_types = {}
+        if self.query_durations is None:
+            self.query_durations = []
+        if self.query_types is None:
+            self.query_types = {}
+        if self.slow_queries is None:
+            self.slow_queries = []
 
     def record_connection_start(self):
         """Record new connection start"""
@@ -45,6 +58,26 @@ class ResourceStats:
         self.query_count += 1
         self.last_query_time = datetime.now()
 
+    def record_query_duration(self, sql: str, duration: float):
+        """Record query execution time and type
+
+        Args:
+            sql: SQL query text
+            duration: Execution time in seconds
+        """
+        self.query_durations.append(duration)
+        
+        # Record query type
+        query_type = sql.strip().split()[0].upper()
+        self.query_types[query_type] = self.query_types.get(query_type, 0) + 1
+        
+        # Record slow queries (over 100ms)
+        if duration > 0.1:  # 100ms
+            # Keep at most 10 slow queries
+            if len(self.slow_queries) >= 10:
+                self.slow_queries.pop(0)
+            self.slow_queries.append((sql[:100], duration))  # Truncate SQL to avoid excessive length
+
     def record_error(self, error_type: str):
         """Record error occurrence
 
@@ -61,7 +94,73 @@ class ResourceStats:
         Args:
             obj: Object to estimate size for
         """
-        self.estimated_memory = sys.getsizeof(obj)
+        current_memory = sys.getsizeof(obj)
+        self.estimated_memory = current_memory
+        self.peak_memory = max(self.peak_memory, current_memory)
+
+    def get_query_time_stats(self) -> dict:
+        """Get query time statistics
+
+        Returns:
+            Dictionary with min, max, avg, median query times
+        """
+        if not self.query_durations:
+            return {
+                "min": 0,
+                "max": 0,
+                "avg": 0,
+                "median": 0
+            }
+        
+        return {
+            "min": min(self.query_durations),
+            "max": max(self.query_durations),
+            "avg": sum(self.query_durations) / len(self.query_durations),
+            "median": statistics.median(self.query_durations) if len(self.query_durations) > 0 else 0
+        }
+
+    def get_performance_stats(self) -> str:
+        """Get formatted performance statistics
+
+        Returns:
+            Formatted string with performance statistics
+        """
+        stats = []
+        stats.append(f"Database Performance Statistics")
+        stats.append(f"-----------------------------")
+        stats.append(f"Query Count: {self.query_count}")
+        
+        # Query time statistics
+        if self.query_durations:
+            time_stats = self.get_query_time_stats()
+            stats.append(f"Query Times: avg={time_stats['avg']*1000:.2f}ms, min={time_stats['min']*1000:.2f}ms, max={time_stats['max']*1000:.2f}ms, median={time_stats['median']*1000:.2f}ms")
+        
+        # Query type distribution
+        if self.query_types:
+            stats.append(f"Query Types:")
+            for qtype, count in self.query_types.items():
+                percentage = (count / self.query_count) * 100 if self.query_count else 0
+                stats.append(f"  - {qtype}: {count} ({percentage:.1f}%)")
+        
+        # Slow queries
+        if self.slow_queries:
+            stats.append(f"Slow Queries:")
+            for sql, duration in self.slow_queries:
+                stats.append(f"  - {duration*1000:.2f}ms: {sql}...")
+        
+        # Error statistics
+        if self.error_count > 0:
+            error_rate = (self.error_count / self.query_count) * 100 if self.query_count else 0
+            stats.append(f"Error Rate: {error_rate:.2f}% ({self.error_count} errors)")
+            stats.append(f"Error Types:")
+            for etype, count in self.error_types.items():
+                stats.append(f"  - {etype}: {count}")
+        
+        # Resource usage
+        stats.append(f"Memory Usage: current={self.estimated_memory/1024:.2f}KB, peak={self.peak_memory/1024:.2f}KB")
+        stats.append(f"Connections: active={self.active_connections}, total={self.total_connections}")
+        
+        return "\n".join(stats)
 
     def to_dict(self) -> dict:
         """Convert stats to dictionary for logging
@@ -74,12 +173,22 @@ class ResourceStats:
         if self.connection_start_time:
             connection_duration = (now - self.connection_start_time).total_seconds()
 
+        time_stats = self.get_query_time_stats()
+        
         return {
             "active_connections": self.active_connections,
             "total_connections": self.total_connections,
             "connection_duration": connection_duration,
             "query_count": self.query_count,
+            "query_times_ms": {
+                "min": time_stats["min"] * 1000 if self.query_durations else 0,
+                "max": time_stats["max"] * 1000 if self.query_durations else 0,
+                "avg": time_stats["avg"] * 1000 if self.query_durations else 0,
+                "median": time_stats["median"] * 1000 if self.query_durations else 0
+            },
+            "query_types": self.query_types,
             "error_count": self.error_count,
             "error_types": self.error_types,
-            "estimated_memory_bytes": self.estimated_memory
+            "estimated_memory_bytes": self.estimated_memory,
+            "peak_memory_bytes": self.peak_memory
         }

{mcp_dbutils-0.7.0.dist-info → mcp_dbutils-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-dbutils
-Version: 0.7.0
+Version: 0.9.0
 Summary: MCP Database Utilities Service
 Author: Dong Hao
 License-Expression: MIT
@@ -148,7 +148,7 @@ Add to Claude configuration:
 The project requires a YAML configuration file, specified via the `--config` parameter. Configuration example:
 
 ```yaml
-databases:
+connections:
   # Standard PostgreSQL configuration example
   my_postgres:
     type: postgres
@@ -257,7 +257,7 @@ The abstraction layer design is the core architectural concept in MCP Database U
 
 ### Basic Query
 ```python
-# Access through database name
+# Access through connection name
 async with server.get_handler("my_postgres") as handler:
     # Execute SQL query
     result = await handler.execute_query("SELECT * FROM users")
@@ -275,7 +275,7 @@ schema = await handler.get_schema("users")
 ### Error Handling
 ```python
 try:
-    async with server.get_handler("my_db") as handler:
+    async with server.get_handler("my_connection") as handler:
         result = await handler.execute_query("SELECT * FROM users")
 except ValueError as e:
     print(f"Configuration error: {e}")
@@ -298,19 +298,53 @@ Core server class providing:
 
 ### MCP Tools
 
-#### list_tables
+#### dbutils-list-tables
 Lists all tables in the specified database.
 - Parameters:
-  * database: Database configuration name
+  * connection: Database connection name
 - Returns: Text content with a list of table names
 
-#### query
+#### dbutils-run-query
 Executes a SQL query on the specified database.
 - Parameters:
-  * database: Database configuration name
+  * connection: Database connection name
   * sql: SQL query to execute (SELECT only)
 - Returns: Query results in a formatted text
 
+#### dbutils-get-stats
+Get table statistics information.
+- Parameters:
+  * connection: Database connection name
+  * table: Table name
+- Returns: Statistics including row count, size, column stats
+
+#### dbutils-list-constraints
+List table constraints (primary key, foreign keys, etc).
+- Parameters:
+  * connection: Database connection name
+  * table: Table name
+- Returns: Detailed constraint information
+
+#### dbutils-explain-query
+Get query execution plan with cost estimates.
+- Parameters:
+  * connection: Database connection name
+  * sql: SQL query to explain
+- Returns: Formatted execution plan
+
+#### dbutils-get-performance
+Get database performance statistics.
+- Parameters:
+  * connection: Database connection name
+- Returns: Detailed performance statistics including query times, query types, error rates, and resource usage
+
+#### dbutils-analyze-query
+Analyze a SQL query for performance and provide optimization suggestions.
+- Parameters:
+  * connection: Database connection name
+  * sql: SQL query to analyze
+- Returns: Query analysis with execution plan, timing information, and optimization suggestions
+
 ### DatabaseHandler
 Abstract base class defining interfaces:
 - get_tables(): Get table resource list

mcp_dbutils-0.9.0.dist-info/RECORD

@@ -0,0 +1,18 @@
+mcp_dbutils/__init__.py,sha256=9dqT2BdVvwforQVWOvJq2M0zCHxFamWIYWC4-6aY70k,1905
+mcp_dbutils/base.py,sha256=2fLSPL-BxdgKIOT_eq73wQGla6MZB03jUrWqBRSWVBQ,25474
+mcp_dbutils/config.py,sha256=uFI4Haw4en5gxHfCM9zugUqNCQFikdlHJZU_NTht7gQ,1905
+mcp_dbutils/log.py,sha256=fibVIwsb1HVU5zriGrDZTMEirKjgIuxuN_B_YTdAJ7I,996
+mcp_dbutils/stats.py,sha256=WYD9NAKHH2bFKmUSTOg18-SuIXergpznPq8AtIjtxdI,7101
+mcp_dbutils/postgres/__init__.py,sha256=XanCXw-kVE6ayqFqjuLJ9swWPcCVlcKZXB2Et2Wjl9w,154
+mcp_dbutils/postgres/config.py,sha256=Kqq0ZCa7PhtvMaBX3sfdEscGosHR1dWuiP8IkI5lzDo,5495
+mcp_dbutils/postgres/handler.py,sha256=O45K0MOosNQ68SJ8piXXH3aZ74RVQ981hudk6faj28Y,24397
+mcp_dbutils/postgres/server.py,sha256=FROogvjr3xlKAZcAvJMNHGgyHqT_LaMK9-vYbA0Dbm0,8716
+mcp_dbutils/sqlite/__init__.py,sha256=lTUOkSfSWNw_BohJ_KNxYKMCANdtpGa3JK_ZyJsg_Ls,134
+mcp_dbutils/sqlite/config.py,sha256=2ekTp89rcCu4qQN1O3sZjIcOwjhzWi4tisD20T2WAik,4533
+mcp_dbutils/sqlite/handler.py,sha256=VxBVbwpNSQHyiudQXYFYpcdoScvbK2IGzYMbkcpvFcI,17731
+mcp_dbutils/sqlite/server.py,sha256=49ug43nyXZApclMXo0t5dKCN4wjDyDUZ-vmf746ac8M,7709
+mcp_dbutils-0.9.0.dist-info/METADATA,sha256=I52Q7IV8un9A5tmTTmEOFKZiW_UYCrWwBDhDq1tzFyE,11986
+mcp_dbutils-0.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_dbutils-0.9.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
+mcp_dbutils-0.9.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
+mcp_dbutils-0.9.0.dist-info/RECORD,,

mcp_dbutils-0.7.0.dist-info/RECORD

@@ -1,18 +0,0 @@
-mcp_dbutils/__init__.py,sha256=xcfE1spAaONAoxBYB1ZyDX8tw7nxV1PMqo_RwxLDp0A,1892
-mcp_dbutils/base.py,sha256=aawU0mY78Ga8YhSYvzOaBi1V8qSYTWyAuZzmXTc37Rg,11232
-mcp_dbutils/config.py,sha256=EwnPNuQVCBKd5WOXQfROyDTM-YpM_Odp0GhCPRg8YwE,1863
-mcp_dbutils/log.py,sha256=fibVIwsb1HVU5zriGrDZTMEirKjgIuxuN_B_YTdAJ7I,996
-mcp_dbutils/stats.py,sha256=2hiKi_M8V4xhVHlH5FS-Df5GuMEpuzif12C8ik06Khs,2538
-mcp_dbutils/postgres/__init__.py,sha256=Y6v_RsI79pqAfpKM3SrT1T1I9r5yWuKT0GUUNmHD3DE,146
-mcp_dbutils/postgres/config.py,sha256=Np0GS5iUaUuWpYI8QfnyjUDy7v-vResQEAexR4W92-o,5447
-mcp_dbutils/postgres/handler.py,sha256=JC_Qyw1s6qjVR0tdH7FLAEAs9EPl_-wgFQLryQyMq5c,6089
-mcp_dbutils/postgres/server.py,sha256=_S3HF1KooxPB9gX1FedoOOGn93tHlIevCab6vjCt2TU,8715
-mcp_dbutils/sqlite/__init__.py,sha256=QV4th2ywzUmCCa3GHCcBf8blJ_E8OYy0dJ2fSf1TfSU,134
-mcp_dbutils/sqlite/config.py,sha256=RTHT2Xx--g-osD73CpT8DrCk0VHpHfPil3D6YUzXD-g,4519
-mcp_dbutils/sqlite/handler.py,sha256=bf_k93rCcJn09zc7tsqrlbiTGUg3FspimfWKxK_JQTs,4970
-mcp_dbutils/sqlite/server.py,sha256=7Bbq9l7Ca_4dzkAbbdRcXxvHoO_NFLzZHwlhKB0HIJc,7724
-mcp_dbutils-0.7.0.dist-info/METADATA,sha256=smgCaufFbIkvOFhDGqrajHYC9WxeQcDCWRerr_bxHp8,10840
-mcp_dbutils-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_dbutils-0.7.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
-mcp_dbutils-0.7.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
-mcp_dbutils-0.7.0.dist-info/RECORD,,