mcp-code-indexer 2.0.0__py3-none-any.whl → 2.0.2__py3-none-any.whl

mcp_code_indexer/database/database.py

@@ -39,16 +39,26 @@ class DatabaseManager:
     and caching with proper transaction management and error handling.
     """
     
-    def __init__(self, db_path: Path, pool_size: int = 3):
+    def __init__(self, 
+                 db_path: Path, 
+                 pool_size: int = 3,
+                 retry_count: int = 5,
+                 timeout: float = 10.0,
+                 enable_wal_mode: bool = True,
+                 health_check_interval: float = 30.0):
         """Initialize database manager with path to SQLite database."""
         self.db_path = db_path
         self.pool_size = pool_size
+        self.retry_count = retry_count
+        self.timeout = timeout
+        self.enable_wal_mode = enable_wal_mode
+        self.health_check_interval = health_check_interval
         self._connection_pool: List[aiosqlite.Connection] = []
         self._pool_lock = None  # Will be initialized in async context
         self._write_lock = None  # Write serialization lock, initialized in async context
         
-        # Retry and recovery components
-        self._retry_handler = create_retry_handler()
+        # Retry and recovery components - configure with provided settings
+        self._retry_handler = create_retry_handler(max_attempts=retry_count)
         self._recovery_manager = None  # Initialized in async context
         
         # Health monitoring and metrics
@@ -66,8 +76,12 @@ class DatabaseManager:
         # Initialize connection recovery manager
         self._recovery_manager = ConnectionRecoveryManager(self)
         
-        # Initialize health monitoring
-        self._health_monitor = ConnectionHealthMonitor(self)
+        # Initialize health monitoring with configured interval
+        self._health_monitor = ConnectionHealthMonitor(
+            self, 
+            check_interval=self.health_check_interval,
+            timeout_seconds=self.timeout
+        )
         await self._health_monitor.start_monitoring()
         
         # Ensure database directory exists
@@ -82,7 +96,7 @@ class DatabaseManager:
             db.row_factory = aiosqlite.Row
             
             # Configure WAL mode and optimizations for concurrent access
-            await self._configure_database_optimizations(db)
+            await self._configure_database_optimizations(db, include_wal_mode=self.enable_wal_mode)
             
             # Apply each migration
             for migration_file in migration_files:
@@ -108,6 +122,7 @@ class DatabaseManager:
         # WAL mode is database-level, only set during initialization
         if include_wal_mode:
             optimizations.append("PRAGMA journal_mode = WAL")
+            logger.info("Enabling WAL mode for database concurrency")
         
         # Connection-level optimizations that can be set per connection
         optimizations.extend([
@@ -422,7 +437,7 @@ class DatabaseManager:
     
     async def create_project(self, project: Project) -> None:
         """Create a new project record."""
-        async with self.get_write_connection() as db:
+        async with self.get_write_connection_with_retry("create_project") as db:
             await db.execute(
                 """
                 INSERT INTO projects (id, name, remote_origin, upstream_origin, aliases, created, last_accessed)
@@ -609,7 +624,7 @@ class DatabaseManager:
     
     async def update_project_access_time(self, project_id: str) -> None:
         """Update the last accessed time for a project."""
