refactor(spring-ai-mcp-server): enhance MCP server auto-configuration

- Update WebFlux and WebMvc server configurations to use ObjectProvider for ObjectMapper
- Add integration tests for MCP server auto-configurations
- Improve conditional checks for stdio configuration

Signed-off-by: Ahoo Wang <[email protected]>

Author: Ahoo-Wang

auto-configurations/spring-ai-mcp-server/src/main/java/org/springframework/ai/autoconfigure/mcp/server/McpWebFluxServerAutoConfiguration.java

@@ -20,6 +20,7 @@
 import io.modelcontextprotocol.server.transport.WebFluxSseServerTransport;
 import io.modelcontextprotocol.spec.ServerMcpTransport;
 
+import org.springframework.beans.factory.ObjectProvider;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
@@ -72,8 +73,10 @@ public class McpWebFluxServerAutoConfiguration {
 
 	@Bean
 	@ConditionalOnMissingBean
-	public WebFluxSseServerTransport webFluxTransport(McpServerProperties serverProperties) {
-		return new WebFluxSseServerTransport(new ObjectMapper(), serverProperties.getSseMessageEndpoint());
+	public WebFluxSseServerTransport webFluxTransport(ObjectProvider<ObjectMapper> objectMapperProvider,
+			McpServerProperties serverProperties) {
+		ObjectMapper objectMapper = objectMapperProvider.getIfAvailable(ObjectMapper::new);
+		return new WebFluxSseServerTransport(objectMapper, serverProperties.getSseMessageEndpoint());
 	}
 
 	// Router function for SSE transport used by Spring WebFlux to start an HTTP server.

auto-configurations/spring-ai-mcp-server/src/main/java/org/springframework/ai/autoconfigure/mcp/server/McpWebMvcServerAutoConfiguration.java

@@ -20,6 +20,7 @@
 import io.modelcontextprotocol.server.transport.WebMvcSseServerTransport;
 import io.modelcontextprotocol.spec.ServerMcpTransport;
 
+import org.springframework.beans.factory.ObjectProvider;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
@@ -68,8 +69,9 @@ public class McpWebMvcServerAutoConfiguration {
 
 	@Bean
 	@ConditionalOnMissingBean
-	public WebMvcSseServerTransport webMvcSseServerTransport(ObjectMapper objectMapper,
+	public WebMvcSseServerTransport webMvcSseServerTransport(ObjectProvider<ObjectMapper> objectMapperProvider,
 			McpServerProperties serverProperties) {
+		ObjectMapper objectMapper = objectMapperProvider.getIfAvailable(ObjectMapper::new);
 		return new WebMvcSseServerTransport(objectMapper, serverProperties.getSseMessageEndpoint());
 	}
 

auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpWebFluxServerAutoConfigurationIT.java

@@ -0,0 +1,40 @@
+package org.springframework.ai.autoconfigure.mcp.server;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import io.modelcontextprotocol.server.transport.WebFluxSseServerTransport;
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.autoconfigure.AutoConfigurations;
+import org.springframework.boot.test.context.runner.ApplicationContextRunner;
+import org.springframework.web.reactive.function.server.RouterFunction;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class McpWebFluxServerAutoConfigurationIT {
+
+	private final ApplicationContextRunner contextRunner = new ApplicationContextRunner().withConfiguration(
+			AutoConfigurations.of(McpWebFluxServerAutoConfiguration.class, McpServerAutoConfiguration.class));
+
+	@Test
+	void defaultConfiguration() {
+		this.contextRunner.run(context -> {
+			assertThat(context).hasSingleBean(WebFluxSseServerTransport.class);
+			assertThat(context).hasSingleBean(RouterFunction.class);
+		});
+	}
+
+	@Test
+	void objectMapperConfiguration() {
+		this.contextRunner.withBean(ObjectMapper.class, ObjectMapper::new).run(context -> {
+			assertThat(context).hasSingleBean(WebFluxSseServerTransport.class);
+			assertThat(context).hasSingleBean(RouterFunction.class);
+		});
+	}
+
+	@Test
+	void stdioEnabledConfiguration() {
+		this.contextRunner.withPropertyValues("spring.ai.mcp.server.stdio=true").run(context -> {
+			assertThat(context).doesNotHaveBean(WebFluxSseServerTransport.class);
+		});
+	}
+
+}

auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpWebMvcServerAutoConfigurationTest.java

@@ -0,0 +1,39 @@
+package org.springframework.ai.autoconfigure.mcp.server;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import io.modelcontextprotocol.server.transport.WebMvcSseServerTransport;
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.autoconfigure.AutoConfigurations;
+import org.springframework.boot.test.context.runner.ApplicationContextRunner;
+import org.springframework.web.servlet.function.RouterFunction;
+import static org.assertj.core.api.Assertions.assertThat;
+
+class McpWebMvcServerAutoConfigurationTest {
+
+	private final ApplicationContextRunner contextRunner = new ApplicationContextRunner().withConfiguration(
+			AutoConfigurations.of(McpWebMvcServerAutoConfiguration.class, McpServerAutoConfiguration.class));
+
+	@Test
+	void defaultConfiguration() {
+		this.contextRunner.run(context -> {
+			assertThat(context).hasSingleBean(WebMvcSseServerTransport.class);
+			assertThat(context).hasSingleBean(RouterFunction.class);
+		});
+	}
+
+	@Test
+	void objectMapperConfiguration() {
+		this.contextRunner.withBean(ObjectMapper.class, ObjectMapper::new).run(context -> {
+			assertThat(context).hasSingleBean(WebMvcSseServerTransport.class);
+			assertThat(context).hasSingleBean(RouterFunction.class);
+		});
+	}
+
+	@Test
+	void stdioEnabledConfiguration() {
+		this.contextRunner.withPropertyValues("spring.ai.mcp.server.stdio=true").run(context -> {
+			assertThat(context).doesNotHaveBean(WebMvcSseServerTransport.class);
+		});
+	}
+
+}