nl.vpro.elasticsearch.highlevel.HighLevelClientFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vpro-shared-elasticsearch-high-level-client Show documentation

There is a newer version: 5.3.2

Show newest version

package nl.vpro.elasticsearch.highlevel;

import lombok.Getter;
import lombok.extern.slf4j.Slf4j;

import java.util.function.Consumer;

import jakarta.inject.Inject;

import org.apache.http.HttpHost;
import org.elasticsearch.client.*;

import nl.vpro.elasticsearchclient.ClientElasticSearchFactory;
import nl.vpro.elasticsearchclient.ESClientFactory;

/**
 */
@Slf4j
public class HighLevelClientFactory  implements ESClientFactory  {

    @Getter
    private final ClientElasticSearchFactory lowLevelFactory;

    private RestHighLevelClient client = null;

    @Inject
    public HighLevelClientFactory(ClientElasticSearchFactory lowLevelFactory) {
        this.lowLevelFactory = lowLevelFactory;
    }

    @Override
    public void setHosts(HttpHost... hosts) {
        this.lowLevelFactory.setHosts(hosts);
        invalidate();
    }

    @Override
    public RestClient client(String logName, Consumer callback) {
        return highLevelClient(logName).getLowLevelClient();
    }

    public RestHighLevelClient highLevelClient() {
        return highLevelClient(null);
    }

    public synchronized RestHighLevelClient highLevelClient(String logName){
        if (client == null) {
            client = constructClient(logName);
        }
        return client;
    }

    private RestHighLevelClient constructClient(String logName) {
        RestClientBuilder builder = lowLevelFactory.getClientBuilder();
        if (builder == null) {
            throw new IllegalStateException();
        }
        return new RestHighLevelClient(builder);
    }

    @Override
    public String toString() {
        return "highlevel["+ lowLevelFactory + "]";
    }

}