modelcontextprotocol / modelcontextprotocol/java-sdk

Request customizers can't tell which connection a request belongs to

未关闭
#1,074 3 条评论 0 个 reaction 已指派 1 人 在 GitHub 查看

@Kehrlann 已经在做这个了。

开始于 2026年8月7日。

area/client area/transport P3 waiting for user
主要语言
Java
星标
3.7k
派生
1.1k
平均合并
1 天 15 小时
30 天内合并 PR
9

描述

Expected Behavior

A request customizer should be able to find out which configured connection the request it is customizing belongs to.

The cheaper option is a well-known key in the transport context, seeded by the transport, so no signature changes:

Publisher<HttpRequest.Builder> customize(HttpRequest.Builder builder, String method, URI endpoint,
        String body, McpTransportContext context) {
    String connection = (String) context.get(McpTransportContext.CONNECTION_NAME);
    builder.setHeader("Authorization", tokenFor(connection));
    return Mono.just(builder);
}

The transport already knows the name when it is built, so it would set it once in the builder and merge it into whatever context the caller supplied.

The other option is to put it in the signature:

Publisher<HttpRequest.Builder> customize(String connectionName, HttpRequest.Builder builder, String method,
        URI endpoint, String body, McpTransportContext context);

This is breaking, but it lines up with McpClientCustomizer#customize(String name, T spec) in Spring AI, which already takes the name first. If the interface is going to change shape at some point anyway, that's the more consistent end state.

Current Behavior

A request customizer runs for every outbound HTTP request on a transport, but isn't told which connection the request belongs to. On main at fd00498:

// McpAsyncHttpClientRequestCustomizer:27
Publisher<HttpRequest.Builder> customize(HttpRequest.Builder builder, String method, URI endpoint,
        @Nullable String body, McpTransportContext context);

The only discriminator available is the endpoint URI. There is no connection name in the context either: nothing in mcp-core defines or sets one, and the transport passes the caller's context through untouched. HttpClientStreamableHttpTransport reads it the same way on all four request paths, at lines 205, 291, 430, and 514:

var transportContext = ctx.getOrDefault(McpTransportContext.KEY, McpTransportContext.EMPTY);
return Mono.from(this.httpRequestCustomizer.customize(builder, "POST", uri, jsonBody, transportContext));

So a customizer that needs per-connection behavior has to build its own URL-to-name map and reverse-match on the endpoint for every request. That map has to be assembled by re-reading the same configuration that the framework already parsed, and the match breaks as soon as a URL is templated, redirected, or differs by a trailing slash.

The name is known at the point the customizer is installed, since the transport is built per connection. It just isn't carried through to the call.

Context

Callers configure connections by name. In Spring AI, for example:

spring.ai.mcp.client.streamable-http.connections:
  first-server:
    url: https://...
  second-server:
    url: https://...

Anything per-connection needs that name: a different credential or audience per server, per-connection metrics, per-connection logging. We hit this with credentials, where each MCP server is a distinct audience, and the token must be minted for the correct one.

Alternatives considered:

  • One customizer instance per connection, closing over the name, installed by whoever builds the transport. This is what we do today. It works, but it only works for the party that owns the builder, and it interacts badly with the single-customizer-slot problem (https://github.com/modelcontextprotocol/java-sdk/issues/1073).
  • A URL to name map inside the customizer, reverse-matched on endpoint. Fragile for the reasons above, and it duplicates configuration parsing.
  • Putting the name into the caller's context at the call site. Requires every caller to know which connection a given client operation will reach, which they generally don't.

Is there a reason the connection name was deliberately omitted from the customizer contract? I might be missing something.

贡献指南

打开贡献指南

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。