摘要

在日常开发过程中经常需要获取类的扩展集。即获取类的子类集（抽象类），或者接口实现类。比如说状态模式中，状态构建类，策略模式中的，策略构造方式。本文介绍几种获取方式。

实现

以策略模式为例
定义了2种策略

@Getter
@AllArgsConstructor
public enum StrategyEnum {
    APPROVE(1),
    REFUSE(2);
    private final int code;
}

interface Handle,实现类ApproveHandle、RefuseHandle

方式1 Spring大法好

Spring为java web开发提供了很多便捷，这么刚需的需求，Spring当然可以搞定
ListableBeanFactory 扩展了FactoryBean,自然能够获取bean的定义，所以可以从ApplicationContext获取类的扩展类

public static <T> Map<String, T> getBeansOfType(final Class<T> type) {
        if (applicationContext != null) {
            final ListableBeanFactory listableBeanFactory = applicationContext;
            return listableBeanFactory.getBeansOfType(type);
        }
        return null;
    }

方式2 还是Spring

Spring提供了很多的扩展，允许我们在初始化bean之前，之后做一些自定义逻辑。当然也可以在Spring装配整个过程完成后做一些注入的逻辑

@Service
public class InstanceBySpring implements InstanceFactory, InitializingBean {
    private static Map<StateEnum, AbstractState> map = new HashMap<>();
    private static Map<StrategyEnum, Handle> mapStrategy = new HashMap<>();
    @Autowired
    private List<AbstractState> states;

    @Autowired
    private List<Handle> handles;
   // 加载完了处理自己的逻辑
    @Override
    public void afterPropertiesSet() throws Exception {
        if (!CollectionUtils.isEmpty(states)) {
            map = states.stream().collect(Collectors.toMap(v -> v
                    .getCode(), Function.identity()));
        }

        if (!CollectionUtils.isEmpty(handles)) {
            mapStrategy = handles.stream().collect(Collectors.toMap(v -> v
                    .getStrateType(), Function.identity()));
        }

    }

    @Override
    public AbstractState getStateInstance(final StateEnum stateEnum) {
        return map.get(stateEnum);
    }

    @Override
    public Handle getStrategyInstance(final StrategyEnum strategyEnum) {
        return mapStrategy.get(strategyEnum);
    }
}

方式3 能不能不用Spring

离了Spring还能不能活，当然可以了。jdk本身提供了扩展支持ServiceLoader类。

1. META-INF/services配置类
2. ServiceLoader.load

public class InstanceByServiceLoader implements InstanceFactory {
    private final ServiceLoader<AbstractState> abstractStateServiceLoader = ServiceLoader.load
            (AbstractState.class);
    private final ServiceLoader<Handle> handleServiceLoader = ServiceLoader.load
            (Handle.class);

    @Override
    public AbstractState getStateInstance(final StateEnum stateEnum) {
        final List<AbstractState> result = new ArrayList<>();
        for (final AbstractState state : abstractStateServiceLoader) {
            result.add(state);
        }
        return result.stream().collect(Collectors.toMap(p -> p.getCode(), Function.identity())).get
                (stateEnum);
    }
}

总结

这边是以接口类来介绍的，对于接口也同样适用。完整代码见github

关注公众号【方丈的寺院】，第一时间收到文章的更新，与方丈一起开始技术修行之路

在这里插入图片描述