Make sure to flush output stream before writing the worker response

lucidsoftware · Nov 12, 2024 · 1ce90c3 · 1ce90c3
1 parent 4001ea6
commit 1ce90c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/scala/higherkindness/rules_scala/common/worker/WorkerMain.scala b/src/main/scala/higherkindness/rules_scala/common/worker/WorkerMain.scala
@@ -145,6 +145,7 @@ abstract class WorkerMain[S](stdin: InputStream = System.in, stdout: PrintStream
                   e.getCause() match {
                     // Task successfully cancelled
                     case cancelError: InterruptedException =>
+                      out.flush()
                       writeResponse(requestId, None, None, wasCancelled = true)
                       logVerbose(
                         s"Cancellation WorkResponse sent for request id: $requestId in response to an" +
@@ -165,6 +166,7 @@ abstract class WorkerMain[S](stdin: InputStream = System.in, stdout: PrintStream
 
                 // Task successfully cancelled
                 case Failure(e: CancellationException) =>
+                  out.flush()
                   writeResponse(requestId, None, None, wasCancelled = true)
                   logVerbose(
                     s"Cancellation WorkResponse sent for request id: $requestId in response to a" +