tech.ytsaurus.core.cypress.Exact Maven / Gradle / Ivy

Go to download
package tech.ytsaurus.core.cypress;

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

import javax.annotation.Nullable;

import tech.ytsaurus.ysontree.YTree;
import tech.ytsaurus.ysontree.YTreeBuilder;
import tech.ytsaurus.ysontree.YTreeNode;


/**
 * @author and-hom
 */
public class Exact extends RangeCriteria {
    @SuppressWarnings("VisibilityModifier")
    public final RangeLimit exact;

    public Exact(RangeLimit exact) {
        this.exact = exact;
    }

    public static Exact of(RangeLimit exact) {
        return new Exact(exact);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        Exact exact1 = (Exact) o;
        return Objects.equals(exact, exact1.exact);
    }

    @Override
    public int hashCode() {
        return Objects.hash(exact);
    }

    @Override
    public YTreeBuilder addRangeCriteria(YTreeBuilder builder) {
        addReadLimit(builder, exact, "exact");
        return builder;
    }

    @Override
    @Nullable
    public RangeCriteria forRetry(long nextRowIndex) {
        if (!exact.key.isEmpty() && exact.rowIndex == -1 && exact.offset == -1) {
            YTreeNode lastPart = YTree.builder()
                    .beginAttributes()
                        .key("type")
                        .value("max")
                    .endAttributes()
                    .entity()
                    .build();
            List upperKey = new ArrayList<>(exact.key);
            upperKey.add(lastPart);
            return new Range(
                    RangeLimit.builder().setKey(exact.key).setRowIndex(nextRowIndex).build(),
                    RangeLimit.key(upperKey));
        } else if (exact.rowIndex == nextRowIndex && exact.offset == -1) {
            return this;
        } else {
            // return new Exact(exact.rowIndex(nextRowIndex));
            // see YTADMINREQ-12492
            // throw new IllegalStateException();
            return null;
        }
    }
}