refactor(server): Convert McpHttpServer to a record and update start method signature

Author: codeboyzhou

src/main/java/com/github/codeboyzhou/mcp/declarative/server/McpHttpServer.java

@@ -8,24 +8,15 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public class McpHttpServer {
+public record McpHttpServer(int port) {
 
   private static final Logger logger = LoggerFactory.getLogger(McpHttpServer.class);
 
   private static final String DEFAULT_SERVLET_CONTEXT_PATH = "/";
 
   private static final String DEFAULT_SERVLET_PATH = "/*";
 
-  private final HttpServletSseServerTransportProvider transportProvider;
-
-  private final int port;
-
-  public McpHttpServer(HttpServletSseServerTransportProvider transportProvider, int port) {
-    this.transportProvider = transportProvider;
-    this.port = port;
-  }
-
-  public void start() {
+  public void start(HttpServletSseServerTransportProvider transportProvider) {
     ServletContextHandler handler = new ServletContextHandler(ServletContextHandler.SESSIONS);
     handler.setContextPath(DEFAULT_SERVLET_CONTEXT_PATH);
 
@@ -62,16 +53,16 @@ public void start() {
   }
 
   private void addShutdownHook(Server httpserver) {
-    Runtime.getRuntime()
-        .addShutdownHook(
-            new Thread(
-                () -> {
-                  try {
-                    logger.info("Shutting down HTTP server and MCP server");
-                    httpserver.stop();
-                  } catch (Exception e) {
-                    logger.error("Error stopping HTTP server and MCP server", e);
-                  }
-                }));
+    Runnable runnable =
+        () -> {
+          try {
+            logger.info("Shutting down HTTP server and MCP server");
+            httpserver.stop();
+          } catch (Exception e) {
+            logger.error("Error stopping HTTP server and MCP server", e);
+          }
+        };
+    Thread shutdownHook = new Thread(runnable);
+    Runtime.getRuntime().addShutdownHook(shutdownHook);
   }
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/factory/ConfigurableMcpHttpSseServerFactory.java

@@ -8,6 +8,7 @@
 import io.modelcontextprotocol.server.McpServer;
 import io.modelcontextprotocol.server.transport.HttpServletSseServerTransportProvider;
 import java.time.Duration;
+import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 
 public class ConfigurableMcpHttpSseServerFactory
@@ -40,9 +41,10 @@ public McpAsyncServer create() {
             .instructions(configuration.instructions())
             .requestTimeout(Duration.ofMillis(configuration.requestTimeout()))
             .build();
-    McpHttpServer httpServer = new McpHttpServer(transportProvider, configuration.sse().port());
+    McpHttpServer httpServer = new McpHttpServer(configuration.sse().port());
     NamedThreadFactory threadFactory = new NamedThreadFactory(McpHttpServer.class.getSimpleName());
-    Executors.newSingleThreadExecutor(threadFactory).execute(httpServer::start);
+    ExecutorService executor = Executors.newSingleThreadExecutor(threadFactory);
+    executor.execute(() -> httpServer.start(transportProvider));
     return server;
   }
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/factory/McpHttpSseServerFactory.java

@@ -6,6 +6,7 @@
 import io.modelcontextprotocol.server.McpAsyncServer;
 import io.modelcontextprotocol.server.McpServer;
 import io.modelcontextprotocol.server.transport.HttpServletSseServerTransportProvider;
+import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 
 public class McpHttpSseServerFactory
@@ -33,9 +34,10 @@ public McpAsyncServer create(McpSseServerInfo serverInfo) {
             .instructions(serverInfo.instructions())
             .requestTimeout(serverInfo.requestTimeout())
             .build();
-    McpHttpServer httpServer = new McpHttpServer(transportProvider, serverInfo.port());
+    McpHttpServer httpServer = new McpHttpServer(serverInfo.port());
     NamedThreadFactory threadFactory = new NamedThreadFactory(McpHttpServer.class.getSimpleName());
-    Executors.newSingleThreadExecutor(threadFactory).execute(httpServer::start);
+    ExecutorService executor = Executors.newSingleThreadExecutor(threadFactory);
+    executor.execute(() -> httpServer.start(transportProvider));
     return server;
   }
 }