自定义MCP Server

1. MCP Server
引入依赖

1. <dependency>
2.     <groupId>org.springframework.boot</groupId>
3.     spring-boot-starter-webflux</artifactId>
4. </dependency>
5. <dependency>
6.     <groupId>org.springframework.ai</groupId>
7.     spring-ai-starter-mcp-server-webflux</artifactId>
8. </dependency>

复制代码

定义工具

1. @Service
2. public class DateService {
4.     private static final String[] WEEKDAY_NAMES = { "星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期日" };
6.     /**
7.      * 获取今天是星期几
8.      *
9.      * @return 星期描述
10.      */
11.     @Tool(description = "获取今天是星期几")
12.     public String getTodayWeekday() {
13.         System.out.println("getTodayWeekday 被调用");
14.         DayOfWeek dayOfWeek = LocalDate.now().getDayOfWeek();
15.         int index = dayOfWeek.getValue() - 1;
16.         if (index < 0 || index >= WEEKDAY_NAMES.length) {
17.             return "未知星期";
18.         }
19.         return String.format("今天是%s", WEEKDAY_NAMES[index]);
20.     }
22. }
25. @Service
26. public class NumberService {
28.     /**
29.      * 判断一个数字是奇数还是偶数
30.      *
31.      * @param number 要判断的数字
32.      * @return 判断结果，返回"奇数"或"偶数"
33.      */
34.     @Tool(description = "判断一个数字是奇数还是偶数")
35.     public String checkOddOrEven(@ToolParam(description = "要判断的数字") int number) {
36.         System.out.println("checkOddOrEven 被调用");
37.         if (number % 2 == 0) {
38.             return String.format("数字 %d 是偶数", number);
39.         } else {
40.             return String.format("数字 %d 是奇数", number);
41.         }
42.     }
43. }
46. @Service
47. public class MyToolService {
49.     /**
50.      * 打印 1-9 的九九乘法表
51.      *
52.      * @return 九九乘法表文本
53.      */
54.     @Tool(description = "打印标准九九乘法表")
55.     public String printMultiplicationTable() {
56.         System.out.println("printMultiplicationTable 被调用");
57.         StringBuilder table = new StringBuilder();
58.         for (int i = 1; i <= 9; i++) {
59.             for (int j = 1; j <= i; j++) {
60.                 table.append(String.format("%d×%d=%-2d", j, i, i * j));
61.                 if (j < i) {
62.                     table.append("  ");
63.                 }
64.             }
65.             if (i < 9) {
66.                 table.append('\n');
67.             }
68.         }
69.         return table.toString();
70.     }
71. }
74. @Service
75. public class BankService {
77.     private static final String BASE_URL = "https://www.abc.xyz.com/payBankInfo/getByBankNo";
79.     private final RestClient restClient;
80.     private final ObjectMapper objectMapper;
82.     public BankService(RestClient.Builder restClientBuilder, ObjectMapper objectMapper) {
83.         this.restClient = restClientBuilder
84.                 .baseUrl(BASE_URL)
85.                 .build();
86.         this.objectMapper = objectMapper;
87.     }
89.     @Tool(description = "通过行号查询开户行信息")
90.     public String queryBankInfo(@ToolParam(description = "开户行行号") String bankNo) {
91.         System.out.println("queryBankInfo 被调用");
92.         try {
93.             // 参数验证
94.             if (bankNo == null || bankNo.trim().isEmpty()) {
95.                 return "行号不能为空";
96.             }
98.             String responseBody = restClient.post()
99.                     .uri(uriBuilder -> uriBuilder.queryParam("bankNo", bankNo.trim()).build())
100.                     .contentType(MediaType.APPLICATION_JSON)
101.                     .accept(MediaType.APPLICATION_JSON)
102.                     .retrieve()
103.                     .body(String.class);
105.             // 尝试格式化 JSON
106.             try {
107.                 Object json = objectMapper.readValue(responseBody, Object.class);
108.                 return objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(json);
109.             } catch (Exception e) {
110.                 // 如果不是 JSON，直接返回原始响应
111.                 return responseBody;
112.             }
113.         } catch (RestClientResponseException e) {
114.             return String.format("请求失败: HTTP %d - %s", e.getStatusCode().value(), e.getStatusText());
115.         } catch (RestClientException e) {
116.             return String.format("请求异常: %s", e.getMessage());
117.         } catch (Exception e) {
118.             return String.format("处理异常: %s", e.getMessage());
119.         }
120.     }
121. }

