Spring如何消除代码中的if-else/switch-case
前言
在很多时候，我们代码中会有很多分支，而且分支下面的代码又有一些复杂的逻辑，相信很多人都喜欢用if-else/switch-case去实现。做的不好的会直接把实现的代码放在if-else/switch-case的分支之下：
switch(type){
casecase1:
...
...
break;
casecase2:
...
...
break;
casecase3:
...
...
break
default:
returnnull;
}
这样的代码不仅冗长，读起来也非常困难。做的好一点的会把这些逻辑封装成函数然后在分支中调用：
switch(type){
casecase1:
returncase1Func();
casecase2:
returncase2Func();
casecase3:
returncase3Func();
default:
returnnull;
}
即使这样也是面向过程思维的写法，以前写C程序的时候也总喜欢这样写，毫无设计模式可言。不仅违背开闭原则，而且随着switch-case分支的增多，该段代码只会越来越冗长。其实这种代码已经有成熟的模式去消除诸多的if-else/switch-case分支。本文就教大家在Spring中如何用注解+策略模式+简单工厂的方式消除if-else/switch-case。我们就拿QQ空间的个人中心举例子，假如QQ空间个人中心有四个tab分别是列出我的说说、我的日志、我的照片和我的访客。一般的后台代码很有可能如下：
//各个tab名称的枚举：
publicenumUserRelatedType{
/**
*说说
*/
SHUOSHUO("说说"),
/**
*日志
*/
RIZHI("日志"),
/**
*发布
*/
ZHAOPIAN("照片"),
/**
*访客
*/
FANGKE("");
privateStringdesc;
UserRelatedType(Stringdesc){
this.desc=desc;
}
publicStringgetDesc(){
returndesc;
}
publicvoidsetDesc(Stringdesc){
this.desc=desc;
}
}
列出QQ用户个人中心相关tab的代码：
publicListlistRelated(UserRelatedQueryquery){ UserRelatedTyperelatedType=UserRelatedType.valueOf(StringUtils.upperCase(query.getType())); switch(relatedType){ caseSHUOSHUO: returnlistRelatedShuoshuo(query); caseRIZHI: returnlistRelatedRizhi(query); caseZHAOPIAN: returnlistRelatedZhaopian(query); caseFANGKE: returnlistRelatedFangke(query); default: returnnull; } }
而采用注解+策略模式+简单工厂，重构后代码如下：
1、定义一个注解，用来完全消除if-else：
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public@interfaceRelatedTypeAnnotation{
/**
*用户相关类型名称
*/
UserRelatedTypevalue();
}
2、先定义了个接口，所有tab都要实现该接口。其中list是tab数据展示的方法。
publicinterfaceUserRelated{
/**
*列出详细信息
*
*@paramquery
*@return
*/
Listlist(UserRelatedQueryquery);
}
3、定义具体的各个tab的实现，继承UserRelated策略接口
我的说说
@Component("userRelatedShuoshuo")
@RelatedTypeAnnotation(value=UserRelatedType.SHUOSHUO)
publicclassUserRelatedShuoshuoimplementsUserRelated{
@Override
publicListlist(UserRelatedQueryquery){
System.out.println("我的说说！");
returnlist;
}
}
我的日志
@Component("userRelatedRizhi")
@RelatedTypeAnnotation(value=UserRelatedType.RIZHI)
publicclassUserRelatedRizhiimplementsUserRelated{
@Override
publicListlist(UserRelatedQueryquery){
System.out.println("我的日志！");
returnlist;
}
}
我的照片
@Component("userRelatedZhaopian")
@RelatedTypeAnnotation(value=UserRelatedType.ZHAOPIAN)
publicclassUserRelatedZhaopianimplementsUserRelated{
@Override
publicListlist(UserRelatedQueryquery){
System.out.println("我的照片！");
returnlist;
}
}
我的访客
@Component("userRelatedFangke")
@RelatedTypeAnnotation(value=UserRelatedType.FANGKE)
publicclassUserRelatedFangkeimplementsUserRelated{
@Override
publicListlist(UserRelatedQueryquery){
System.out.println("我的访客！");
returnlist;
}
}
3、定义一个从Springcontext获取bean的工具类
@Component
publicclassSpringContextUtilimplementsApplicationContextAware{
privateApplicationContextcontext;
publicApplicationContextgetContext(){
returncontext;
}
@Override
publicvoidsetApplicationContext(ApplicationContextcontext)throwsBeansException{
this.context=context;
}
}
4、定义一个简单工厂，用来生产各种tab对象。
@Component
publicclassUserRelatedFactory{
@Autowired
SpringContextUtilspringContextUtil;
privatestaticMapuserRelatedMap=Maps.newConcurrentMap();
//工厂将Spring装配的相关的Bean用Map保存起来
publicUserRelatedFactory(){
MapbeanMap=springContextUtil.getContext().getBeansWithAnnotation(RelatedTypeAnnotation.class);
for(ObjectuserRelated:beanMap.values()){
RelatedTypeAnnotationannotation=userRelated.getClass().getAnnotation(RelatedTypeAnnotation.class);
userRelatedMap.put(annotation.value(),(UserRelated)userRelated);
}
}
publicstaticUserRelatedcreateRelated(UserRelatedTyperelatedType){
returnuserRelatedMap.get(relatedType);
}
}
5、调用的代码（listRelated会在controller中被调用）。
publicListlistRelated(UserRelatedQueryquery){ UserRelatedTyperelatedType=UserRelatedType.valueOf(StringUtils.upperCase(query.getType())); UserRelatedrelated=UserRelatedFactory.createRelated(relatedType); if(related!=null){ returnrelated.list(query); }else{ returnnull; } }
重构后的代码如果需要再新增一种tab，比如我的好友，只需要新增一种类型继承UserRelated实现其中的list，并加上相应的注解即可。
其实这是一种通用的解决方案，当你if-else/switch-case的分支超过3个、且分支代码相似且冗长的情况下就应该考虑这种模式。这种模式写出的代码面向对象、清晰、易扩展还高大上，何乐而不为呀，赶紧试试吧！
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对毛票票的支持。