fix(spark-lineage): default timeout for future responses

datahub-project · Jul 20, 2024 · fae44d1 · fae44d1
1 parent 96fb2f7
commit fae44d1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/metadata-integration/java/datahub-client/src/main/java/datahub/client/rest/RestEmitter.java b/metadata-integration/java/datahub-client/src/main/java/datahub/client/rest/RestEmitter.java
@@ -250,6 +250,7 @@ public void completed(SimpleHttpResponse response) {
 
           @Override
           public void failed(Exception ex) {
+            responseLatch.countDown();
             if (callback != null) {
               try {
                 callback.onFailure(ex);
@@ -261,6 +262,7 @@ public void failed(Exception ex) {
 
           @Override
           public void cancelled() {
+            responseLatch.countDown();
             if (callback != null) {
               try {
                 callback.onFailure(new RuntimeException("Cancelled"));
@@ -344,6 +346,7 @@ public void completed(SimpleHttpResponse response) {
 
           @Override
           public void failed(Exception ex) {
+            responseLatch.countDown();
             if (callback != null) {
               try {
                 callback.onFailure(ex);
@@ -355,6 +358,7 @@ public void failed(Exception ex) {
 
           @Override
           public void cancelled() {
+            responseLatch.countDown();
             if (callback != null) {
               try {
                 callback.onFailure(new RuntimeException("Cancelled"));