66capabilities through the Model Context Protocol (MCP) using Joern's Code Property Graph.
77"""
88
9- import asyncio
109import logging
1110import os
12- import signal
1311import shutil
1412import socket
1513from fastmcp import FastMCP
3432
3533# Global service instances
3634services = {}
37- shutdown_event = asyncio .Event ()
3835
3936logger = logging .getLogger (__name__ )
4037
@@ -71,18 +68,6 @@ async def _graceful_shutdown():
7168 logger .error (f"Error during graceful shutdown: { e } " , exc_info = True )
7269
7370
74- def _setup_signal_handlers (mcp : FastMCP ):
75- """Setup signal handlers for graceful shutdown"""
76- def signal_handler (signum , frame ):
77- logger .info (f"Received signal { signum } , initiating graceful shutdown..." )
78- asyncio .create_task (_graceful_shutdown ())
79-
80- # Register handlers for SIGTERM and SIGINT
81- signal .signal (signal .SIGTERM , signal_handler )
82- signal .signal (signal .SIGINT , signal_handler )
83- logger .debug ("Signal handlers registered for SIGTERM and SIGINT" )
84-
85-
8671@lifespan
8772async def app_lifespan (server : FastMCP ):
8873 """Startup and shutdown logic for the FastMCP server"""
@@ -91,9 +76,6 @@ async def app_lifespan(server: FastMCP):
9176 setup_logging (config .server .log_level )
9277 logger .info ("Starting CodeBadger Server" )
9378
94- # Setup signal handlers for graceful shutdown
95- _setup_signal_handlers (server )
96-
9779 # Ensure required directories exist
9880 os .makedirs (config .storage .workspace_root , exist_ok = True )
9981
0 commit comments