[REEF-1721] Close the job submission handler when REEF client closes.
Also add some extra logging in the `REEF.close()` call.
JIRA: [REEF-1721](https://issues.apache.org/jira/browse/REEF-1721)
Closes #1234
diff --git a/lang/java/reef-common/src/main/java/org/apache/reef/runtime/common/client/REEFImplementation.java b/lang/java/reef-common/src/main/java/org/apache/reef/runtime/common/client/REEFImplementation.java
index 80c7924..ac52922 100644
--- a/lang/java/reef-common/src/main/java/org/apache/reef/runtime/common/client/REEFImplementation.java
+++ b/lang/java/reef-common/src/main/java/org/apache/reef/runtime/common/client/REEFImplementation.java
@@ -39,6 +39,7 @@
import javax.inject.Inject;
import java.util.Set;
+import java.util.logging.Level;
import java.util.logging.Logger;
/**
@@ -89,8 +90,21 @@
@Override
public void close() {
+
+ LOG.log(Level.FINE, "Close REEF: shutdown jobs");
this.runningJobs.closeAllJobs();
+
+ LOG.log(Level.FINE, "Close REEF: shutdown client");
this.clientWireUp.close();
+
+ LOG.log(Level.FINE, "Close REEF: shutdown job submitter");
+ try {
+ this.jobSubmissionHandler.close();
+ } catch (final Exception ex) {
+ LOG.log(Level.WARNING, "Could not shutdown job submitter", ex);
+ }
+
+ LOG.log(Level.FINE, "Close REEF: done");
}
@Override
@@ -137,6 +151,4 @@
@NamedParameter(doc = "The driver remote identifier.")
public static final class DriverRemoteIdentifier implements Name<String> {
}
-
-
}