-        async with self.get_write_connection() as db:
+        async with self.get_write_connection_with_retry("update_project_access_time") as db:
             await db.execute(
                 "UPDATE projects SET last_accessed = ? WHERE id = ?",
                 (datetime.utcnow(), project_id)
@@ -618,7 +633,7 @@ class DatabaseManager:
     
     async def update_project(self, project: Project) -> None:
         """Update an existing project record."""
-        async with self.get_write_connection() as db:
+        async with self.get_write_connection_with_retry("update_project") as db:
             await db.execute(
                 """
                 UPDATE projects 
@@ -680,7 +695,7 @@ class DatabaseManager:
     
     async def create_file_description(self, file_desc: FileDescription) -> None:
         """Create or update a file description."""
-        async with self.get_write_connection() as db:
+        async with self.get_write_connection_with_retry("create_file_description") as db:
             await db.execute(
                 """
                 INSERT OR REPLACE INTO file_descriptions 

mcp_code_indexer/server/mcp_server.py

@@ -83,7 +83,14 @@ class MCPCodeIndexServer:
         }
         
         # Initialize components
-        self.db_manager = DatabaseManager(self.db_path, pool_size=db_pool_size)
+        self.db_manager = DatabaseManager(
+            db_path=self.db_path, 
+            pool_size=db_pool_size,
+            retry_count=db_retry_count,
+            timeout=db_timeout,
+            enable_wal_mode=enable_wal_mode,
+            health_check_interval=health_check_interval
+        )
         self.token_counter = TokenCounter(token_limit)
         self.merge_handler = MergeHandler(self.db_manager)
         

{mcp_code_indexer-2.0.0.dist-info → mcp_code_indexer-2.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-code-indexer
-Version: 2.0.0
+Version: 2.0.2
 Summary: MCP server that tracks file descriptions across codebases, enabling AI agents to efficiently navigate and understand code through searchable summaries and token-aware overviews.
 Author: MCP Code Indexer Contributors
 Maintainer: MCP Code Indexer Contributors
@@ -59,8 +59,8 @@ Dynamic: requires-python
 
 # MCP Code Indexer 🚀
 
-[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?11)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?11)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?13)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?13)](https://pypi.org/project/mcp-code-indexer/)
 [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
 
 A production-ready **Model Context Protocol (MCP) server** that revolutionizes how AI agents navigate and understand codebases. Built for high-concurrency environments with advanced database resilience, the server provides instant access to intelligent descriptions, semantic search, and context-aware recommendations while maintaining 800+ writes/sec throughput.

{mcp_code_indexer-2.0.0.dist-info → mcp_code_indexer-2.0.2.dist-info}/RECORD

@@ -10,18 +10,18 @@ mcp_code_indexer/token_counter.py,sha256=WrifOkbF99nWWHlRlhCHAB2KN7qr83GOHl7apE-
 mcp_code_indexer/data/stop_words_english.txt,sha256=7Zdd9ameVgA6tN_zuXROvHXD4hkWeELVywPhb7FJEkw,6343
 mcp_code_indexer/database/__init__.py,sha256=aPq_aaRp0aSwOBIq9GkuMNjmLxA411zg2vhdrAuHm-w,38
 mcp_code_indexer/database/connection_health.py,sha256=XJvUrHRhIroZlIPScVGdKb69lNP67lT9ZTTO67cFSEs,16721
-mcp_code_indexer/database/database.py,sha256=975bJCsj8G970UbRie3ge5ctFV5pmeOhqvNGt78Y5z8,48769
+mcp_code_indexer/database/database.py,sha256=5G_1E-jSVMDJuyFCVki9tbNmoyNvHJa3x7dRIBB2UQE,49603
 mcp_code_indexer/database/models.py,sha256=_vCmJnPXZSiInRzyvs4c7QUWuNNW8qsOoDlGX8J-Gnk,7124
 mcp_code_indexer/database/retry_handler.py,sha256=zwwZ0V1PzzS1rtcfVQOI-CXqWnPGF-KGH4L_3d-_h1Y,11932
 mcp_code_indexer/middleware/__init__.py,sha256=p-mP0pMsfiU2yajCPvokCUxUEkh_lu4XJP1LyyMW2ug,220
 mcp_code_indexer/middleware/error_middleware.py,sha256=5agJTAkkPogfPGnja1V9JtG9RG-BiOALIJYctK3byJQ,11730
 mcp_code_indexer/server/__init__.py,sha256=16xMcuriUOBlawRqWNBk6niwrvtv_JD5xvI36X1Vsmk,41
-mcp_code_indexer/server/mcp_server.py,sha256=1GFvfORtUW3Am8me3joCg3Q61epipsB2I-1uv6WL15Q,66717
+mcp_code_indexer/server/mcp_server.py,sha256=tQTZFnAnASStkbnQOsj2-3eH1DYSmY97q1pwo7geFXI,66934
 mcp_code_indexer/tiktoken_cache/9b5ad71b2ce5302211f9c61530b329a4922fc6a4,sha256=Ijkht27pm96ZW3_3OFE-7xAPtR0YyTWXoRO8_-hlsqc,1681126
 mcp_code_indexer/tools/__init__.py,sha256=m01mxML2UdD7y5rih_XNhNSCMzQTz7WQ_T1TeOcYlnE,49
-mcp_code_indexer-2.0.0.dist-info/licenses/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
-mcp_code_indexer-2.0.0.dist-info/METADATA,sha256=vtQ3OS8AfTQuZmLqVFEyPzMtvvX9-V_KoyI4ygMvQ00,20165
-mcp_code_indexer-2.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mcp_code_indexer-2.0.0.dist-info/entry_points.txt,sha256=8HqWOw1Is7jOP1bvIgaSwouvT9z_Boe-9hd4NzyJOhY,68
-mcp_code_indexer-2.0.0.dist-info/top_level.txt,sha256=yKYCM-gMGt-cnupGfAhnZaoEsROLB6DQ1KFUuyKx4rw,17
-mcp_code_indexer-2.0.0.dist-info/RECORD,,
+mcp_code_indexer-2.0.2.dist-info/licenses/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
+mcp_code_indexer-2.0.2.dist-info/METADATA,sha256=BsEhIKscok46a5pfTyigsaMIHjMHZgmg-7-O_OyTrZY,20165
+mcp_code_indexer-2.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mcp_code_indexer-2.0.2.dist-info/entry_points.txt,sha256=8HqWOw1Is7jOP1bvIgaSwouvT9z_Boe-9hd4NzyJOhY,68
+mcp_code_indexer-2.0.2.dist-info/top_level.txt,sha256=yKYCM-gMGt-cnupGfAhnZaoEsROLB6DQ1KFUuyKx4rw,17
+mcp_code_indexer-2.0.2.dist-info/RECORD,,