[Functions] Check executor null when closing the FileSource (apache#1…

…5247)
drianpauL · Apr 21, 2022 · 06ba587 · 06ba587
1 parent e4a8de1
commit 06ba587
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/pulsar-io/file/src/main/java/org/apache/pulsar/io/file/FileSource.java b/pulsar-io/file/src/main/java/org/apache/pulsar/io/file/FileSource.java
@@ -57,13 +57,15 @@ public void open(Map<String, Object> config, SourceContext sourceContext) throws
 
     @Override
     public void close() throws Exception {
-        executor.shutdown();
-        try {
-            if (!executor.awaitTermination(800, TimeUnit.MILLISECONDS)) {
+        if (executor != null) {
+            executor.shutdown();
+            try {
+                if (!executor.awaitTermination(800, TimeUnit.MILLISECONDS)) {
+                    executor.shutdownNow();
+                }
+            } catch (InterruptedException e) {
                 executor.shutdownNow();
             }
-        } catch (InterruptedException e) {
-            executor.shutdownNow();
         }
     }
 }