slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(LocalRunnerClient): consolidate MCP client transport params

Browse Source 
Context:
Group HTTP and STDIO transport parameter variants under a sealed internal transport parameter hierarchy.
This commit is contained in:
slhafzjw
2025-12-29 18:48:53 +08:00
parent db3435fccf
commit 0e741802d1
1 changed files with 23 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
Partner-Main/src/main/java/work/slhaf/partner/core/action/runner/LocalRunnerClient.java
View File

@@ -242,14 +242,14 @@ public class LocalRunnerClient extends RunnerClient {
private McpClientTransport createTransport(McpClientTransportParams mcpClientTransportParams) { private McpClientTransport createTransport(McpClientTransportParams mcpClientTransportParams) {
return switch (mcpClientTransportParams) { return switch (mcpClientTransportParams) {
case StdioMcpClientTransportParams params -> { case McpClientTransportParams.Stdio params -> {
ServerParameters serverParameters = ServerParameters.builder(params.command) ServerParameters serverParameters = ServerParameters.builder(params.command)
.env(params.env) .env(params.env)
.args(params.args) .args(params.args)
.build(); .build();
yield new StdioClientTransport(serverParameters, McpJsonMapper.getDefault()); yield new StdioClientTransport(serverParameters, McpJsonMapper.getDefault());
} }
case HttpMcpClientTransportParams params -> { case McpClientTransportParams.Http params -> {
McpSyncHttpClientRequestCustomizer customizer = (builder, method, endpoint, body, context) -> { McpSyncHttpClientRequestCustomizer customizer = (builder, method, endpoint, body, context) -> {
params.headers.forEach(builder::setHeader); params.headers.forEach(builder::setHeader);
}; };
@@ -527,37 +527,37 @@ public class LocalRunnerClient extends RunnerClient {
} }
} }
private sealed abstract static class McpClientTransportParams permits HttpMcpClientTransportParams, StdioMcpClientTransportParams { private sealed abstract static class McpClientTransportParams permits McpClientTransportParams.Http, McpClientTransportParams.Stdio {
private final int timeout; private final int timeout;
private McpClientTransportParams(int timeout) { private McpClientTransportParams(int timeout) {
this.timeout = timeout; this.timeout = timeout;
} }
}
private final static class HttpMcpClientTransportParams extends McpClientTransportParams { private final static class Http extends McpClientTransportParams {
private final String baseUri; private final String baseUri;
private final String endpoint; private final String endpoint;
private final Map<String, String> headers; private final Map<String, String> headers;
private HttpMcpClientTransportParams(int timeout, String baseUri, String endpoint, Map<String, String> header) { private Http(int timeout, String baseUri, String endpoint, Map<String, String> header) {
super(timeout); super(timeout);
this.baseUri = baseUri; this.baseUri = baseUri;
this.endpoint = endpoint; this.endpoint = endpoint;
this.headers = header; this.headers = header;
}
} }
}
private final static class StdioMcpClientTransportParams extends McpClientTransportParams { private final static class Stdio extends McpClientTransportParams {
private final String command; private final String command;
private final Map<String, String> env; private final Map<String, String> env;
private final List<String> args; private final List<String> args;
private StdioMcpClientTransportParams(int timeout, String command, Map<String, String> env, List<String> args) { private Stdio(int timeout, String command, Map<String, String> env, List<String> args) {
super(timeout); super(timeout);
this.command = command; this.command = command;
this.env = env; this.env = env;
this.args = args; this.args = args;
}
} }
} }