Spring如何消除代码中的if-else/switch-case
前言
在很多时候,我们代码中会有很多分支,而且分支下面的代码又有一些复杂的逻辑,相信很多人都喜欢用 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 的工具类
@component public 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 对象。
@component public 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 个、且分支代码相似且冗长的情况下就应该考虑这种模式。这种模式写出的代码面向对象、清晰、易扩展还高大上,何乐而不为呀,赶紧试试吧!
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对的支持。