develop.framework.commons.components.EntityRegistry Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of develop-framework-commons Show documentation

公共工具包模块

The newest version!

package develop.framework.commons.components;

import lombok.NonNull;

import java.io.Serializable;
import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Optional;

/**
 * 实体注册器
 *
 * @author qiushui on 2018-05-29.
 * @since 0.1
 */
public abstract class EntityRegistry, K> implements Serializable {

    private static final long serialVersionUID = 8580818076321536793L;

    protected Map entityMap;

    public EntityRegistry() {
        this.entityMap = new LinkedHashMap<>();
        T[] defaultEntities = defaultEntity();
        if(defaultEntities != null) {
            Arrays.stream(defaultEntities).forEach(entity -> entityMap.put(entity.key(), entity));
        }
    }

    /**
     * 提供默认实体
     * @return 默认实体数组
     */
    protected abstract T[] defaultEntity();

    /**
     * 添加自定义实体
     *
     * @param customEntities 自定义实体
     */
    public void addCustomEntities(@NonNull T[] customEntities) {
        Arrays.stream(customEntities).forEach(entity -> entityMap.put(entity.key(), entity));
    }

    /**
     * 提取
     * @param key 标记
     * @return 实体
     */
    public Optional extract(K key) {
        if (entityMap.containsKey(key)) {
            return Optional.of(entityMap.get(key));
        }
        return Optional.empty();
    }

    /**
     * 强制提取
     * @param key 标记
     * @param customRuntimeException 如果实体不存在于注册器则抛出自定义异常
     * @return 实体
     */
    public T extractRequired(K key, RuntimeException customRuntimeException) {
        if (entityMap.containsKey(key)) {
            return entityMap.get(key);
        }
        throw customRuntimeException;
    }

    /**
     * 提取，失败使用默认值
     * @param key 标记
     * @param defaultValue 默认值
     * @return 实体
     */
    public T extractOrDefault(K key, T defaultValue) {
        if (entityMap.containsKey(key)) {
            return entityMap.get(key);
        }
        return defaultValue;
    }
}