From 93ed13a1897bfcd6f339c6622b2309e07ea5a67e Mon Sep 17 00:00:00 2001
From: maqi <maqi@dtstack.com>
Date: Thu, 16 Apr 2020 15:35:43 +0800
Subject: [PATCH] add security config

---
 .../flink/sql/launcher/factory/YarnClusterClientFactory.java  | 4 ++++
 1 file changed, 4 insertions(+)

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
index 3005af2d2..58d32cfac 100644
--- 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);