File tree Expand file tree Collapse file tree 4 files changed +13
-24
lines changed
acp-core/src/main/java/com/agentclientprotocol/sdk
acp-websocket-jetty/src/main/java/com/agentclientprotocol/sdk/agent/transport Expand file tree Collapse file tree 4 files changed +13
-24
lines changed Original file line number Diff line number Diff line change 44
55package com .agentclientprotocol .sdk .agent .transport ;
66
7+ import java .time .Duration ;
78import java .io .BufferedReader ;
89import java .io .IOException ;
910import java .io .InputStream ;
@@ -277,12 +278,9 @@ else if (isClosing.get()) {
277278 @ Override
278279 public Mono <Void > sendMessage (JSONRPCMessage message ) {
279280 return Mono .zip (inboundReady .asMono (), outboundReady .asMono ()).then (Mono .defer (() -> {
280- if (outboundSink .tryEmitNext (message ).isSuccess ()) {
281- return Mono .empty ();
282- }
283- else {
284- return Mono .error (new RuntimeException ("Failed to enqueue message" ));
285- }
281+ outboundSink .emitNext (message ,
282+ Sinks .EmitFailureHandler .busyLooping (Duration .ofMillis (100 )));
283+ return Mono .empty ();
286284 }));
287285 }
288286
Original file line number Diff line number Diff line change @@ -260,12 +260,9 @@ private void handleIncomingErrors() {
260260
261261 @ Override
262262 public Mono <Void > sendMessage (JSONRPCMessage message ) {
263- if (this .outboundSink .tryEmitNext (message ).isSuccess ()) {
264- return Mono .empty ();
265- }
266- else {
267- return Mono .error (new RuntimeException ("Failed to enqueue message" ));
268- }
263+ this .outboundSink .emitNext (message ,
264+ Sinks .EmitFailureHandler .busyLooping (Duration .ofMillis (100 )));
265+ return Mono .empty ();
269266 }
270267
271268 /**
Original file line number Diff line number Diff line change @@ -199,12 +199,9 @@ private void startOutboundProcessing() {
199199 @ Override
200200 public Mono <Void > sendMessage (JSONRPCMessage message ) {
201201 return connectionReady .asMono ().then (Mono .defer (() -> {
202- if (outboundSink .tryEmitNext (message ).isSuccess ()) {
203- return Mono .empty ();
204- }
205- else {
206- return Mono .error (new RuntimeException ("Failed to enqueue message" ));
207- }
202+ outboundSink .emitNext (message ,
203+ Sinks .EmitFailureHandler .busyLooping (Duration .ofMillis (100 )));
204+ return Mono .empty ();
208205 }));
209206 }
210207
Original file line number Diff line number Diff line change @@ -224,12 +224,9 @@ private void startOutboundProcessing() {
224224 @ Override
225225 public Mono <Void > sendMessage (JSONRPCMessage message ) {
226226 return connectionReady .asMono ().then (Mono .defer (() -> {
227- if (outboundSink .tryEmitNext (message ).isSuccess ()) {
228- return Mono .empty ();
229- }
230- else {
231- return Mono .error (new RuntimeException ("Failed to enqueue message" ));
232- }
227+ outboundSink .emitNext (message ,
228+ Sinks .EmitFailureHandler .busyLooping (Duration .ofMillis (100 )));
229+ return Mono .empty ();
233230 }));
234231 }
235232
You can’t perform that action at this time.
0 commit comments