fix(java-emitter): check for null callback (#3830)

datahub-project · Jan 5, 2022 · daa20f8 · daa20f8
1 parent 5b36944
commit daa20f8
Showing 1 changed file with 12 additions and 8 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
@@ -198,19 +198,23 @@ public void completed(HttpResponse response) {
 
       @Override
       public void failed(Exception ex) {
-        try {
-          callback.onFailure(ex);
-        } catch (Exception e) {
-          log.error("Error executing user callback on failure.", e);
+        if (callback != null) {
+          try {
+            callback.onFailure(ex);
+          } catch (Exception e) {
+            log.error("Error executing user callback on failure.", e);
+          }
         }
       }
 
       @Override
       public void cancelled() {
-        try {
-          callback.onFailure(new RuntimeException("Cancelled"));
-        } catch (Exception e) {
-          log.error("Error executing user callback on failure due to cancellation.", e);
+        if (callback != null) {
+          try {
+            callback.onFailure(new RuntimeException("Cancelled"));
+          } catch (Exception e) {
+            log.error("Error executing user callback on failure due to cancellation.", e);
+          }
         }
       }
     };