Fix condition to drop messages in HttpResponseFlowAdapter

restatedev · Nov 29, 2024 · 9a6e902 · 9a6e902
1 parent 858e0d2
commit 9a6e902
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk-http-vertx/src/main/java/dev/restate/sdk/http/vertx/HttpResponseFlowAdapter.java b/sdk-http-vertx/src/main/java/dev/restate/sdk/http/vertx/HttpResponseFlowAdapter.java
@@ -40,7 +40,7 @@ public void onSubscribe(Flow.Subscription subscription) {
 
   @Override
   public void onNext(ByteBuffer byteBuffer) {
-    if (this.httpServerResponse.closed()) {
+    if (this.httpServerResponse.ended()) {
       cancelSubscription();
       return;
     }
@@ -87,8 +87,8 @@ private void endResponse() {
   }
 
   private void cancelSubscription() {
-    LOG.trace("Cancelling subscription");
     if (this.outputSubscription != null) {
+      LOG.trace("Cancelling subscription");
       Flow.Subscription outputSubscription = this.outputSubscription;
       this.outputSubscription = null;
       outputSubscription.cancel();