feat(server): Add support for MCP server configuration

Author: codeboyzhou

pom.xml

@@ -51,6 +51,7 @@
         <maven-surefire-plugin.version>3.2.3</maven-surefire-plugin.version>
         <sortpom-maven-plugin.version>4.0.0</sortpom-maven-plugin.version>
         <!--==================== dependency versions ======================-->
+        <jackson-dataformat-yaml.version>2.18.3</jackson-dataformat-yaml.version>
         <jetty.version>12.0.18</jetty.version>
         <junit5.version>5.10.2</junit5.version>
         <logback.version>1.5.18</logback.version>
@@ -78,6 +79,11 @@
             <version>${logback.version}</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>com.fasterxml.jackson.dataformat</groupId>
+            <artifactId>jackson-dataformat-yaml</artifactId>
+            <version>${jackson-dataformat-yaml.version}</version>
+        </dependency>
         <dependency>
             <groupId>io.modelcontextprotocol.sdk</groupId>
             <artifactId>mcp</artifactId>

src/main/java/com/github/codeboyzhou/mcp/declarative/McpServers.java

@@ -2,6 +2,10 @@
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.github.codeboyzhou.mcp.declarative.annotation.McpComponentScan;
+import com.github.codeboyzhou.mcp.declarative.configuration.McpServerConfiguration;
+import com.github.codeboyzhou.mcp.declarative.configuration.YamlConfigurationLoader;
+import com.github.codeboyzhou.mcp.declarative.enums.ServerType;
+import com.github.codeboyzhou.mcp.declarative.exception.McpServerException;
 import com.github.codeboyzhou.mcp.declarative.listener.DefaultMcpSyncHttpServerStatusListener;
 import com.github.codeboyzhou.mcp.declarative.listener.McpHttpServerStatusListener;
 import com.github.codeboyzhou.mcp.declarative.server.McpHttpServer;
@@ -15,11 +19,16 @@
 import io.modelcontextprotocol.server.transport.StdioServerTransportProvider;
 import io.modelcontextprotocol.spec.McpServerTransportProvider;
 import org.reflections.Reflections;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
