feat: add Docker support and implement SSE in MCP server

Author: alaturqua

.dockerignore

@@ -0,0 +1,13 @@
+.env
+.venv/
+.vscode/
+.github/
+dist/
+__pycache__/
+*.egg-info/
+*.pyc
+*.pyo
+*.pyd
+*.db
+
+docker-compose.yaml

.gitignore

@@ -172,3 +172,5 @@ cython_debug/
 
 # PyPI configuration file
 .pypirc
+
+.vscode/

Dockerfile

@@ -18,4 +18,7 @@ ADD . /app
 
 ENV PATH="/app/.venv/bin:$PATH"
 
-ENTRYPOINT [ "uv", "run", "mcp", "run", "src/server.py" ]
+EXPOSE 8000
+
+ENTRYPOINT [ "uv", "run", "src/server.py" ]
+CMD [ "--host", "0.0.0.0", "--port", "8000" ]   

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcp-trino-python"
-version = "0.4.0"
+version = "0.6.0"
 description = "A Model Context Protocol (MCP) connector for Trino, enabling seamless integration between MCP-compliant services and Trino query engine"
 readme = "README.md"
 license = { text = "Apache-2.0" }

src/server.py

@@ -4,9 +4,15 @@
 functionality through resources and tools, with special support for Iceberg tables.
 """
 
+import uvicorn
+from mcp.server import Server
 from mcp.server.fastmcp import FastMCP
 from mcp.server.fastmcp.prompts import base
+from mcp.server.sse import SseServerTransport
 from pydantic import Field
+from starlette.applications import Starlette
+from starlette.requests import Request
+from starlette.routing import Mount, Route
 
 from config import load_config
 from trino_client import TrinoClient
@@ -533,8 +539,46 @@ def maintain_iceberg(table: str, catalog: str, schema_name: str) -> list[base.Me
     ]
 
 
+def create_starlette_app(mcp_server: Server, *, debug: bool = False) -> Starlette:
+    """Create a Starlette application that can server the provied mcp server with SSE."""
+    sse = SseServerTransport("/messages/")
+
+    async def handle_sse(request: Request) -> None:
+        async with sse.connect_sse(
+            request.scope,
+            request.receive,
+            request._send,  # noqa: SLF001
+        ) as (read_stream, write_stream):
+            await mcp_server.run(
+                read_stream,
+                write_stream,
+                mcp_server.create_initialization_options(),
+            )
+
+    return Starlette(
+        debug=debug,
+        routes=[
+            Route("/sse", endpoint=handle_sse),
+            Mount("/messages/", app=sse.handle_post_message),
+        ],
+    )
+
+
 if __name__ == "__main__":
+    import argparse
+
     from loguru import logger
 
     logger.info("Starting Trino MCP server...")
-    mcp.run(transport="stdio")
+    mcp_server = mcp._mcp_server  # noqa: SLF001
+
+    parser = argparse.ArgumentParser(description="Run MCP Trino Server as a SSE-based server")
+    parser.add_argument("--host", default="127.0.0.1", help="Host to bind to")
+    parser.add_argument("--port", type=int, default=8000, help="Port to listen on")
+    args = parser.parse_args()
+
+    # Bind SSE request handling to MCP server
+    starlette_app = create_starlette_app(mcp_server, debug=True)
+    # Run the server with Uvicorn
+    uvicorn.run(starlette_app, host=args.host, port=args.port)
+    logger.info("Trino MCP server is running.")