Fix tracing

restatedev · Mar 20, 2023 · 81c3a26 · 81c3a26
1 parent 2ac7894
commit 81c3a26
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sdk-core-impl/src/main/java/dev/restate/sdk/core/impl/InvocationStateMachine.java b/sdk-core-impl/src/main/java/dev/restate/sdk/core/impl/InvocationStateMachine.java
@@ -196,6 +196,7 @@ void close() {
       this.readyResultPublisher.abort(ProtocolException.CLOSED);
       this.sideEffectAckPublisher.abort(ProtocolException.CLOSED);
       this.entriesQueue.abort(ProtocolException.CLOSED);
+      this.span.end();
     }
   }
 
@@ -214,6 +215,7 @@ void fail(Throwable cause) {
       this.readyResultPublisher.abort(cause);
       this.sideEffectAckPublisher.abort(cause);
       this.entriesQueue.abort(cause);
+      this.span.end();
     }
   }