获取同一接口多个实现类的bean

最新推荐文章于 2024-07-14 04:00:33 发布

weixin_30505485

最新推荐文章于 2024-07-14 04:00:33 发布

阅读量799

点赞数

文章标签： java 测试

原文链接：http://www.cnblogs.com/assasion/p/9577341.html

版权

@Service("taskExecutorFactory")
public class TaskExecutorFactory implements ApplicationContextAware,InitializingBean {

    private ApplicationContext context;

    private static Map<TaskTypeEnum, TaskExecutor> taskExecutorMap = new HashMap<>();

    public static TaskExecutor getExecutor(TaskTypeEnum taskTypeEnum) {
        if (!taskExecutorMap.containsKey(taskTypeEnum)) {
            throw new RuntimeException("不支持的任务类型:" + taskTypeEnum.getDescription());
        }
        return taskExecutorMap.get(taskTypeEnum);
    }


    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.context = applicationContext;
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        Map<String, TaskExecutor> taskExecutors = context.getBeansOfType(TaskExecutor.class);
        for (TaskExecutor executor : taskExecutors.values()) {
            taskExecutorMap.put(executor.taskType(), executor);
        }

    }
}