[BEAM-7412] shut down executor service in fn harness

apache · May 30, 2019 · 43660a0 · 43660a0
1 parent e8666a4
commit 43660a0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/FnHarness.java b/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/FnHarness.java
@@ -19,6 +19,7 @@
 
 import java.util.EnumMap;
 import java.util.List;
+import java.util.concurrent.ExecutorService;
 import java.util.function.Function;
 import org.apache.beam.fn.harness.control.AddHarnessIdInterceptor;
 import org.apache.beam.fn.harness.control.BeamFnControlClient;
@@ -203,7 +204,9 @@ public static void main(
       JvmInitializers.runBeforeProcessing(options);
 
       LOG.info("Entering instruction processing loop");
-      control.processInstructionRequests(options.as(GcsOptions.class).getExecutorService());
+      ExecutorService executorService = options.as(GcsOptions.class).getExecutorService();
+      control.processInstructionRequests(executorService);
+      executorService.shutdown();
     } finally {
       System.out.println("Shutting SDK harness down.");
     }