[hotfix][python] Fix possible null pointer exception in PythonDriver

deepfield · Apr 13, 2021 · 67b2cb3 · 67b2cb3
1 parent 449048e
commit 67b2cb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flink-python/src/main/java/org/apache/flink/client/python/PythonDriver.java b/flink-python/src/main/java/org/apache/flink/client/python/PythonDriver.java
@@ -132,7 +132,7 @@ public static void main(String[] args) throws Throwable {
  }
  } finally {
  PythonEnvUtils.setGatewayServer(null);
- if (Runtime.getRuntime().removeShutdownHook(shutdownHook)) {
+ if (shutdownHook != null && Runtime.getRuntime().removeShutdownHook(shutdownHook)) {
  shutdownHook.run();
  }
  }