io.polyglotted.eswrapper.services.DocFinder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of es-wrapper Show documentation

Standard set of utilities for elastic search abstraction

The newest version!

package io.polyglotted.eswrapper.services;

import com.google.common.collect.ImmutableList;
import io.polyglotted.eswrapper.query.ResultBuilder;
import io.polyglotted.pgmodel.search.IndexKey;
import org.elasticsearch.action.get.*;
import org.elasticsearch.client.Client;

import java.util.List;

import static com.google.common.base.Strings.isNullOrEmpty;
import static io.polyglotted.eswrapper.ElasticConstants.PARENT_META;
import static io.polyglotted.eswrapper.ElasticConstants.SOURCE_META;
import static io.polyglotted.eswrapper.services.ModelIndexUtil.checkMultiGet;

public abstract class DocFinder {

    public static  T findBy(Client client, String index, String type, String id, String
       parent, ResultBuilder builder) {
        GetResponse response = client.get(new GetRequest(index, type, id).parent(parent)
           .fields(SOURCE_META, PARENT_META)).actionGet();
        return response.isExists() ? builder.buildFrom(response) : null;
    }

    public static  List findAllBy(Client client, Iterable indexKeys, ResultBuilder
       resultBuilder, boolean ignoreFailure) {
        ImmutableList.Builder result = ImmutableList.builder();
        MultiGetRequest multiGetRequest = new MultiGetRequest();
        for (IndexKey key : indexKeys) {
            if (isNullOrEmpty(key.id)) continue;
            multiGetRequest.add(new MultiGetRequest.Item(key.index, key.type, key.id).parent(key.parent)
               .fields(SOURCE_META, PARENT_META));
        }
        if (multiGetRequest.getItems().size() == 0) return result.build();

        MultiGetResponse multiGetItemResponses = client.multiGet(multiGetRequest).actionGet();
        for (MultiGetItemResponse item : multiGetItemResponses.getResponses()) {
            GetResponse get = checkMultiGet(item).getResponse();
            if (!get.isExists()) {
                if (ignoreFailure) continue;
                throw new IllegalStateException("get item not exists");
            }
            result.add(resultBuilder.buildFrom(get));
        }
        return result.build();
    }
}