org.nutz.boot.starter.elasticsearch.ElasticsearchStarter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nutzboot-starter-elasticsearch Show documentation
The newest version!
package org.nutz.boot.starter.elasticsearch;

import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.transport.TransportAddress;
import org.elasticsearch.transport.client.PreBuiltTransportClient;
import org.nutz.boot.annotation.PropDoc;
import org.nutz.boot.starter.ServerFace;
import org.nutz.ioc.Ioc;
import org.nutz.ioc.impl.PropertiesProxy;
import org.nutz.ioc.loader.annotation.Inject;
import org.nutz.ioc.loader.annotation.IocBean;
import org.nutz.lang.Strings;
import org.nutz.log.Log;
import org.nutz.log.Logs;

import java.net.InetAddress;
import java.net.UnknownHostException;

/**
 * Created by wizzer on 2018/6/15.
 * Modify by wendal on 2019/04/15
 */
@IocBean
public class ElasticsearchStarter implements ServerFace {

    private final static Log log = Logs.get();
    @Inject("refer:$ioc")
    protected Ioc ioc;
    @Inject
    protected PropertiesProxy conf;

    protected TransportClient client;

    protected static final String PRE = "elasticsearch.";

    @PropDoc(value = "Elasticsearch的ip地址", defaultValue = "127.0.0.1", need = true)
    public static final String PROP_HOST = PRE + "host";

    @PropDoc(value = "Elasticsearch的端口", defaultValue = "9300", type = "int", need = true)
    public static final String PROP_PORT = PRE + "port";

    @PropDoc(value = "Elasticsearch的集群名称", defaultValue = "elasticsearch", need = true)
    public static final String PROP_CLUSTER_NAME = PRE + "cluster.name";

    @PropDoc(value = "Elasticsearch是否嗅探集群状态", defaultValue = "true", type = "boolean", need = true)
    public static final String PROP_CLIENT_TRANSPORT_SNIFF = PRE + "client.transport.sniff";

    public void start() throws Exception {
        ioc.get(TransportClient.class, "elasticsearchClient");
    }

    @IocBean(name = "elasticsearchClient", depose = "close")
    public TransportClient getElasticsearchClient() throws UnknownHostException {
        log.debug("loading elasticsearchClient...");
        Settings.Builder builder = Settings.builder();

        // 把所有elasticsearch开头的配置都放入Settings里面
        for (String key : conf.keys()) {
            if (key.startsWith(PRE) && !key.equals(PROP_HOST) && !key.equals(PROP_PORT)) {
                builder.put(key.substring(PRE.length()), conf.get(key));
            }
        }

        //设置ES实例的名称
        builder.put("cluster.name", conf.get(PROP_CLUSTER_NAME, conf.get("nutz.application.name", "nutzboot")));
        //自动嗅探整个集群的状态，把集群中其他ES节点的ip添加到本地的客户端列表中
        builder.put("client.transport.sniff", conf.getBoolean(PROP_CLIENT_TRANSPORT_SNIFF, true));

        client = new PreBuiltTransportClient(builder.build());
        String[] hosts = Strings.splitIgnoreBlank(conf.get(PROP_HOST, "127.0.0.1"), ",");
        int port = conf.getInt(PROP_PORT, 9300);
        for (String host : hosts) {
            client.addTransportAddress(new TransportAddress(InetAddress.getByName(host), port));
        }
        return client;
    }
}