/*
 * Copyright 2017, TeamDev Ltd. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.spine3.server.entity.storage;

import com.google.common.collect.ImmutableMap;
import org.spine3.server.reflect.Property;

import java.util.HashMap;
import java.util.Map;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;

/**
 * @author Dmytro Dashenkov
 */
public class ColumnTypeRegistry<R, C> {

    private final Map<Class, ColumnType> propertyColumnType;

    private ColumnTypeRegistry(Map<Class, ColumnType> propertyColumnType) {
        this.propertyColumnType = propertyColumnType;
    }

    public ColumnType get(Property<?> field) {
        checkNotNull(field);
        final Class javaType = field.getType();
        final ColumnType type = propertyColumnType.get(javaType);
        checkState(type != null,
                   "Could not find storage type for %s.",
                   javaType.getName());
        return type;
    }

    public static <R, C> Builder<R, C> newBuilder() {
        return new Builder<>();
    }

    public static class Builder<R, C> {

        private final Map<Class, ColumnType> propertyColumnType = new HashMap<>();

        private Builder() {
        }

        public <J, S> Builder put(Class<J> javaType, ColumnType<J, S, R, C> columnType) {
            propertyColumnType.put(javaType, columnType);
            return this;
        }

        public ColumnTypeRegistry<R, C> build() {
            return new ColumnTypeRegistry<>(ImmutableMap.copyOf(propertyColumnType));
        }
    }
}

/*
 * Copyright 2017, TeamDev Ltd. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.spine3.server.entity.storage;

import com.google.common.collect.ImmutableMap;
import org.spine3.server.reflect.Property;

import java.util.HashMap;
import java.util.Map;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;

/**
 * @author Dmytro Dashenkov
 */
public class ColumnTypeRegistry {

    private final Map<Class, ColumnType> propertyColumnType;

    private ColumnTypeRegistry(Map<Class, ColumnType> propertyColumnType) {
        this.propertyColumnType = propertyColumnType;
    }

    public ColumnType get(Property<?> field) {
        checkNotNull(field);
        final Class javaType = field.getType();
        final ColumnType type = propertyColumnType.get(javaType);
        checkState(type != null,
                   "Could not find storage type for %s.",
                   javaType.getName());
        return type;
    }

    public static <R, C> Builder<R, C> newBuilder() {
        return new Builder<>();
    }

    public static class Builder<R, C> {

        private final Map<Class, ColumnType> propertyColumnType = new HashMap<>();

        private Builder() {
        }

        public <J, S> Builder put(Class<J> javaType, ColumnType<J, S, R, C> columnType) {
            propertyColumnType.put(javaType, columnType);
            return this;
        }

        public ColumnTypeRegistry build() {
            return new ColumnTypeRegistry(ImmutableMap.copyOf(propertyColumnType));
        }
    }
}

Delete redundant generics.