Fix test failures due to Span not being properly closed

Signed-off-by: Shreyansh Ray <[email protected]>
rayshrey · Oct 3, 2023 · 43f504a · 43f504a
1 parent 530c15d
commit 43f504a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...r/src/main/java/org/opensearch/action/support/replication/TransportReplicationAction.java b/...r/src/main/java/org/opensearch/action/support/replication/TransportReplicationAction.java
@@ -423,6 +423,7 @@ protected void handlePrimaryRequest(final ConcreteShardRequest<Request> request,
             new AsyncPrimaryAction(request, TraceableActionListener.create(listener, span, getTracer()), (ReplicationTask) task).run();
         } catch (RuntimeException e) {
             listener.onFailure(e);
+            span.endSpan();
         }
     }
 
@@ -708,6 +709,7 @@ protected void handleReplicaRequest(
                 .run();
         } catch (RuntimeException e) {
             listener.onFailure(e);
+            span.endSpan();
         }
     }