com.speedment.runtime.core.internal.util.DefaultClassMapper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of tool-deploy Show documentation

A Speedment bundle that shades all dependencies into one jar. This is useful when deploying an application on a server.

The newest version!

/*
 *
 * Copyright (c) 2006-2019, Speedment, Inc. All Rights Reserved.
 *
 * 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 com.speedment.runtime.core.internal.util;

import com.speedment.runtime.core.util.ClassMapper;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.stream.Stream;

import static java.util.Objects.requireNonNull;

/**
 *
 * @author pemi
 * @param  the base type
 */
public abstract class DefaultClassMapper implements ClassMapper {

    protected final Consumer NOTHING = v -> {};

    private final Map, V> map;

    public DefaultClassMapper() {
        this.map = new ConcurrentHashMap<>();
    }

    protected  T put(T newItem, Function> keyMapper) {
        requireNonNull(newItem);
        requireNonNull(keyMapper);
        return put(newItem, NOTHING, NOTHING, keyMapper);
    }

    protected  T put(T newItem, Consumer added, Consumer removed, Function> keyMapper) {
        requireNonNull(newItem);
        requireNonNull(added);
        requireNonNull(removed);
        requireNonNull(keyMapper);
        added.accept(newItem);

        @SuppressWarnings("unchecked") // Must be same type!
        final T oldItem = (T) map.put(keyMapper.apply(newItem), newItem);
        if (oldItem != null) {
            removed.accept(oldItem);
        }

        return oldItem;
    }

    @SuppressWarnings("unchecked") // Must be same type!
    @Override
    public  R getOrThrow(Class clazz) {
        requireNonNull(clazz);
        return (R) map.get(clazz);
    }

    @Override
    public Stream, V>> stream() {
        return map.entrySet().stream();
    }

}