io.polyglotted.pgmodel.search.index.FieldMapping 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.index;

import com.google.common.collect.ImmutableMap;
import lombok.*;
import lombok.experimental.Accessors;

import java.util.Map;
import java.util.TreeMap;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.collect.ImmutableMap.copyOf;
import static com.google.common.collect.Iterables.transform;
import static com.google.common.collect.Maps.uniqueIndex;
import static java.util.Collections.singleton;

@RequiredArgsConstructor
@Accessors(fluent = true)
@ToString(includeFieldNames = false, doNotUseGetters = true)
public final class FieldMapping implements Comparable {
    @Getter
    public final String field;
    public final FieldType type;
    public final Indexed indexed;
    public final String analyzer;
    public final String copyTo;
    public final Boolean docValues;
    public final Boolean includeInAll;
    public final Boolean hasFields;
    public final Boolean stored;
    public final ImmutableMap argsMap;
    public final ImmutableMap properties;

    @Override
    public boolean equals(Object o) {
        return this == o || !(o == null || getClass() != o.getClass()) && field.equals(((FieldMapping) o).field);
    }

    @Override
    public int hashCode() {
        return field.hashCode();
    }

    @Override
    public int compareTo(FieldMapping other) {
        return (other == null) ? -1 : field.compareTo(other.field);
    }

    public boolean hasProperties() {
        return (type == FieldType.NESTED || type == FieldType.OBJECT) && properties.size() > 0;
    }

    public boolean hasFields() {
        return Boolean.TRUE.equals(hasFields) && properties.size() > 0;
    }

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

    public static FieldMapping.Builder objectField(String field) {
        return simpleField(field, FieldType.OBJECT);
    }

    public static FieldMapping.Builder nestedField(String field) {
        return simpleField(field, FieldType.NESTED);
    }

    public static FieldMapping.Builder notAnalyzedStringField(String field) {
        return notAnalyzedField(field, FieldType.STRING).docValues(true);
    }

    public static FieldMapping.Builder notAnalyzedField(String field, FieldType fieldType) {
        return simpleField(field, fieldType).indexed(Indexed.NOT_ANALYZED);
    }

    public static FieldMapping.Builder simpleField(String field, FieldType fieldType) {
        return fieldBuilder().field(field).type(fieldType);
    }

    @Setter
    @Accessors(fluent = true, chain = true)
    @NoArgsConstructor(access = AccessLevel.PRIVATE)
    public static class Builder {
        private String field;
        private FieldType type;
        private Indexed indexed = null;
        private String analyzer = null;
        private String copyTo = null;
        private Boolean docValues = null;
        private Boolean includeInAll = null;
        private Boolean hasFields = null;
        private Boolean stored = null;
        private final Map args = new TreeMap<>();
        private final Map properties = new TreeMap<>();

        public Builder property(Iterable properties) {
            this.properties.putAll(uniqueIndex(transform(properties, Builder::build), FieldMapping::field));
            return this;
        }

        public Builder isAPath() {
            hasFields(true);
            return property(singleton(simpleField("tree", FieldType.STRING).analyzer("path_analyzer")));
        }

        public Builder addRawFields() {
            hasFields(true);
            return property(singleton(notAnalyzedStringField("raw")));
        }

        public Builder extra(String name, Object value) {
            args.put(name, value);
            return this;
        }

        public FieldMapping build() {
            return new FieldMapping(checkNotNull(field, "field is required"), checkNotNull(type, "type is required"),
               indexed, analyzer, copyTo, docValues, includeInAll, hasFields, stored, copyOf(args), copyOf(properties));
        }
    }
}