做一个通用的spi机制的工厂类

    1. import lombok.experimental.UtilityClass;
    3. import java.util.*;
    5. /**
    6.  * @author shizi
    7.  * @since 2020/2/25 上午11:28
    8.  */
    9. @UtilityClass
    10. public class ServiceLoaderFactory {
    12.     private static final Map<Class, Collection<Object>> SERVICE_MAP = new HashMap<>();
    14.     /**
    15.      * 初始化某些类
    16.      *
    17.      * @param interfaceClasses 多个待扩展的spi类型接口
    18.      */
    19.     public void init(Class<?>... interfaceClasses) {
    20.         Arrays.stream(interfaceClasses).forEach(interfaceClass -> {
    21.             for (Object instance : ServiceLoader.load(interfaceClass)) {
    22.                 register(interfaceClass, instance);
    23.             }
    24.         });
    25.     }
    27.     /**
    28.      * 获取接口类对应的所有spi对象集合
    29.      *
    30.      * @param interfaceClass 目标接口类
    31.      * @param <T>            类型
    32.      * @return spi对象集合
    33.      */
    34.     @SuppressWarnings("unchecked")
    35.     public <T> Collection<T> getChildObject(Class<T> interfaceClass) {
    36.         return (Collection<T>) SERVICE_MAP.get(interfaceClass);
    37.     }
    39.     /**
    40.      * 获取指定类的对象
    41.      *
    42.      * @param tClass 目标类
    43.      * @param <T>    类型
    44.      * @return 目标类的实例
    45.      */
    46.     @SuppressWarnings("unchecked")
    47.     public <T> T getTarget(Class<T> tClass) {
    48.         return (T) SERVICE_MAP.entrySet().stream().flatMap(e -> {
    49.             if (e.getKey().isAssignableFrom(tClass)) {
    50.                 return e.getValue().stream();
    51.             }
    52.             return null;
    53.         }).filter(Objects::nonNull).filter(o -> o.getClass().equals(tClass)).findFirst().orElse(null);
    54.     }
    56.     /**
    57.      * 注册目标类以及对应的子类
    58.      *
    59.      * @param interfaceClass spi对外接口
    60.      * @param childInstance  接口的子类
    61.      */
    62.     private void register(Class interfaceClass, Object childInstance) {
    63.         SERVICE_MAP.compute(interfaceClass, (k, v) -> {
    64.             if (null == v) {
    65.                 Collection<Object> valueCollection = new LinkedList<>();
    66.                 valueCollection.add(childInstance);
    67.                 return valueCollection;
    68.             } else {
    69.                 v.add(childInstance);
    70.                 return v;
    71.             }
    72.         });
    73.     }
    74. }

    使用方：

    1. /**
    2.  * @author shizi
    3.  * @since 2020/2/25 下午2:19
    4.  */
    5. public class TestDemo implements LoadServiceDemo {
    7.     static {
    8.         System.out.println("static fun");
    9.         ServiceLoaderFactory.init(LoadServiceDemo.class);
    10.     }
    12.     @Override
    13.     public void testnihao() {
    14.         System.out.println("kkk");
    15.     }
    17.     @Test
    18.     public void test1(){
    19.         ServiceLoaderFactory.getChildObject(LoadServiceDemo.class).forEach(e->{
    20.             e.testnihao();
    21.         });
    22.     }
    24.     @Test
    25.     public void test2(){
    26.         ServiceLoaderFactory.getTarget(LoadServiceDemoImpl1.class).testnihao();
    27.     }
    28. }