JestClientWrapper.java

package com.codetroopers.play.elasticsearch.jest;


import com.codetroopers.play.elasticsearch.AsyncUtils;
import com.codetroopers.play.elasticsearch.IndexClient;
import com.codetroopers.play.elasticsearch.IndexConfig;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.Lists;
import io.searchbox.Action;
import io.searchbox.client.JestClient;
import io.searchbox.client.JestClientFactory;
import io.searchbox.client.JestResult;
import io.searchbox.client.config.HttpClientConfig;
import play.Logger;
import play.libs.F;

import javax.annotation.Nullable;
import javax.validation.constraints.NotNull;
import java.util.List;

import static io.searchbox.client.config.HttpClientConfig.Builder;

public class JestClientWrapper {
    public static JestClient jestClient(IndexConfig config) throws Exception {

        final List<String> connectionUrls = Lists.newArrayList();
        if (config.isLocalMode()){
            connectionUrls.add("http://localhost:9200");
        }else{
            final String[] urlChunks = config.client.split(",");
            for (String urlChunk : urlChunks) {
                if (!urlChunk.startsWith("http")){
                    connectionUrls.add("http://" + urlChunk);      
                }else{
                    connectionUrls.add(urlChunk);
                }
            }
        }
        return buildJestClient(connectionUrls);
    }

    @VisibleForTesting
    static JestClient buildJestClient(List<String> connectionUrls) {
        final Builder builder = new Builder(connectionUrls)
                .multiThreaded(true);

        builder.gson(JestRichResult.createGsonWithDateFormat());
        HttpClientConfig clientConfig = builder.build();

        // Construct a new Jest client according to configuration via factory
        JestClientFactory factory = new JestClientFactory();
        factory.setHttpClientConfig(clientConfig);
        return factory.getObject();
    }

    @NotNull
    public static JestRichResult execute(final JestRequest jestRequest){
        return execute(jestRequest.getAction());
    }

    @NotNull
    public static JestRichResult execute(final Action action){
        JestResult jestResult;
        try {
            jestResult = IndexClient.client.execute(action);
        } catch (Exception e) {
            jestResult = null;
            Logger.error("ElasticSearch : Unable to execute request {}", e);
        }
        return new JestRichResult(jestResult);
    }
    
    
    public static F.Promise<JestRichResult> executeAsync(final Action action){
        return F.Promise.wrap(AsyncUtils.executeAsync(IndexClient.client, action));
    }

    public static void log(@Nullable JestResult jestResult, String prefix){
        if (Logger.isDebugEnabled() && jestResult != null) {
            Logger.debug("ElasticSearch : " + prefix + " " + jestResult.getJsonString());
        }
    }
    
    public static void log(@NotNull JestRichResult jestResult, String prefix){
        log(jestResult.getResult(), prefix);
    }
}