Merge pull request #548 from SDCISA-13736-ApplyFindings-2024011001-1312

swisspost · Jan 10, 2024 · 6978e34 · 6978e34
2 parents acafdf6 + f4d685d
commit 6978e34
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/gateleen-logging/src/main/java/org/swisspush/gateleen/logging/LoggingWriteStream.java b/gateleen-logging/src/main/java/org/swisspush/gateleen/logging/LoggingWriteStream.java
@@ -3,6 +3,7 @@
 import io.vertx.core.AsyncResult;
 import io.vertx.core.Future;
 import io.vertx.core.Handler;
+import io.vertx.core.Promise;
 import io.vertx.core.buffer.Buffer;
 import io.vertx.core.streams.WriteStream;
 
@@ -36,15 +37,17 @@ public WriteStream<Buffer> exceptionHandler(Handler<Throwable> handler) {
 
     @Override
     public Future<Void> write(Buffer data) {
+        Promise<Void> p = Promise.promise();
         write(data, event -> {
-
+            if( event.failed() ) p.fail(event.cause());
+            else p.complete();
         });
-        return Future.succeededFuture();
+        return p.future();
     }
 
     @Override
     public void write(Buffer data, Handler<AsyncResult<Void>> handler) {
-        wrappedWriteStream.write(data);
+        wrappedWriteStream.write(data, handler);
         if (isRequest) {
             loggingHandler.appendRequestPayload(data);
         } else {