接口存在多个实现类时的动态调用

2019-03-06  本文已影响0人  最终幻想_5b57

接口存在多个实现类时的动态调用

https://blog.csdn.net/jtf8525140/article/details/77509153

在之前做过的项目有这块的需求,也参考过网上大神们的设计方式,主要整理出一下两种方式供大家参考

/**
 * 交通方式
 */
public interface TrafficMode {

    /**
     * 查询交通方式编码
     * @return 编码
     */
    TrafficCode getCode();

    /**
     * 查询交通方式的费用,单位:分
     * @return 费用
     */
    Integer getFee();

}
枚举:

/**
 * 交通类型枚举
 */
public enum TrafficCode {

    TRAIN,
    BUS

}
接口有两个实现类:

/**
 * 汽车方式
 */
@Component
public class BusMode implements TrafficMode {

    @Override
    public TrafficCode getCode() {
        return TrafficCode.BUS;
    }

    @Override
    public Integer getFee() {
        return 10000;
    }

}
/**
 * 火车方式
 */
@Component
public class TrainMode implements TrafficMode {

    @Override
    public TrafficCode getCode() {
        return TrafficCode.TRAIN;
    }

    @Override
    public Integer getFee() {
        return 9000;
    }

}
工厂类:

/**
 * 交通方式工厂类
 */
@Component
public class TrafficModeFactory implements ApplicationContextAware {

    private static Map<TrafficCode, TrafficMode> trafficBeanMap;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        Map<String, TrafficMode> map = applicationContext.getBeansOfType(TrafficMode.class);
        trafficBeanMap = new HashMap<>();
        map.forEach((key, value) -> trafficBeanMap.put(value.getCode(), value));
    }

    public static <T extends TrafficMode> T getTrafficMode(TrafficCode code) {
        return (T)trafficBeanMap.get(code);
    }

}
验证
有了上面的代码之后,我们一起通过单元测试来看一下效果,单元测试代码片段如下:

 @Test
    public void testGetTrafficMode() {
        TrafficMode mode = TrafficModeFactory.getTrafficMode(TrafficCode.BUS);
        Assert.assertEquals(mode.getFee().intValue(), 10000);

        mode = TrafficModeFactory.getTrafficMode(TrafficCode.TRAIN);
        Assert.assertEquals(mode.getFee().intValue(), 9000);
    }
运行之后的结果呢?必然是通过。

/**
 * 汽车方式
 */
@Component
public class BusMode implements TrafficMode , InitializingBean{

    @Autowired
    public ITrafficModeFactory iTrafficModeFactory ;

    @Override
    public Integer getFee() {
        return 10000;
    }

     @Override
    public void afterPropertiesSet() {
        iTrafficModeFactory.register(TrafficCode.BUS,this);
    }
}
/**
 * 火车方式
 */
@Component
public class TrainMode implements TrafficMode  , InitializingBean{

    @Autowired
    public ITrafficModeFactory iTrafficModeFactory ;

    @Override
    public Integer getFee() {
        return 9000;
    }

    @Override
    public void afterPropertiesSet() {
        iTrafficModeFactory.register(TrafficCode.TRAIN,this);
    }

}

工厂类:

/**
 * 交通方式工厂类
 */
@Component
public class TrafficModeFactory {

    private static Map<TrafficCode, TrafficMode> trafficBeanMap = new ConcurrentHashMap<>();;

    @Override
    public void register(String code, TrafficMode trafficMode) {
        trafficBeanMap.put(code, trafficMode);
    }

    public static <T extends TrafficMode> T getTrafficMode(TrafficCode code) {
        return (T)trafficBeanMap.get(code);
    }

}
这里没有写工厂类的接口,写的是实现类,大家就自己实现下吧
测试效果和第一种是一样的

当然实现方法还有很多种,不使用spring,使用抽象类也是可以的,只要可以找到对应关系,实现类可以实例化就可以了。

上一篇 下一篇

猜你喜欢

热点阅读