core/store/dist/src/main/java/org/onosproject/store/consistent/impl/UpdateOperation.java

package org.onosproject.store.consistent.impl;

import static com.google.common.base.Preconditions.*;

import com.google.common.base.MoreObjects;

/**
 * Database update operation.
 *
 * @param <K> key type.
 * @param <V> value type.
 */
public class UpdateOperation<K, V> {

    /**
     * Type of database update operation.
     */
    public static enum Type {
        PUT,
        PUT_IF_ABSENT,
        PUT_IF_VERSION_MATCH,
        PUT_IF_VALUE_MATCH,
        REMOVE,
        REMOVE_IF_VERSION_MATCH,
        REMOVE_IF_VALUE_MATCH,
    }

    private Type type;
    private String tableName;
    private K key;
    private V value;
    private V currentValue;
    private long currentVersion = -1;

    /**
     * Returns the type of update operation.
     * @return type of update.
     */
    public Type type() {
        return type;
    }

    /**
     * Returns the tableName being updated.
     * @return table name.
     */
    public String tableName() {
        return tableName;
    }

    /**
     * Returns the item key being updated.
     * @return item key
     */
    public K key() {
        return key;
    }

    /**
     * Returns the new value.
     * @return item's target value.
     */
    public V value() {
        return value;
    }

    /**
     * Returns the expected current value in the database value for the key.
     * @return current value in database.
     */
    public V currentValue() {
        return currentValue;
    }

    /**
     * Returns the expected current version in the database for the key.
     * @return expected version.
     */
    public long currentVersion() {
        return currentVersion;
    }

    @Override
    public String toString() {
        return MoreObjects.toStringHelper(this)
            .add("type", type)
            .add("tableName", tableName)
            .add("key", key)
            .add("value", value)
            .add("currentValue", currentValue)
            .add("currentVersion", currentVersion)
            .toString();
    }

    /**
     * Creates a new builder instance.
     * @param <K> key type.
     * @param <V> value type.
     *
     * @return builder.
     */
    public static <K, V> Builder<K, V> newBuilder() {
        return new Builder<>();
    }

    /**
     * UpdatOperation builder.
     *
     * @param <K> key type.
     * @param <V> value type.
     */
    public static final class Builder<K, V> {

        private UpdateOperation<K, V> operation = new UpdateOperation<>();

        public UpdateOperation<K, V> build() {
            validateInputs();
            return operation;
        }

        public Builder<K, V> withType(Type type) {
            operation.type = checkNotNull(type, "type cannot be null");
            return this;
        }

        public Builder<K, V> withTableName(String tableName) {
            operation.tableName = checkNotNull(tableName, "tableName cannot be null");
            return this;
        }

        public Builder<K, V> withKey(K key) {
            operation.key = checkNotNull(key, "key cannot be null");
            return this;
        }

        public Builder<K, V> withCurrentValue(V value) {
            operation.currentValue = checkNotNull(value, "currentValue cannot be null");
            return this;
        }

        public Builder<K, V> withValue(V value) {
            operation.value = checkNotNull(value, "value cannot be null");
            return this;
        }

        public Builder<K, V> withCurrentVersion(long version) {
            checkArgument(version >= 0, "version cannot be negative");
            operation.currentVersion = version;
            return this;
        }

        private void validateInputs() {
            checkNotNull(operation.type, "type must be specified");
            checkNotNull(operation.tableName, "table name must be specified");
            checkNotNull(operation.key, "key must be specified");
            switch (operation.type) {
            case PUT:
            case PUT_IF_ABSENT:
                checkNotNull(operation.value, "value must be specified.");
                break;
            case PUT_IF_VERSION_MATCH:
                checkNotNull(operation.value, "value must be specified.");
                checkState(operation.currentVersion >= 0, "current version must be specified");
                break;
            case PUT_IF_VALUE_MATCH:
                checkNotNull(operation.value, "value must be specified.");
                checkNotNull(operation.currentValue, "currentValue must be specified.");
                break;
            case REMOVE:
                break;
            case REMOVE_IF_VERSION_MATCH:
                checkState(operation.currentVersion >= 0, "current version must be specified");
                break;
            case REMOVE_IF_VALUE_MATCH:
                checkNotNull(operation.currentValue, "currentValue must be specified.");
                break;
            default:
                throw new IllegalStateException("Unknown operation type");
            }
        }
    }
}