io.polyglotted.pgmodel.search.SimpleDoc Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pg-model Show documentation

Standard set of models for geo coding, access control and elastic search abstraction

The newest version!

package io.polyglotted.pgmodel.search;

import com.google.common.collect.ImmutableMap;
import io.polyglotted.pgmodel.ObjectRef;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.ToString;
import lombok.experimental.Accessors;
import lombok.experimental.Delegate;

import java.util.Map;
import java.util.Objects;

import static com.google.common.base.Optional.fromNullable;
import static com.google.common.collect.Maps.filterKeys;
import static io.polyglotted.pgmodel.search.index.HiddenFields.BASEVERSION_FIELD;
import static io.polyglotted.pgmodel.search.index.HiddenFields.BYTES_FIELD;
import static io.polyglotted.pgmodel.search.index.HiddenFields.STATUS_FIELD;
import static io.polyglotted.pgmodel.util.ModelUtil.jsonEquals;

@Accessors(fluent = true)
@RequiredArgsConstructor
@ToString(includeFieldNames = false, doNotUseGetters = true)
public final class SimpleDoc implements ObjectRef> {
    @Delegate(excludes = KeyExclude.class)
    @Getter public final IndexKey key;
    @Getter public final ImmutableMap source;

    @Override
    public boolean equals(Object o) { return jsonEquals(this, o); }

    @Override
    public int hashCode() { return Objects.hash(key, source); }

    public Map filteredCopy(boolean includeBase) {
        return filterKeys(source, includeBase ? SimpleDoc::baseVersionKey : SimpleDoc::validKey);
    }

    public DocStatus status() { return DocStatus.fromStatus(strVal(STATUS_FIELD)); }

    public Long baseVersion() { return hasItem(BASEVERSION_FIELD) ? longStrVal(BASEVERSION_FIELD) : null; }

    public boolean hasItem(String property) { return source.containsKey(property); }

    public boolean boolVal(String property) { return fromNullable((Boolean) source.get(property)).or(false); }

    public int intVal(String property) { return fromNullable((Integer) source.get(property)).or(Integer.MIN_VALUE); }

    public long longVal(String property) { return fromNullable(longObjVal(property)).or(Long.MIN_VALUE); }

    public long longStrVal(String property) { return Long.parseLong(strVal(property)); }

    public Long longObjVal(String property) { return (Long) source.get(property); }

    public String strVal(String property) { return (String) source.get(property); }

    private static boolean validKey(String key) { return BYTES_FIELD.equals(key) || !key.startsWith("&"); }

    private static boolean baseVersionKey(String key) { return BASEVERSION_FIELD.equals(key) || validKey(key); }
}