io.inversion.Index Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of inversion-core-api Show documentation
Inversion Cloud API Engine
The newest version!
/*
 * Copyright (c) 2015-2019 Rocket Partners, LLC
 * https://github.com/inversion-api
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.inversion;

import io.inversion.query.Projection;
import io.inversion.utils.Utils;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 *
 */
public class Index implements Serializable {

    public static final String TYPE_INDEX        = "INDEX";
    public static final String TYPE_RESOURCE_KEY = "RESOURCE_KEY";
    public static final String TYPE_PRIMARY_KEY  = "PRIMARY_KEY";
    public static final String TYPE_FOREIGN_KEY  = "FOREIGN_KEY";


    protected final List properties = new ArrayList<>();
    protected       Collection     collection = null;
    protected       String         name       = null;
    protected       String         type       = null;           // primary, partition, sort, localsecondary, etc
    protected boolean    unique     = true;
    protected Projection projection = null;

    public Index() {
        super();
    }

    public Index(String name, String type, boolean unique, Property... properties) {
        withName(name);
        withType(type);
        withUnique(unique);
        withProperties(properties);
    }

    public Index(String name, String type, boolean unique, int sequence, Property property) {
        withName(name);
        withType(type);
        withUnique(unique);
        withProperty(sequence, property);
    }

    public boolean equals(Object object) {
        if (object == this)
            return true;

        if (object instanceof Index) {
            Index index = ((Index) object);
            return ((collection == null || collection == index.collection) && Utils.equal(name, index.name));
        }
        return false;
    }

    public String toString() {

        StringBuffer buff = new StringBuffer("[Index: ").append(collection == null ? null : collection.getName()).append(":").append(name).append("(");
        for (int i = 0; i < size(); i++) {
            buff.append(getJsonName(i));
            if (i < size() - 1)
                buff.append(", ");
        }
        buff.append(")]");
        return buff.toString();
    }

    public boolean isExclude() {
        if (collection.isExclude())
            return true;

        for (Property property : properties)
            if (property.isExclude())
                return true;

        return false;
    }

    public Collection getCollection() {
        return collection;
    }

    public Index withCollection(Collection coll) {
        if (this.collection != coll) {
            this.collection = coll;
            coll.withIndexes(this);
        }

        return this;
    }

    public String getName() {
        return name;
    }

    public Index withName(String name) {
        this.name = name;
        return this;
    }

    public String getType() {
        return type;
    }

    public Index withType(String type) {
        this.type = type;
        return this;
    }

    public boolean isType(String... types) {
        for (String type : types) {
            if (type.equalsIgnoreCase(this.type))
                return true;
        }
        return false;
    }

    public boolean isUnique() {
        return unique;
    }

    public Index withUnique(boolean unique) {
        this.unique = unique;
        return this;
    }

    public int size() {
        return properties.size();
    }


    public Index withProperty(int sequence, Property property){
        if(!this.properties.contains(property)){
            if(sequence > 0){
                while(this.properties.size() < sequence){
                    this.properties.add(null);
                }
                this.properties.set(sequence -1, property);
            }
            else{
                this.properties.add(property);
            }
        }
        return this;
    }

    public Index withProperties(Property... properties) {
        for (int i = 0; properties != null && i < properties.length; i++) {
            if (properties[i] != null && !this.properties.contains(properties[i]))
                this.properties.add(properties[i]);
        }
        return this;
    }

    public List getProperties() {
        return new ArrayList(properties);
    }

    public Property getProperty(int idx) {
        return properties.get(idx);
    }

    public String getColumnName(int index) {
        return index < properties.size() ? properties.get(index).getColumnName() : null;
    }

    public List getColumnNames() {
        return properties.stream().map(Property::getColumnName).collect(Collectors.toList());
    }

    public String getJsonName(int index) {
        return index < properties.size() ? properties.get(index).getJsonName() : null;
    }

    public List getJsonNames() {
        return properties.stream().map(Property::getJsonName).collect(Collectors.toList());
    }

    public Projection getProjection() {
        return projection;
    }

    public Index withProjection(Projection projection) {
        this.projection = projection;
        return this;
    }
}