io.polyglotted.eswrapper.indexing.Alias Maven / Gradle / Ivy

Go to download
package io.polyglotted.eswrapper.indexing;

import com.google.common.collect.ImmutableList;
import io.polyglotted.eswrapper.query.request.Expression;
import lombok.AccessLevel;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import lombok.ToString;
import lombok.experimental.Accessors;
import org.elasticsearch.action.admin.indices.alias.IndicesAliasesRequest;
import org.elasticsearch.cluster.metadata.AliasAction;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.collect.Iterables.toArray;
import static io.polyglotted.eswrapper.query.ExpressionType.buildFilter;
import static java.util.Arrays.asList;

@RequiredArgsConstructor
@ToString(includeFieldNames = false, doNotUseGetters = true)
public final class Alias {
    public final String alias;
    public final ImmutableList indices;
    public final Expression filter;
    public final boolean remove;

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

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Alias that = (Alias) o;
        return alias.equals(that.alias) && indices.equals(that.indices) && remove == that.remove &&
           (filter == null ? that.filter==null : filter.equals(that.filter));
    }

    @Override
    public int hashCode() {
        return Objects.hash(alias, indices, filter, remove);
    }

    public static Builder aliasBuilder() {
        return new Builder();
    }

    @Setter
    @Accessors(fluent = true, chain = true)
    @RequiredArgsConstructor(access = AccessLevel.PRIVATE)
    public static class Builder {
        private String alias;
        private final List indices = new ArrayList<>();
        private Expression filter;
        private boolean remove = false;

        public Builder index(String... indices) {
            this.indices.addAll(asList(indices));
            return this;
        }

        public Builder remove() {
            this.remove(true);
            return this;
        }

        public Alias build() {
            checkArgument(!indices.isEmpty(), "atleast one index must be added");
            return new Alias(checkNotNull(alias, "alias required"), ImmutableList.copyOf(indices), filter, remove);
        }
    }
}