add security config

DTStack · Apr 16, 2020 · 93ed13a · 93ed13a
1 parent 0a7f3f9
commit 93ed13a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/launcher/src/main/java/com/dtstack/flink/sql/launcher/factory/YarnClusterClientFactory.java b/launcher/src/main/java/com/dtstack/flink/sql/launcher/factory/YarnClusterClientFactory.java
@@ -23,6 +23,8 @@
 import org.apache.flink.configuration.ConfigConstants;
 import org.apache.flink.configuration.Configuration;
 import org.apache.flink.core.fs.FileSystem;
+import org.apache.flink.runtime.security.SecurityConfiguration;
+import org.apache.flink.runtime.security.SecurityUtils;
 import org.apache.flink.util.FileUtils;
 import org.apache.flink.util.function.FunctionUtils;
 import org.apache.flink.yarn.YarnClientYarnClusterInformationRetriever;
@@ -54,6 +56,8 @@ public ClusterDescriptor createClusterDescriptor(String yarnConfDir, Configurati
                 flinkConfig.setString(ConfigConstants.PATH_HADOOP_CONFIG, yarnConfDir);
                 FileSystem.initialize(flinkConfig, null);
 
+                SecurityUtils.install(new SecurityConfiguration(flinkConfig));
+
                 YarnConfiguration yarnConf = getYarnConf(yarnConfDir);
                 YarnClient yarnClient = YarnClient.createYarnClient();
                 yarnClient.init(yarnConf);