Data Generator Java MCP Server

Data Generator Java MCP 服务

一个用于数据生成的 Spring Boot 应用，实现了 Model Completion Protocol（MCP）服务。此项目演示了如何使用 Spring AI 创建自定义 MCP 服务，并将其与 AI 模型集成，以根据提示词生成数据。

概述

此应用是一个提供数据生成工具的 MCP 服务。它使用 Spring AI 的 MCP 服务功能，将 Java 方法公开为可供 AI 模型调用的工具。该服务配置为以同步模式运行。

功能

• 实现 Model Completion Protocol（MCP）服务
• 提供数据生成工具
• 支持同步执行模式
• 包含用于外部 API 集成的 REST 客户端
• 实现简单的 User 数据模型

前提条件

• Java 21 或更高版本
• Maven 3.9+（或使用附带的 Maven wrapper）
• 具备 Spring Boot 和 Spring AI 的基础知识

快速开始

构建应用

1. 克隆仓库：

   git clone <repository-url>
   cd data-generator-java-mcp-server
   

2. 使用 Maven 构建应用：

   ./mvnw clean install
   

运行应用

使用 Maven wrapper 运行应用：

./mvnw spring-boot:run

MCP 服务将启动并准备好接收请求。

配置

MCP 服务配置

MCP 服务在 application.properties 文件中配置：

# Application name
spring.application.name=data-generator-java-mcp-server

# MCP server name
spring.ai.mcp.server.name=data-generator-java-mcp-server

# MCP server type (SYNC or ASYNC)
spring.ai.mcp.server.type=SYNC

# Disable Spring Boot banner
spring.main.banner-mode=off

# Run as a non-web application
spring.main.web-application-type=none

自定义配置

你可以通过修改 McpServerConfig 类来自定义 MCP 服务：

@Configuration
public class McpServerConfig {

    @Bean
    public ToolCallbackProvider dataGeneratorToolCallbackProvider(DataGeneratorService dataGeneratorService) {
        return MethodToolCallbackProvider.builder()
                .toolObjects(dataGeneratorService)
                .build();
    }
}

可用工具

MCP 服务提供以下工具：

1. generateData - 根据提供的提示词生成数据

   • 参数：prompt（String）
   • 返回：以 String 形式返回生成的数据

2. getUserData - 根据提供的 ID 和年龄生成用户数据

   • 参数：
     • id（String）
     • age（int）
   • 返回：包含 ID、姓名和年龄的 User 对象

扩展应用

要向 MCP 服务添加新工具：

1. 创建新的服务类，或扩展现有的 DataGeneratorService
2. 添加使用 @Tool 注解的方法
3. 在 McpServerConfig 类中注册该服务

示例：

@Service
public class MyCustomService {

    @Tool(
        name = "myCustomTool",
        description = "Description of what the tool does"
    )
    public SomeReturnType myCustomTool(String param1, int param2) {
        // Implementation
        return result;
    }
}

然后在配置中注册：

@Bean
public ToolCallbackProvider myCustomToolCallbackProvider(MyCustomService myCustomService) {
    return MethodToolCallbackProvider.builder()
            .toolObjects(myCustomService)
            .build();
}

与 AI 模型集成

此 MCP 服务可以与支持 Model Completion Protocol 的 AI 模型集成。AI 模型可以调用此服务公开的工具，在处理过程中生成数据。