keep start if no server

Author: daimor

src/mcp_server_iris/interoperability.py

@@ -63,7 +63,7 @@ async def interoperability_production_create(name: str, ctx: Context) -> bool:
             raise ValueError(
                 "Production name must in format packagenamespace.productionname, where packagenamespace can have multiple parts separated by dots"
             )
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         with transaction(iris):
             prod = iris.classMethodObject(
                 "%Dictionary.ClassDefinition", "%OpenId", name
@@ -86,7 +86,7 @@ async def interoperability_production_status(
         full_status: bool = False,
     ) -> str:
         logger.info("Interoperability Production Status" + f": {name}" if name else "")
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         refname = IRISReference(iris)
         refname.setValue(name)
         refstatus = IRISReference(iris)
@@ -123,7 +123,7 @@ async def interoperability_production_start(
         logger.info(
             "Starting Interoperability Production" + f": {name}" if name else "."
         )
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         raise_on_error(
             iris,
             iris.classMethodString(
@@ -152,7 +152,7 @@ async def interoperability_production_stop(
         force: bool = False,
     ) -> str:
         logger.info("Sopping Interoperability Production.")
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         raise_on_error(
             iris,
             iris.classMethodString(
@@ -166,7 +166,7 @@ async def interoperability_production_recover(
         ctx: Context,
     ) -> str:
         logger.info("Recovering Interoperability Production")
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         raise_on_error(
             iris, iris.classMethodString("Ens.Director", "RecoverProduction")
         )
@@ -177,7 +177,7 @@ async def interoperability_production_needsupdate(
         ctx: Context,
     ) -> str:
         logger.info("Checking if Interoperability Production needs update")
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         reason = IRISReference(iris)
         result = iris.classMethodBoolean(
             "Ens.Director", "ProductionNeedsUpdate", reason
@@ -192,7 +192,7 @@ async def interoperability_production_update(
         timeout: int = None,
         force: bool = False,
     ) -> str:
-        iris = ctx.request_context.lifespan_context["iris"]
+        iris = ctx.iris
         raise_on_error(
             iris,
             iris.classMethodString("Ens.Director", "UpdateProduction", timeout, force),
@@ -215,7 +215,7 @@ async def interoperability_production_logs(
         log_type_alert and log_type.append(LogType.Alert.value)
         log_type_error and log_type.append(LogType.Error.value)
         log_type_warning and log_type.append(LogType.Warning.value)
-        db = ctx.request_context.lifespan_context["db"]
+        db = ctx.db
         with db.cursor() as cur:
             sql = f"""
 select top ? TimeLogged , %External(Type) Type, ConfigName, Text
@@ -236,7 +236,7 @@ async def interoperability_production_queues(
         ctx: Context,
     ) -> str:
         queues = []
-        db = ctx.request_context.lifespan_context["db"]
+        db = ctx.db
         with db.cursor() as cur:
             sql = "select * from Ens.Queue_Enumerate()"
             cur.execute(sql)

src/mcp_server_iris/server.py

@@ -35,6 +35,10 @@ async def server_lifespan(server: MCPServer) -> AsyncIterator[dict]:
         db = irisnative.connect(**config)
         iris = irisnative.createIRIS(db)
         yield {"db": db, "iris": iris}
+    except Exception:
+        db = None
+        iris = None
+        yield {"db": db, "iris": iris}
     finally:
         if db:
             db.close()
@@ -148,7 +152,7 @@ async def execute_sql(
 ) -> list[types.TextContent]:
     # params = arguments.get("params", [])
     logger.info(f"Executing SQL query: {query}")
-    conn = ctx.request_context.lifespan_context["db"]
+    conn = ctx.db
     with conn.cursor() as cursor:
         cursor.execute(query, params)
         # limit by 100 rows

src/mcp_server_iris/sql.py

@@ -0,0 +1,21 @@
+import logging
+import mcp.types as types
+from mcp_server_iris.mcpserver import MCPServer, Context
+
+
+def init(server: MCPServer, logger: logging.Logger) -> None:
+    """Initialize the SQL server with a tool to execute SQL queries."""
+    logger.info("Initializing SQL tool for InterSystems IRIS MCP Server")
+
+    @server.tool(description="Execute an SQL query on the Server")
+    async def execute_sql(
+        query: str, ctx: Context, params: list[str] = []
+    ) -> list[types.TextContent]:
+        # params = arguments.get("params", [])
+        logger.info(f"Executing SQL query: {query}")
+        conn = ctx.db
+        with conn.cursor() as cursor:
+            cursor.execute(query, params)
+            # limit by 100 rows
+            rows = cursor.fetchall()[:100]
+            return [types.TextContent(type="text", text=str(rows))]