modelcontextprotocol/java-sdk
Watch
1
0
Fork
You've already forked java-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(mcp) Increase the server stdio shutdown timeout

Browse source
This commit is contained in:
Christian Tzolov 2024-12-30 09:08:56 +01:00
commit 47b09165c5
2 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
mcp/src/main/java/org/springframework/ai/mcp/server/transport/SseServerTransport.java
View file

@ -6,7 +6,6 @@ import java.util.List;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Function;
import java.util.stream.Collectors;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;

2
mcp/src/main/java/org/springframework/ai/mcp/server/transport/StdioServerTransport.java
View file

@ -252,7 +252,7 @@ public class StdioServerTransport implements McpTransport {
// First complete the sinks to stop processing
inboundSink.tryEmitComplete();
outboundSink.tryEmitComplete();
return Mono.delay(Duration.ofMillis(100));
return Mono.delay(Duration.ofMillis(200));
})).then(Mono.fromRunnable(() -> {
try {
// Dispose schedulers first