复制代码

启动MCP服务

1. package com.example.springaimcpserver;
3. import com.example.springaimcpserver.service.BankService;
4. import com.example.springaimcpserver.service.DateService;
5. import com.example.springaimcpserver.service.MyToolService;
6. import com.example.springaimcpserver.service.NumberService;
7. import org.springframework.ai.tool.ToolCallback;
8. import org.springframework.ai.tool.ToolCallbackProvider;
9. import org.springframework.ai.tool.function.FunctionToolCallback;
10. import org.springframework.ai.tool.method.MethodToolCallbackProvider;
11. import org.springframework.boot.SpringApplication;
12. import org.springframework.boot.autoconfigure.SpringBootApplication;
13. import org.springframework.context.annotation.Bean;
15. import java.time.LocalDateTime;
17. @SpringBootApplication
18. public class SpringAiMcpServerApplication {
20.         public static void main(String[] args) {
21.                 SpringApplication.run(SpringAiMcpServerApplication.class, args);
22.         }
25.         /**
26.          * ToolCallback：单个工具，适合简单、独立的工具
27.          * ToolCallbackProvider：工具提供者，适合批量注册（通过扫描 @Tool 方法）
28.          * 两种方式都会被 Spring AI 识别并注册为 MCP 工具。
29.          */
31.         @Bean
32.         public ToolCallbackProvider customTools(MyToolService myToolService) {
33.                 return MethodToolCallbackProvider.builder().toolObjects(myToolService).build();
34.         }
36.         @Bean
37.         public ToolCallbackProvider customTools2(NumberService numberService, DateService dateService, BankService bankService) {
38.                 return MethodToolCallbackProvider.builder().toolObjects(numberService, dateService, bankService).build();
39.         }
42.         public record TextInput(String input) {
43.         }
45.         public record CalculatorInput(double a, double b, String operation) {
46.         }
48.         /**
49.          * 将输入字母转为大写
50.          */
51.         @Bean
52.         public ToolCallback toUpperCase() {
53.                 System.out.println("toUpperCase 被调用");
54.                 return FunctionToolCallback.builder("toUpperCase", (TextInput input) -> input.input().toUpperCase())
55.                                 .inputType(TextInput.class)
56.                                 .description("将字母转成大写")
57.                                 .build();
58.         }
60.         /**
61.          * 计算器工具 - 支持加减乘除运算
62.          */
63.         @Bean
64.         public ToolCallback calculator() {
65.                 System.out.println("calculator 被调用");
66.                 return FunctionToolCallback.builder("calculator", (CalculatorInput input) -> {
67.                                         double a = input.a();
68.                                         double b = input.b();
69.                                         String op = input.operation().toLowerCase().trim();
71.                                         return switch (op) {
72.                                                 case "add", "+" -> String.format("%.2f + %.2f = %.2f", a, b, a + b);
73.                                                 case "subtract", "-" -> String.format("%.2f - %.2f = %.2f", a, b, a - b);
74.                                                 case "multiply", "*", "×" -> String.format("%.2f × %.2f = %.2f", a, b, a * b);
75.                                                 case "divide", "/", "÷" -> {
76.                                                         if (b == 0) {
77.                                                                 yield "错误：除数不能为零";
78.                                                         }
79.                                                         yield String.format("%.2f ÷ %.2f = %.2f", a, b, a / b);
80.                                                 }
81.                                                 default -> "错误：不支持的操作符。支持的操作：add/+, subtract/-, multiply/*, divide//";
82.                                         };
83.                                 })
84.                                 .inputType(CalculatorInput.class)
85.                                 .description("执行基本的数学运算（加法、减法、乘法、除法）。支持的操作：add/+, subtract/-, multiply/*, divide//")
86.                                 .build();
87.         }
89.         /**
90.          * 获取当前时间
91.          */
92.         @Bean
93.         public ToolCallback getCurrentTime() {
94.                 System.out.println("getCurrentTime 被调用");
95.                 return FunctionToolCallback.builder("getCurrentTime",
96.                                                 () -> LocalDateTime.now().toString())
97.                                 .inputType(Void.class)
98.                                 .description("获取当前时间")
99.                                 .build();
100.         }
101. }

