Spring如何消除代码中的if-else/switch-case

2019-11-26 09:06:37

字体：大中小

来源：转载

供稿：网友

前言

在很多时候，我们代码中会有很多分支，而且分支下面的代码又有一些复杂的逻辑，相信很多人都喜欢用 if-else/switch-case 去实现。做的不好的会直接把实现的代码放在 if-else/switch-case 的分支之下：

switch ( type ) { case case1:  ...  ...  break; case case2:  ...  ...  break; case case3:  ...  ...  break default:  return null;}

这样的代码不仅冗长，读起来也非常困难。做的好一点的会把这些逻辑封装成函数然后在分支中调用：

switch ( type ) { case case1:  return case1Func(); case case2:  return case2Func(); case case3:  return case3Func(); default:  return null;}

即使这样也是面向过程思维的写法，以前写 C 程序的时候也总喜欢这样写，毫无设计模式可言。不仅违背开闭原则，而且随着 switch-case 分支的增多，该段代码只会越来越冗长。其实这种代码已经有成熟的模式去消除诸多的 if-else/switch-case 分支。本文就教大家在 Spring 中如何用注解+策略模式+简单工厂的方式消除 if-else/switch-case 。我们就拿 QQ 空间的个人中心举例子，假如 QQ 空间个人中心有四个 tab 分别是列出我的说说、我的日志、我的照片和我的访客。一般的后台代码很有可能如下：

//各个 tab 名称的枚举：public enum UserRelatedType { /**  * 说说  */ SHUOSHUO("说说"), /**  * 日志  */ RIZHI("日志"), /**  * 发布  */ ZHAOPIAN("照片"), /**  * 访客  */ FANGKE(""); private String desc; UserRelatedType(String desc) {  this.desc = desc; } public String getDesc() {  return desc; } public void setDesc(String desc) {  this.desc = desc; }}

列出 QQ 用户个人中心相关 tab 的代码：

public List<UserRelatedVO> listRelated(UserRelatedQuery query){ UserRelatedType relatedType = UserRelatedType.valueOf(StringUtils.upperCase(query.getType()) ); switch ( relatedType ) {  case SHUOSHUO:   return listRelatedShuoshuo( query );  case RIZHI:   return listRelatedRizhi( query );  case ZHAOPIAN:   return listRelatedZhaopian( query );  case FANGKE:   return listRelatedFangke( query );  default:   return null; }}

而采用注解+策略模式+简单工厂，重构后代码如下：

1、定义一个注解，用来完全消除 if-else：

@Target(ElementType.TYPE)@Retention(RetentionPolicy.RUNTIME)public @interface RelatedTypeAnnotation { /**  * 用户相关类型名称  */ UserRelatedType value();}

2、先定义了个接口，所有 tab 都要实现该接口。其中 list 是 tab 数据展示的方法。

public interface UserRelated { /**  * 列出详细信息  *  * @param query  * @return  */ List<UserRelatedVO> list(UserRelatedQuery query);}

3、定义具体的各个 tab 的实现，继承 UserRelated 策略接口

我的说说

@Component("userRelatedShuoshuo")@RelatedTypeAnnotation( value = UserRelatedType.SHUOSHUO )public class UserRelatedShuoshuo implements UserRelated { @Override public List<UserRelatedVO> list(UserRelatedQuery query) {  System.out.println("我的说说！");  return list; }}

我的日志

@Component("userRelatedRizhi")@RelatedTypeAnnotation( value = UserRelatedType.RIZHI )public class UserRelatedRizhi implements UserRelated { @Override public List<UserRelatedVO> list(UserRelatedQuery query) {  System.out.println("我的日志！");  return list; }}

我的照片

@Component("userRelatedZhaopian")@RelatedTypeAnnotation( value = UserRelatedType.ZHAOPIAN )public class UserRelatedZhaopian implements UserRelated { @Override public List<UserRelatedVO> list(UserRelatedQuery query) {  System.out.println("我的照片！");  return list; }}

我的访客

@Component("userRelatedFangke")@RelatedTypeAnnotation( value = UserRelatedType.FANGKE )public class UserRelatedFangke implements UserRelated { @Override public List<UserRelatedVO> list(UserRelatedQuery query) {  System.out.println("我的访客！");  return list; }}

3、定义一个从 Spring context 获取 bean 的工具类

@Componentpublic class SpringContextUtil implements ApplicationContextAware { private ApplicationContext context; public ApplicationContext getContext() {  return context; } @Override public void setApplicationContext(ApplicationContext context)throws BeansException {  this.context = context; }}

4、定义一个简单工厂，用来生产各种 tab 对象。

@Componentpublic class UserRelatedFactory { @Autowired SpringContextUtil springContextUtil;  private static Map<UserRelatedType, UserRelated> userRelatedMap = Maps.newConcurrentMap(); //工厂将 Spring 装配的相关的 Bean 用 Map 保存起来 public UserRelatedFactory(){  Map<String, Object> beanMap = springContextUtil.getContext().getBeansWithAnnotation(RelatedTypeAnnotation.class);  for(Object userRelated : beanMap.values()) {   RelatedTypeAnnotation annotation = userRelated.getClass().getAnnotation(RelatedTypeAnnotation.class);   userRelatedMap.put(annotation.value(), (UserRelated)userRelated);  } } public static UserRelated createRelated(UserRelatedType relatedType) {  return userRelatedMap.get( relatedType ); }}

5、调用的代码（listRelated 会在 controller 中被调用）。

public List<UserRelatedVO> listRelated(UserRelatedQuery query){ UserRelatedType relatedType = UserRelatedType.valueOf(StringUtils.upperCase(query.getType()) ); UserRelated related = UserRelatedFactory.createRelated( relatedType ); if( related != null ) {  return related.list( query ); } else {  return null; }}

重构后的代码如果需要再新增一种 tab，比如我的好友，只需要新增一种类型继承 UserRelated 实现其中的 list，并加上相应的注解即可。

其实这是一种通用的解决方案，当你 if-else/switch-case 的分支超过 3 个、且分支代码相似且冗长的情况下就应该考虑这种模式。这种模式写出的代码面向对象、清晰、易扩展还高大上，何乐而不为呀，赶紧试试吧！

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对武林网的支持。

上一篇：详解java_ 集合综合案例：斗地主

下一篇：单例模式的反射漏洞和反序列化漏洞代码实例