refactor(server): Refactor Immutable class to use Java records and streamline McpServerComponentRegister registration methods

Author: codeboyzhou

src/main/java/com/github/codeboyzhou/mcp/declarative/common/Immutable.java

@@ -1,13 +1,12 @@
 package com.github.codeboyzhou.mcp.declarative.common;
 
-public final class Immutable<T> {
-  private final T value;
+public record Immutable<T>(T value) {
 
-  private Immutable(T value) {
-    this.value = value;
+  public static <T> Immutable<T> of(T value) {
+    return new Immutable<>(value);
   }
 
-  public static <T> T of(T value) {
-    return new Immutable<>(value).value;
+  public T get() {
+    return value;
   }
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/component/McpServerComponentRegister.java

@@ -7,15 +7,23 @@
 import com.github.codeboyzhou.mcp.declarative.common.InjectorProvider;
 import com.google.inject.Injector;
 import io.modelcontextprotocol.server.McpSyncServer;
+import java.lang.annotation.Annotation;
 import java.lang.reflect.Method;
 import java.util.Set;
+import java.util.function.BiConsumer;
 import org.reflections.Reflections;
 
 public final class McpServerComponentRegister {
 
-  private final McpSyncServer server;
+  private final Injector injector;
+
+  private final Reflections reflections;
+
+  private final Immutable<McpSyncServer> server;
 
   public McpServerComponentRegister(McpSyncServer server) {
+    this.injector = InjectorProvider.getInstance().getInjector();
+    this.reflections = injector.getInstance(Reflections.class);
     this.server = Immutable.of(server);
   }
 
@@ -24,38 +32,21 @@ public static McpServerComponentRegister of(McpSyncServer server) {
   }
 
   public void registerComponents() {
-    registerResources();
-    registerPrompts();
-    registerTools();
+    register(McpResource.class, McpServerResourceFactory.class, McpSyncServer::addResource);
+    register(McpPrompt.class, McpServerPromptFactory.class, McpSyncServer::addPrompt);
+    register(McpTool.class, McpServerToolFactory.class, McpSyncServer::addTool);
   }
 
-  private void registerResources() {
-    Injector injector = InjectorProvider.getInstance().getInjector();
-    McpServerResourceFactory resourceFactory = injector.getInstance(McpServerResourceFactory.class);
-    Reflections reflections = injector.getInstance(Reflections.class);
-    Set<Method> resourceMethods = reflections.getMethodsAnnotatedWith(McpResource.class);
-    for (Method method : resourceMethods) {
-      server.addResource(resourceFactory.create(method.getDeclaringClass(), method));
-    }
-  }
-
-  private void registerPrompts() {
-    Injector injector = InjectorProvider.getInstance().getInjector();
-    McpServerPromptFactory promptFactory = injector.getInstance(McpServerPromptFactory.class);
-    Reflections reflections = injector.getInstance(Reflections.class);
-    Set<Method> promptMethods = reflections.getMethodsAnnotatedWith(McpPrompt.class);
-    for (Method method : promptMethods) {
-      server.addPrompt(promptFactory.create(method.getDeclaringClass(), method));
-    }
-  }
+  private <T> void register(
+      Class<? extends Annotation> annotationClass,
+      Class<? extends McpServerComponentFactory<T>> factoryClass,
+      BiConsumer<McpSyncServer, T> serverAddComponent) {
 
-  private void registerTools() {
-    Injector injector = InjectorProvider.getInstance().getInjector();
-    McpServerToolFactory toolFactory = injector.getInstance(McpServerToolFactory.class);
-    Reflections reflections = injector.getInstance(Reflections.class);
-    Set<Method> toolMethods = reflections.getMethodsAnnotatedWith(McpTool.class);
-    for (Method method : toolMethods) {
-      server.addTool(toolFactory.create(method.getDeclaringClass(), method));
+    Set<Method> methods = reflections.getMethodsAnnotatedWith(annotationClass);
+    McpServerComponentFactory<T> factory = injector.getInstance(factoryClass);
+    for (Method method : methods) {
+      T component = factory.create(method.getDeclaringClass(), method);
+      serverAddComponent.accept(server.get(), component);
     }
   }
 }