Elasticsearch Java Client接続プール

5039 ワード

Elasticsearch APIによると、Java側で使用するのはESサービスであるためJava Clientを作成する必要があるが、接続ごとにclientをインスタンス化し、システムに対する消費が大きく、使用後にclient closeを落としても、サーバがsocketリソースをタイムリーに回収できないため、極端な場合、サーバが最大接続数に達する.
上記の問題を解決し、clientの利用率を向上させるために、プール化技術を用いてclientを多重化することを参照することができる.
import org.elasticsearch.client.Client;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.settings.ImmutableSettings;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.transport.InetSocketTransportAddress;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/** * Created by tgg on 16-3-17. */
public class ClientHelper {

    private Settings setting;

    private Map<String, Client> clientMap = new ConcurrentHashMap<String, Client>();

    private Map<String, Integer> ips = new HashMap<String, Integer>(); // hostname port

    private String clusterName = "elasticsearch";

    private ClientHelper() {
        init();
        //TO-DO       client helper
    }

    public static final ClientHelper getInstance() {
        return ClientHolder.INSTANCE;
    }

    private static class ClientHolder {
        private static final ClientHelper INSTANCE = new ClientHelper();
    }

    /** *       client */
    public void init() {
        ips.put("127.0.0.1", 9300);
        setting = ImmutableSettings
                .settingsBuilder()
                .put("client.transport.sniff",true)
                .put("cluster.name","elasticsearch").build();
        addClient(setting, getAllAddress(ips));
    }

    /** *           * * @return */
    public List<InetSocketTransportAddress> getAllAddress(Map<String, Integer> ips) {
        List<InetSocketTransportAddress> addressList = new ArrayList<InetSocketTransportAddress>();
        for (String ip : ips.keySet()) {
            addressList.add(new InetSocketTransportAddress(ip, ips.get(ip)));
        }
        return addressList;
    }

    public Client getClient() {
        return getClient(clusterName);
    }

    public Client getClient(String clusterName) {
        return clientMap.get(clusterName);
    }

    public void addClient(Settings setting, List<InetSocketTransportAddress> transportAddress) {
        Client client = new TransportClient(setting)
                .addTransportAddresses(transportAddress
                        .toArray(new InetSocketTransportAddress[transportAddress.size()]));
        clientMap.put(setting.get("cluster.name"), client);
    }
}