@Bean
@ConditionalOnMissingBean
public WebFluxSseServerTransportProvider webFluxTransport(ObjectProvider<ObjectMapper> objectMapperProvider,
		McpServerProperties serverProperties,
		ObjectProvider<McpServerSessionListener<ServerRequest>> mcpServerSessionListeners) {
	ObjectMapper objectMapper = objectMapperProvider.getIfAvailable(ObjectMapper::new);
	WebFluxSseServerTransportProvider webFluxSseServerTransportProvider = new WebFluxSseServerTransportProvider(
			objectMapper, serverProperties.getSseMessageEndpoint(), serverProperties.getSseEndpoint());
	webFluxSseServerTransportProvider.setMcpServerSessionListener(mcpServerSessionListeners.getIfAvailable());
	return webFluxSseServerTransportProvider;
}

@Configuration
public class ToolProviderAutoConfiguration {

    public static final Map<String, HttpHeaders> sessionHeaders = new ConcurrentHashMap<>();

    @Bean
    public ToolCallbackProvider toolCallbackProvider(WeatherToolProvider weatherToolProvider) {
        return MethodToolCallbackProvider.builder().toolObjects(weatherToolProvider).build();
    }

    @Bean
    public McpServerSessionListener<ServerRequest> mcpServerSessionListener() {
        return new McpServerSessionListener<>() {

            @Override
            public void onConnection(McpServerSession session, ServerRequest request) {
                HttpHeaders httpHeaders = request.headers().asHttpHeaders();
                sessionHeaders.put(session.getId(), httpHeaders);
            }

            @Override
            public void onMessage(McpServerSession session, ServerRequest request) {
            }
        };
    }
}

@Service
public class WeatherToolProvider {

    @Tool(name = "query_weather", description = "查询天气信息。")
    public String queryWeather(String city,
                               ToolContext toolContext) {
        Object o = toolContext.getContext().get("exchange");
        if (o instanceof McpSyncServerExchange) {
            String sessionId = ((McpSyncServerExchange) o).getSessionId();
            log.info("sessionId: {}", sessionId);

            HttpHeaders httpHeaders = ToolProviderAutoConfiguration.sessionHeaders.get(sessionId);
            log.info("httpHeaders: {}", httpHeaders);
        }

        return "今天晴转多云，18-26°C，东南风3-4级。";
    }
}