MCP 模型上下文协议：AI 应用互操作的 Type-C 时代

package com.example.mcp.service;

import org.springframework.ai.mcp.server.tool.Tool;
import org.springframework.stereotype.Service;

/**
 * MCP 工具服务示例
 *
 * @author MamimiJa Nai
 */
@Service
public class WeatherService {

    /**
     * 获取指定城市的天气信息
     *
     * @param city 城市名称
     * @return 天气描述
     */
    @Tool(
        name = "get_weather",
        description = "获取指定城市的当前天气信息，包括温度、天气状况和湿度"
    )
    public String getWeather(String city) {
        // 实际项目中应该调用真实天气 API
        return String.format(
            "🌍 %s 的天气：晴天，温度 22°C，湿度 45%%",
            city
        );
    }

    /**
     * 获取城市列表
     *
     * @return 支持的城市列表
     */
    @Tool(
        name = "list_cities",
        description = "获取所有支持天气查询的城市列表"
    )
    public String listCities() {
        return """
            📍 支持的城市：
            - 北京
            - 上海
            - 广州
            - 深圳
            - 杭州
            """;
    }
}

package com.example.mcp.config;

import com.example.mcp.service.WeatherService;
import org.springframework.ai.mcp.server.ToolCallbackProvider;
import org.springframework.ai.mcp.server.config.McpServerProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * MCP 服务器配置
 *
 * @author MamimiJa Nai
 */
@Configuration
public class McpServerConfig {

    /**
     * 注册工具回调提供者
     */
    @Bean
    public ToolCallbackProvider toolCallbackProvider(
            WeatherService weatherService,
            McpServerProperties properties) {

        return ToolCallbackProvider.builder()
            .toolObjects(weatherService)  // 注册工具服务
            .serverProperties(properties)  // MCP 服务器属性
            .build();
    }
}

@Configuration
public class McpSecurityConfig {

    /**
     * 配置工具访问权限
     */
    @Bean
    public ToolCallbackProvider secureToolProvider(
            WeatherService weatherService) {

        return ToolCallbackProvider.builder()
            .toolObjects(weatherService)
            // 添加访问控制
            .accessControl((tool, principal) -> {
                // 只允许特定角色访问
                return principal.hasRole("AI_USER");
            })
            .build();
    }
}

@Service
public class OptimizedWeatherService {

    @Tool(name = "get_weather_fast")
    @Cacheable(value = "weather", key = "#city")  // 添加缓存
    public String getWeather(String city) {
        // 实现逻辑
    }

    @Tool(name = "get_weather_batch")
    public Map<String, String> getWeatherBatch(List<String> cities) {
        // 批量处理，减少调用次数
        return cities.stream()
            .collect(Collectors.toMap(
                city -> city,
                this::getWeather
            ));
    }
}

@Service
public class KnowledgeBaseService {

    @Tool(
        name = "search_docs",
        description = "在企业知识库中搜索相关文档"
    )
    @PreAuthorize("hasRole('EMPLOYEE')")  # 权限控制
    @AuditLog  # 审计日志
    public List<Document> searchDocuments(
        @ToolParam(description = "搜索关键词") String query,
        @ToolParam(description = "最大结果数", defaultValue = "5") int limit
    ) {
        // 1. 参数验证
        validateQuery(query);

        // 2. 调用搜索服务
        List<Document> results = searchService.search(query, limit);

        // 3. 过滤敏感信息
        return filterSensitiveInfo(results);
    }

    @Tool(name = "get_doc_summary")
    public String getDocumentSummary(
        @ToolParam(description = "文档ID") String docId
    ) {
        // 实现逻辑
    }
}