如何优雅地处理switch-case

使用注解和多态。

思路:将自定义注解打在相应的处理器上,在Service类初始化时,自动扫描打了响应注解的处理器类,并保存下来,然后通过具体的传参取出具体的类来处理。

具体看这个例子,现有一批资源Pojo类,资源Pojo主要有2个属性,主键resourceId和资源类型resourceType,现要根据不同的资源类型用resourceId去查不同的接口,返回资源的详细信息,返回的信息都是每个资源都会有的,比如资源名称。

资源Pojo的定义是:

class ResourcePojo{
    private Long resourceId;
    private EnumResource resourceType;
}

再来定义一个查询资源接口:

interface IResourceInfoGetter{
    //根据单个资源查询
    ResourceModel getResourceInfo(ResourcePojo resource);
    //根据多个资源查询
    List<ResourceModel> getResourceInfoList(List<ResourcePojo> resources);
}

class ResourceModel{
    private Long resourceId;
    private Integer resourceType;
    //这里为了简洁就只写个资源名称,但其实可能还有更多的资源信息。
    private String resourceName;
}

/**
* 枚举中有4个值的资源类型,分别表示机票、汽车票、长途汽车票、火车票的资源
*/
enum EnumResource{
    FLIGHT,BUS, LONG_DISTANCE_BUS,TRAIN;
}

现在可以开始写代码了,这里还有要注意的东西就是BUS,LONG_DISTANCE_BUS这两种虽然是不同资源,但是查询的接口是同一个,所以虽然枚举值有4个,但是外部接口其实只有3个。

//新增注解
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE})
@interface ResourceType {
    EnumResource type();
}

//把注解打到对应的类上
@ResourceType(type = EnumResource.FLIGHT)
public class FlightResourceInfoGetter implements IResourceInfoGetter {
       ...
}

//service类
@Service
public class ResourceInfoGetter implements IResourceInfoGetter {
    private final static Map<EnumResource, IResourceInfoGetter> GETTER_STORE = new EnumMap<>(EnumResource.class);

    //有Getter所在包的包名
    private final static String PACKAGE_NAME = "xxx";
    //自动扫描带有自定义注解的类
    @PostConstruct
    private void init() {
        //需要引入第三方jar包 org.reflections
        Reflections reflections = new Reflections(PACKAGE_NAME);
        //获取包下所有IResourceInfoGetter
        Set<Class<? extends IResourceInfoGetter>> classes = reflections.getSubTypesOf(IResourceInfoGetter.class);
        for (Class<? extends IResourceInfoGetter> clazz : classes) {
            //因为对象是被spring管理的,所以需要通过applicationContext拿bean。
            //为了解耦ApplicationContext和该类关系,使用了一个SpringContextHolder类。
            GETTER_STORE.put(clazz.getAnnotation(ResourceType.class).type(), SpringContextHolder.getBean(clazz));
        }
    }
    //根据不同的resources调用不同接口取出信息
    public List<ResourceModel> getResourceInfoList(List<ResourcePojo> resources){
        Map<IResourceInfoGetter, List<ResourcePojo>> getterMap = resources.stream().collect(Collectors.groupingBy(r -> GETTER_STORE.get(r.getResourceType())));
        List<ResourceModel> result = new ArrayList<>(resources.size());
        for (Map.Entry<IResourceInfoGetter, List<ResourcePojo>> entry : getterMap.entrySet()) {
            result.addAll(entry.getKey().getResourceInfoList(entry.getValue()));
        }
        return result;
    }
}


现在再来看如果新增一种resourceType,需要改动哪些地方:

1.Enum类新增一种资源类型。
2.新增一个实现IResourceInfoGetter的类 (类上的注解 ),用来查询第五种资源。

代码比switch-case简洁,扩展性也比switch-case强。

总结:利用多态和注解来替代switch-case代码块。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,406评论 6 503
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,732评论 3 393
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,711评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,380评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,432评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,301评论 1 301
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,145评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,008评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,443评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,649评论 3 334
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,795评论 1 347
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,501评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,119评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,731评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,865评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,899评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,724评论 2 354

推荐阅读更多精彩内容