复制代码

2. MCP Client
pom.xml

1. spring:
2.   application:
3.     name: spring-ai-mcp-server
4.   main:
5.     banner-mode: off
6.   ai:
7.     mcp:
8.       server:
9.         name: my-spring-ai-mcp-server
10.         version: 1.0.0
11.         type: async
12.         sse-endpoint: /sse
13.         sse-message-endpoint: /mcp/message
14.         capabilities:
15.           tool: true
16.           resource: true
17.           prompt: true
18.           completion: true

复制代码

application.yml

1. <?xml version="1.0" encoding="UTF-8"?>
2. <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
3.         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
4.         <modelVersion>4.0.0</modelVersion>
5.         <parent>
6.                 <groupId>org.springframework.boot</groupId>
7.                 spring-boot-starter-parent</artifactId>
8.                 <version>3.5.7</version>
9.                 <relativePath/>
10.         </parent>
11.         <groupId>com.example</groupId>
12.         spring-ai-mcp-client</artifactId>
13.         <version>0.0.1-SNAPSHOT</version>
14.         <name>spring-ai-mcp-client</name>
15.         <description>spring-ai-mcp-client</description>
17.         <properties>
18.                 <java.version>17</java.version>
19.                 <spring-ai.version>1.0.3</spring-ai.version>
20.                 <spring-ai-alibaba.version>1.0.0.4</spring-ai-alibaba.version>
21.         </properties>
23.         <dependencies>
24.                 <dependency>
25.                         <groupId>org.springframework.ai</groupId>
26.                         spring-ai-starter-mcp-client-webflux</artifactId>
27.                 </dependency>
29.                 <dependency>
30.                         <groupId>com.alibaba.cloud.ai</groupId>
31.                         spring-ai-alibaba-starter-dashscope</artifactId>
32.                 </dependency>
33.         </dependencies>
35.         <dependencyManagement>
36.                 <dependencies>
37.                         <dependency>
38.                                 <groupId>org.springframework.ai</groupId>
39.                                 spring-ai-bom</artifactId>
40.                                 <version>${spring-ai.version}</version>
41.                                 <type>pom</type>
42.                                 <scope>import</scope>
43.                         </dependency>
44.                         <dependency>
45.                                 <groupId>com.alibaba.cloud.ai</groupId>
46.                                 spring-ai-alibaba-bom</artifactId>
47.                                 <version>${spring-ai-alibaba.version}</version>
48.                                 <type>pom</type>
49.                                 <scope>import</scope>
50.                         </dependency>
51.                 </dependencies>
52.         </dependencyManagement>
54.         <build>
55.                 <plugins>
56.                         <plugin>
57.                                 <groupId>org.springframework.boot</groupId>
58.                                 spring-boot-maven-plugin</artifactId>
59.                         </plugin>
60.                 </plugins>
61.         </build>
63. </project>

复制代码

测试工具调用

1. server:
2.   port: 8081
4. spring:
5.   application:
6.     name: my-spring-ai-mcp-client
7.   main:
8.     web-application-type: none
9.   ai:
10.     dashscope:
11.       api-key: sk-66217287a102487c4c52
12.       chat:
13.         enabled: true
14.         options:
15.           model: qwen3-max
16.           temperature: 0
17.     mcp:
18.       client:
19.         toolcallback:
20.           enabled: true
21.         sse:
22.           connections:
23.             server1:
24.               url: http://localhost:8080
25.               sse-endpoint: /sse
26.             mcp-chinese-fortune:
27.               url: https://mcp.api-inference.modelscope.net
28.               sse-endpoint: /d103803834594b/sse

复制代码

测试结果：
服务器端控制台输出：

