java基礎(1)-幾種獲取類的擴展方式

摘要

在日常開發過程中經常需要獲取類的擴展集。即獲取類的子類集（抽象類），或者接口實現類。比如說狀態模式中，狀態構建類，策略模式中的，策略構造方式。本文介紹幾種獲取方式。

實現

以策略模式爲例
定義了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

關注公衆號【方丈的寺院】，第一時間收到文章的更新，與方丈一起開始技術修行之路