瀏覽代碼

es增加权限校验

lixing 4 年之前
父節點
當前提交
59c81fc2c0

+ 7 - 2
src/main/java/com/persagy/apm/energy/report/workorder/config/ElasticsearchCustomConfig.java

@@ -61,14 +61,19 @@ public class ElasticsearchCustomConfig extends ElasticsearchConfigurationSupport
     public Client elasticsearchClient() throws UnknownHostException {
         String clusterName = environment.getProperty("spring.data.elasticsearch.cluster-name");
         String clusterNodes = environment.getProperty("spring.data.elasticsearch.cluster-nodes");
+        String user = environment.getProperty("elasticsearch.userName");
+        String password = environment.getProperty("elasticsearch.password");
         if (StringUtils.isBlank(clusterName)) {
             clusterName = "elasticsearch";
         }
         if (StringUtils.isBlank(clusterNodes)) {
             clusterNodes = "127.0.0.1:9300";
         }
-        Settings settings = Settings.builder().put("cluster.name", clusterName).build();
-        TransportClient client = new PreBuiltTransportClient(settings);
+        Settings.Builder builder = Settings.builder().put("cluster.name", clusterName);
+        if (StringUtils.isNotBlank(user) && StringUtils.isNotBlank(password)) {
+            builder.put("xpack.security.user", user + ":" + password);
+        }
+        TransportClient client = new PreBuiltTransportClient(builder.build());
         String[] split = clusterNodes.split(",");
         for (int i = 0; i < split.length; i++) {
             String[] addrs = split[i].split(":");