refactor(server): Implement async specification method in server factories and streamline server creation process

Author: codeboyzhou

src/main/java/com/github/codeboyzhou/mcp/declarative/server/simple/SimpleMcpHttpSseServerFactory.java

@@ -1,17 +1,17 @@
 package com.github.codeboyzhou.mcp.declarative.server.simple;
 
-import com.github.codeboyzhou.mcp.declarative.common.NamedThreadFactory;
-import com.github.codeboyzhou.mcp.declarative.server.McpHttpServer;
-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 SimpleMcpHttpSseServerFactory
     implements SimpleMcpServerFactory<
         HttpServletSseServerTransportProvider, SimpleMcpHttpSseServerInfo> {
 
+  @Override
+  public McpServer.AsyncSpecification<?> specification(SimpleMcpHttpSseServerInfo serverInfo) {
+    return McpServer.async(transportProvider(serverInfo));
+  }
+
   @Override
   public HttpServletSseServerTransportProvider transportProvider(SimpleMcpHttpSseServerInfo info) {
     return HttpServletSseServerTransportProvider.builder()
@@ -20,21 +20,4 @@ public HttpServletSseServerTransportProvider transportProvider(SimpleMcpHttpSseS
         .messageEndpoint(info.messageEndpoint())
         .build();
   }
-
-  @Override
-  public McpAsyncServer create(SimpleMcpHttpSseServerInfo info) {
-    HttpServletSseServerTransportProvider transportProvider = transportProvider(info);
-    McpAsyncServer server =
-        McpServer.async(transportProvider)
-            .serverInfo(info.name(), info.version())
-            .capabilities(serverCapabilities())
-            .instructions(info.instructions())
-            .requestTimeout(info.requestTimeout())
-            .build();
-    McpHttpServer httpServer = new McpHttpServer();
-    NamedThreadFactory threadFactory = new NamedThreadFactory(McpHttpServer.class.getSimpleName());
-    ExecutorService executor = Executors.newSingleThreadExecutor(threadFactory);
-    executor.execute(() -> httpServer.use(transportProvider).bind(info.port()).start());
-    return server;
-  }
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/simple/SimpleMcpHttpStreamableServerFactory.java

@@ -1,18 +1,19 @@
 package com.github.codeboyzhou.mcp.declarative.server.simple;
 
-import com.github.codeboyzhou.mcp.declarative.common.NamedThreadFactory;
-import com.github.codeboyzhou.mcp.declarative.server.McpHttpServer;
 import com.github.codeboyzhou.mcp.declarative.util.ObjectMappers;
-import io.modelcontextprotocol.server.McpAsyncServer;
 import io.modelcontextprotocol.server.McpServer;
 import io.modelcontextprotocol.server.transport.HttpServletStreamableServerTransportProvider;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
 
 public class SimpleMcpHttpStreamableServerFactory
     implements SimpleMcpServerFactory<
         HttpServletStreamableServerTransportProvider, SimpleMcpHttpStreamableServerInfo> {
 
+  @Override
+  public McpServer.AsyncSpecification<?> specification(
+      SimpleMcpHttpStreamableServerInfo serverInfo) {
+    return McpServer.async(transportProvider(serverInfo));
+  }
+
   @Override
   public HttpServletStreamableServerTransportProvider transportProvider(
       SimpleMcpHttpStreamableServerInfo info) {
@@ -24,21 +25,4 @@ public HttpServletStreamableServerTransportProvider transportProvider(
         .keepAliveInterval(info.keepAliveInterval())
         .build();
   }
-
-  @Override
-  public McpAsyncServer create(SimpleMcpHttpStreamableServerInfo info) {
-    HttpServletStreamableServerTransportProvider transportProvider = transportProvider(info);
-    McpAsyncServer server =
-        McpServer.async(transportProvider)
-            .serverInfo(info.name(), info.version())
-            .capabilities(serverCapabilities())
-            .instructions(info.instructions())
-            .requestTimeout(info.requestTimeout())
-            .build();
-    McpHttpServer httpServer = new McpHttpServer();
-    NamedThreadFactory threadFactory = new NamedThreadFactory(McpHttpServer.class.getSimpleName());
-    ExecutorService executor = Executors.newSingleThreadExecutor(threadFactory);
-    executor.execute(() -> httpServer.use(transportProvider).bind(info.port()).start());
-    return server;
-  }
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/simple/SimpleMcpServerFactory.java

@@ -1,6 +1,7 @@
 package com.github.codeboyzhou.mcp.declarative.server.simple;
 
 import io.modelcontextprotocol.server.McpAsyncServer;
+import io.modelcontextprotocol.server.McpServer;
 import io.modelcontextprotocol.spec.McpSchema;
 import io.modelcontextprotocol.spec.McpServerTransportProviderBase;
 
@@ -15,7 +16,16 @@ default McpSchema.ServerCapabilities serverCapabilities() {
         .build();
   }
 
-  T transportProvider(S serverInfo);
+  default McpAsyncServer create(S serverInfo) {
+    return specification(serverInfo)
+        .serverInfo(serverInfo.name(), serverInfo.version())
+        .capabilities(serverCapabilities())
+        .instructions(serverInfo.instructions())
+        .requestTimeout(serverInfo.requestTimeout())
+        .build();
+  }
 
-  McpAsyncServer create(S serverInfo);
+  McpServer.AsyncSpecification<?> specification(S serverInfo);
+
+  T transportProvider(S serverInfo);
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/simple/SimpleMcpStdioServerFactory.java

@@ -1,24 +1,18 @@
 package com.github.codeboyzhou.mcp.declarative.server.simple;
 
-import io.modelcontextprotocol.server.McpAsyncServer;
 import io.modelcontextprotocol.server.McpServer;
 import io.modelcontextprotocol.server.transport.StdioServerTransportProvider;
 
 public class SimpleMcpStdioServerFactory
     implements SimpleMcpServerFactory<StdioServerTransportProvider, SimpleMcpServerBaseInfo> {
 
   @Override
-  public StdioServerTransportProvider transportProvider(SimpleMcpServerBaseInfo serverInfo) {
-    return new StdioServerTransportProvider();
+  public McpServer.AsyncSpecification<?> specification(SimpleMcpServerBaseInfo serverInfo) {
+    return McpServer.async(transportProvider(serverInfo));
   }
 
   @Override
-  public McpAsyncServer create(SimpleMcpServerBaseInfo serverInfo) {
-    return McpServer.async(transportProvider(serverInfo))
-        .serverInfo(serverInfo.name(), serverInfo.version())
-        .capabilities(serverCapabilities())
-        .instructions(serverInfo.instructions())
-        .requestTimeout(serverInfo.requestTimeout())
-        .build();
+  public StdioServerTransportProvider transportProvider(SimpleMcpServerBaseInfo serverInfo) {
+    return new StdioServerTransportProvider();
   }
 }