Bläddra i källkod

fix #1556 add createTransportClient

Shinsuke Sugaya 7 år sedan
förälder
incheckning
72b75d809d
1 ändrade filer med 15 tillägg och 11 borttagningar
  1. 15 11
      src/main/java/org/codelibs/fess/es/client/FessEsClient.java

+ 15 - 11
src/main/java/org/codelibs/fess/es/client/FessEsClient.java

@@ -269,17 +269,7 @@ public class FessEsClient implements Client {
             client = runner.client();
             addTransportAddress("localhost", runner.node().settings().getAsInt("transport.tcp.port", 9300));
         } else {
-            final Builder settingsBuilder = Settings.builder();
-            settingsBuilder.put("cluster.name", fessConfig.getElasticsearchClusterName());
-            settingsBuilder.put("client.transport.sniff", fessConfig.isElasticsearchTransportSniff());
-            settingsBuilder.put("client.transport.ping_timeout", fessConfig.getElasticsearchTransportPingTimeout());
-            settingsBuilder.put("client.transport.nodes_sampler_interval", fessConfig.getElasticsearchTransportNodesSamplerInterval());
-            final Settings settings = settingsBuilder.build();
-            final TransportClient transportClient = new PreBuiltTransportClient(settings);
-            for (final TransportAddress address : transportAddressList) {
-                transportClient.addTransportAddress(address);
-            }
-            client = transportClient;
+            client = createTransportClient(fessConfig);
         }
 
         if (StringUtil.isBlank(transportAddressesValue)) {
@@ -353,6 +343,20 @@ public class FessEsClient implements Client {
         });
     }
 
+    protected Client createTransportClient(final FessConfig fessConfig) {
+        final Builder settingsBuilder = Settings.builder();
+        settingsBuilder.put("cluster.name", fessConfig.getElasticsearchClusterName());
+        settingsBuilder.put("client.transport.sniff", fessConfig.isElasticsearchTransportSniff());
+        settingsBuilder.put("client.transport.ping_timeout", fessConfig.getElasticsearchTransportPingTimeout());
+        settingsBuilder.put("client.transport.nodes_sampler_interval", fessConfig.getElasticsearchTransportNodesSamplerInterval());
+        final Settings settings = settingsBuilder.build();
+        final TransportClient transportClient = new PreBuiltTransportClient(settings);
+        for (final TransportAddress address : transportAddressList) {
+            transportClient.addTransportAddress(address);
+        }
+        return transportClient;
+    }
+
     public boolean existsIndex(final String indexName) {
         final FessConfig fessConfig = ComponentUtil.getFessConfig();
         boolean exists = false;