+import java.io.IOException;
 import java.time.Duration;
 
 public class McpServers {
 
+    private static final Logger logger = LoggerFactory.getLogger(McpServers.class);
+
     private static final McpServers INSTANCE = new McpServers();
 
     private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();
@@ -73,4 +82,59 @@ public void startSyncSseServer(McpSseServerInfo serverInfo) {
         startSyncSseServer(serverInfo, new DefaultMcpSyncHttpServerStatusListener());
     }
 
+    public void startServer(String configFileName) {
+        McpServerConfiguration configuration = loadConfiguration(configFileName);
+        if (configuration.enabled()) {
+            startServerWith(configuration);
+        } else {
+            logger.info("MCP server is disabled.");
+        }
+    }
+
+    public void startServer() {
+        // Load configuration from default file
+        startServer(null);
+    }
+
+    private McpServerConfiguration loadConfiguration(String configFileName) {
+        YamlConfigurationLoader configurationLoader = new YamlConfigurationLoader();
+        McpServerConfiguration configuration;
+        try {
+            if (configFileName == null || configFileName.isBlank()) {
+                configuration = configurationLoader.loadConfiguration();
+            } else {
+                configuration = configurationLoader.load(configFileName);
+            }
+        } catch (IOException e) {
+            throw new McpServerException("Error loading configuration file", e);
+        }
+        return configuration;
+    }
+
+    private void startServerWith(McpServerConfiguration configuration) {
+        if (ServerType.SYNC.name().equalsIgnoreCase(configuration.type())) {
+            if (configuration.stdio()) {
+                McpServerInfo serverInfo = McpServerInfo.builder()
+                    .name(configuration.name())
+                    .version(configuration.version())
+                    .instructions(configuration.instructions())
+                    .requestTimeout(Duration.ofSeconds(configuration.requestTimeout()))
+                    .build();
+                startSyncStdioServer(serverInfo);
+            } else {
+                McpSseServerInfo serverInfo = McpSseServerInfo.builder()
+                    .name(configuration.name())
+                    .version(configuration.version())
+                    .instructions(configuration.instructions())
+                    .requestTimeout(Duration.ofSeconds(configuration.requestTimeout()))
+                    .baseUrl(configuration.baseUrl())
+                    .messageEndpoint(configuration.sseMessageEndpoint())
+                    .sseEndpoint(configuration.sseEndpoint())
+                    .port(configuration.ssePort())
+                    .build();
+                startSyncSseServer(serverInfo);
+            }
+        }
+    }
+
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/configuration/McpServerConfiguration.java

@@ -0,0 +1,45 @@
+package com.github.codeboyzhou.mcp.declarative.configuration;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+import java.util.Map;
+
+public record McpServerConfiguration(
+    @JsonProperty("enabled") boolean enabled,
+    @JsonProperty("stdio") boolean stdio,
+    @JsonProperty("name") String name,
+    @JsonProperty("version") String version,
+    @JsonProperty("instructions") String instructions,
+    @JsonProperty("request-timeout") long requestTimeout,
+    @JsonProperty("type") String type,
+    @JsonProperty("resource-change-notification") boolean resourceChangeNotification,
+    @JsonProperty("prompt-change-notification") boolean promptChangeNotification,
+    @JsonProperty("tool-change-notification") boolean toolChangeNotification,
+    @JsonProperty("tool-response-mime-type") Map<String, String> toolResponseMimeType,
+    @JsonProperty("sse-message-endpoint") String sseMessageEndpoint,
+    @JsonProperty("sse-endpoint") String sseEndpoint,
+    @JsonProperty("base-url") String baseUrl,
+    @JsonProperty("sse-port") int ssePort
+) {
+
+    public static McpServerConfiguration defaultConfiguration() {
+        return new McpServerConfiguration(
+            true,
+            false,
+            "mcp-server",
+            "1.0.0",
+            "mcp-server",
+            10000,
+            "SYNC",
+            true,
+            true,
+            true,
+            Map.of(),
+            "/mcp/message",
+            "/sse",
+            "",
+            8080
+        );
+    }
+
+}

src/main/java/com/github/codeboyzhou/mcp/declarative/configuration/YamlConfigurationLoader.java

@@ -0,0 +1,49 @@
+package com.github.codeboyzhou.mcp.declarative.configuration;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.nio.file.NoSuchFileException;
+import java.util.Objects;
+
+import static java.util.stream.Collectors.joining;
+
+public class YamlConfigurationLoader {
+
+    private static final Logger logger = LoggerFactory.getLogger(YamlConfigurationLoader.class);
+
+    private final ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
+
+    public McpServerConfiguration loadConfiguration() {
+        try {
+            McpServerConfiguration configuration = load("mcp-server.yml");
+            if (configuration == null) {
+                configuration = load("mcp-server.yaml");
+            }
+            return Objects.requireNonNullElseGet(configuration, McpServerConfiguration::defaultConfiguration);
+        } catch (IOException e) {
+            logger.error("Error loading configuration file, will use default configuration", e);
+            return McpServerConfiguration.defaultConfiguration();
+        }
+    }
+
+    public McpServerConfiguration load(String configFileName) throws IOException {
+        ClassLoader classLoader = YamlConfigurationLoader.class.getClassLoader();
+        try (InputStream inputStream = classLoader.getResourceAsStream(configFileName)) {
+            if (inputStream == null) {
+                throw new NoSuchFileException(configFileName);
+            }
+            InputStreamReader inputStreamReader = new InputStreamReader(inputStream);
+            BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
+            final String content = bufferedReader.lines().collect(joining(System.lineSeparator()));
+            return mapper.readValue(content, McpServerConfiguration.class);
+        }
+    }
+
+}

src/main/java/com/github/codeboyzhou/mcp/declarative/enums/ServerType.java

@@ -0,0 +1,6 @@
+package com.github.codeboyzhou.mcp.declarative.enums;
+
+public enum ServerType {
+    SYNC,
+    ASYNC
+}

src/main/java/com/github/codeboyzhou/mcp/declarative/exception/McpServerException.java

@@ -0,0 +1,13 @@
+package com.github.codeboyzhou.mcp.declarative.exception;
+
+public class McpServerException extends RuntimeException {
+
+    public McpServerException(String message) {
+        super(message);
+    }
+
+    public McpServerException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+}