1. package com.example.springaimcpclient;
3. import org.springframework.ai.chat.client.ChatClient;
4. import org.springframework.ai.tool.ToolCallbackProvider;
5. import org.springframework.boot.CommandLineRunner;
6. import org.springframework.boot.SpringApplication;
7. import org.springframework.boot.autoconfigure.SpringBootApplication;
8. import org.springframework.context.ConfigurableApplicationContext;
9. import org.springframework.context.annotation.Bean;
11. @SpringBootApplication
12. public class SpringAiMcpClientApplication {
14.         public static void main(String[] args) {
15.                 SpringApplication.run(SpringAiMcpClientApplication.class, args);
16.         }
19.         @Bean
20.         public CommandLineRunner predefinedQuestions(ChatClient.Builder chatClientBuilder, ToolCallbackProvider tools,
21.                                                                                                  ConfigurableApplicationContext context) {
23.                 // 打印所有可用的工具
24.                 System.out.println("\n========== 所有可用的工具 ==========");
25.                 var toolCallbacks = tools.getToolCallbacks();
26.                 System.out.println("工具总数: " + toolCallbacks.length);
27.                 for (int i = 0; i < toolCallbacks.length; i++) {
28.                         var toolDef = toolCallbacks[i].getToolDefinition();
29.                         System.out.println("\n【工具 " + (i + 1) + "】");
30.                         System.out.println("  名称: " + toolDef.name());
31.                         System.out.println("  描述: " + toolDef.description());
32.                 }
33.                 System.out.println("=====================================\n");
36.         return args -> {
38.                 String systemPrompt = """
39.                                         你是一个智能助手，拥有多个可用的工具。
40.                                         当用户提问时，你必须优先考虑使用可用的工具来回答问题。
41.                                         只有在没有合适的工具时，才使用你自己的知识来回答。
42.                                        
43.                                         重要规则：
44.                                         1. 如果有工具可以获取实时信息（如时间、日期），必须使用工具
45.                                         2. 如果有工具可以执行操作（如字符串转换），必须使用工具
46.                                         3. 使用工具后，将工具返回的结果展示给用户
47.                                         """;
48.                 var chatClient = chatClientBuilder
49.                                 .defaultSystem(systemPrompt)
50.                                 .defaultToolCallbacks(tools)
51.                                 .build();
53.                 // 测试问题1：当前时间（应该调用getCurrentTime工具）
54.                 System.out.println("\n>>> QUESTION: 当前时间是多少");
55.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("当前时间是多少").call().content());
57.                 // 测试问题2：字符串转大写（应该调用toUpperCase工具）
58.                 System.out.println("\n>>> QUESTION: 将abcd转成大写");
59.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("将abcd转成大写").call().content());
61.                 // 测试问题3：计算（应该调用calculator工具）
62.                 System.out.println("\n>>> QUESTION: 计算 123 加 456 等于多少");
63.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("计算 123 加 456 等于多少").call().content());
65.                 // 测试问题4：今天星期几（应该调用getTodayWeekday工具）
66.                 System.out.println("\n>>> QUESTION: 今天星期几");
67.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("今天星期几").call().content());
69.                 // 测试问题5：打印九九乘法表（应该调用printMultiplicationTable工具）
70.                 System.out.println("\n>>> QUESTION: 打印九九乘法表");
71.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("打印九九乘法表").call().content());
73.                 // 测试问题6：查询开户行信息（应该调用queryBankInfo工具）
74.                 System.out.println("\n>>> QUESTION: 查询行号为303100000629的开户行信息");
75.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("查询行号为303100000629的开户行信息").call().content());
77.                 // 测试问题7：判断奇偶数（应该调用checkOddOrEven工具）
78.                 System.out.println("\n>>> QUESTION: 数字16是奇数还是偶数");
79.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("数字16是奇数还是偶数").call().content());
81.                 // 测试问题8：帮我算下命，出生时间 2015年 10月19日8点
82.                 System.out.println("\n>>> QUESTION: 帮我算下命，出生时间 2015年 10月19日8点");
83.                 System.out.println(">>> ASSISTANT: " + chatClient.prompt("帮我算下命，出生时间 2015年 10月19日8点").call().content());
84.                
85.                 context.close();
86.         };
87.         }
88. }

复制代码

客户端控制台输出：

1. getTodayWeekday 被调用
2. printMultiplicationTable 被调用
3. queryBankInfo 被调用
4. checkOddOrEven 被调用

复制代码

抓包截图

SSE方式通信流程：

参考文档：
https://docs.spring.io/spring-ai/reference/getting-started.html
https://github.com/spring-ai-community/awesome-spring-ai
 

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！