deliver/src/main/java/com/nosum/deliver/file/factory/UploadFileFactory.java

package com.nosum.deliver.file.factory;

import com.nosum.common.constant.Constants;
import com.nosum.common.validator.annotion.NotNull;
import com.nosum.deliver.file.service.UploadFileTemplateService;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 文件存储实例工厂
 */
public class UploadFileFactory {

    private static final Map<String, UploadFileTemplateService> uploadFileServiceMap = new ConcurrentHashMap<>();

    /**
     * 获取工厂UploadFileTemplateService
     *
     * @return
     */
    public static UploadFileTemplateService getUploadFileService(String storyType) {
        return uploadFileServiceMap.get(storyType);
    }

    /**
     * 工厂注册
     *
     * @param storyType
     * @param uploadFileTemplateService
     */
    public static void register(@NotNull final String storyType, final UploadFileTemplateService uploadFileTemplateService) {
        uploadFileServiceMap.put(storyType, uploadFileTemplateService);
    }

}