io.polyglotted.eswrapper.services.ModelIndexUtil 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

There is a newer version: 2.4.3

package io.polyglotted.eswrapper.services;

import io.polyglotted.pgmodel.search.IndexKey;
import io.polyglotted.pgmodel.search.SimpleDoc;
import io.polyglotted.pgmodel.search.index.Alias;
import io.polyglotted.pgmodel.search.query.ResponseHeader;
import org.elasticsearch.action.ActionRequest;
import org.elasticsearch.action.admin.indices.alias.IndicesAliasesRequest;
import org.elasticsearch.action.bulk.BulkItemResponse;
import org.elasticsearch.action.get.MultiGetItemResponse;
import org.elasticsearch.action.get.MultiGetResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.cluster.metadata.AliasAction;
import org.elasticsearch.index.VersionType;

import static com.google.common.collect.Iterables.toArray;
import static io.polyglotted.eswrapper.query.ExprConverter.buildFilter;

public abstract class ModelIndexUtil {

    public static IndicesAliasesRequest.AliasActions aliasActions(Alias alias) {
        return new IndicesAliasesRequest.AliasActions(alias.remove ? AliasAction.Type.REMOVE : AliasAction.Type.ADD,
           toArray(alias.indices, String.class), new String[]{alias.alias}).filter(buildFilter(alias.filter));
    }

    public static ActionRequest forcedRequest(SimpleDoc doc) {
        return new IndexRequest(doc.key.index, doc.key.type, doc.key.id).version(doc.key.version)
           .versionType(VersionType.FORCE).source(doc.source);
    }

    public static IndexKey keyFrom(BulkItemResponse response) {
        return IndexKey.keyFrom(response.getIndex(), response.getType(), response.getId(), response.getVersion());
    }

    public static ResponseHeader headerFrom(SearchResponse response) {
        return new ResponseHeader(response.getTookInMillis(), getTotalHits(response),
           getReturnedHits(response), response.getScrollId());
    }

    public static long getTotalHits(SearchResponse response) { return response.getHits().getTotalHits(); }

    public static int getReturnedHits(SearchResponse response) { return response.getHits().hits().length; }

    public static MultiGetItemResponse checkMultiGet(MultiGetItemResponse item) {
        if (item.isFailed())
            throw new IllegalStateException("error multi-get item " + failureMessage(item.getFailure()));
        return item;
    }

    private static String failureMessage(MultiGetResponse.Failure fail) {
        return fail.getIndex() + "/" + fail.getType() + "/" + fail.getId() + ": " + fail.getMessage();
    }
}