消除又臭又長的if...else判斷的技巧,幫你實現(xiàn)更優(yōu)雅的編程!

作者:蘇三說技術(shù)
鏈接:https://juejin.cn/post/6903430159567847432
最近在做代碼重構(gòu),發(fā)現(xiàn)了很多代碼的爛味道。其他的不多說,今天主要說說那些又臭又長的if...else要如何重構(gòu)。
在介紹更更優(yōu)雅的編程之前,讓我們一起回顧一下,不好的if...else代碼
一、又臭又長的if...else
廢話不多說,先看看下面的代碼。
public interface IPay {void pay();}@Servicepublic class AliaPay implements IPay {@Overridepublic void pay() {System.out.println("===發(fā)起支付寶支付===");}}@Servicepublic class WeixinPay implements IPay {@Overridepublic void pay() {System.out.println("===發(fā)起微信支付===");}}@Servicepublic class JingDongPay implements IPay {@Overridepublic void pay() {System.out.println("===發(fā)起京東支付===");}}
@Servicepublic class PayService {@Autowiredprivate AliaPay aliaPay;@Autowiredprivate WeixinPay weixinPay;@Autowiredprivate JingDongPay jingDongPay;????public?void?toPay(String?code)?{if ("alia".equals(code)) {aliaPay.pay();} else if ("weixin".equals(code)) {weixinPay.pay();} else if ("jingdong".equals(code)) {????????????jingDongPay.pay();} else {System.out.println("找不到支付方式");????????}????}}
PayService類的toPay方法主要是為了發(fā)起支付,根據(jù)不同的code,決定調(diào)用用不同的支付類(比如:aliaPay)的pay方法進行支付。這段代碼有什么問題呢?也許有些人就是這么干的。 試想一下,如果支付方式越來越多,比如:又加了百度支付、美團支付、銀聯(lián)支付等等,就需要改toPay方法的代碼,增加新的else...if判斷,判斷多了就會導(dǎo)致邏輯越來越多?很明顯,這里違法了設(shè)計模式六大原則的:開閉原則 和?單一職責(zé)原則。
?開閉原則:對擴展開放,對修改關(guān)閉。就是說增加新功能要盡量少改動已有代碼。
單一職責(zé)原則:顧名思義,要求邏輯盡量單一,不要太復(fù)雜,便于復(fù)用。
?
那有什么辦法可以解決這個問題呢?

二、使用注解
代碼中之所以要用code判斷使用哪個支付類,是因為code和支付類沒有一個綁定關(guān)系,如果綁定關(guān)系存在了,就可以不用判斷了。
我們先定義一個注解。
(RetentionPolicy.RUNTIME)(ElementType.TYPE)public PayCode {String value();String name();}
然后在所有的支付類上都加上注解
(value = "alia", name = "支付寶支付")public class AliaPay implements IPay {public void pay() {System.out.println("===發(fā)起支付寶支付===");}}(value = "weixin", name = "微信支付")public class WeixinPay implements IPay {public void pay() {System.out.println("===發(fā)起微信支付===");}}(value = "jingdong", name = "京東支付")public class JingDongPay implements IPay {public void pay() {System.out.println("===發(fā)起京東支付===");}}
然后增加最關(guān)鍵的類:
public class PayService2 implements ApplicationListener{ private static Map<String, IPay> payMap = null;public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {ApplicationContext applicationContext = contextRefreshedEvent.getApplicationContext();Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(PayCode.class);if (beansWithAnnotation != null) {payMap = new HashMap<>();beansWithAnnotation.forEach((key, value) -> {String bizType = value.getClass().getAnnotation(PayCode.class).value();payMap.put(bizType, (IPay) value);});}}public void pay(String code) {payMap.get(code).pay();????}}
PayService2類實現(xiàn)了ApplicationListener接口,這樣在onApplicationEvent方法中,就可以拿到ApplicationContext的實例。我們再獲取打了PayCode注解的類,放到一個map中,map中的key就是PayCode注解中定義的value,跟code參數(shù)一致,value是支付類的實例。
這樣,每次就可以每次直接通過code獲取支付類實例,而不用if...else判斷了。如果要加新的支付方法,只需在支付類上面打上PayCode注解定義一個新的code即可。
注意:這種方式的code可以沒有業(yè)務(wù)含義,可以是純數(shù)字,只有不重復(fù)就行。

三、動態(tài)拼接名稱
再看看這種方法,主要針對code是有業(yè)務(wù)含義的場景。
@Servicepublic class PayService3 implements ApplicationContextAware {private ApplicationContext applicationContext;private static final String SUFFIX = "Pay";@Overridepublic void setApplicationContext(ApplicationContext applicationContext) throws BeansException {this.applicationContext = applicationContext;}????public?void?toPay(String?payCode)?{((IPay) applicationContext.getBean(getBeanName(payCode))).pay();}public String getBeanName(String payCode) {return payCode + SUFFIX;}}
我們可以看到,支付類bean的名稱是由code和后綴拼接而成,比如:aliaPay、weixinPay和jingDongPay。這就要求支付類取名的時候要特別注意,前面的一段要和code保持一致。調(diào)用的支付類的實例是直接從ApplicationContext實例中獲取的,默認情況下bean是單例的,放在內(nèi)存的一個map中,所以不會有性能問題。
特別說明一下,這種方法實現(xiàn)了ApplicationContextAware接口跟上面的ApplicationListener接口不一樣,是想告訴大家獲取ApplicationContext實例的方法不只一種。

四、模板方法判斷
當(dāng)然除了上面介紹的兩種方法之外,spring的源碼實現(xiàn)中也告訴我們另外一種思路,解決if...else問題。
我們先一起看看spring AOP的部分源碼,看一下DefaultAdvisorAdapterRegistry的wrap方法
public Advisor wrap(Object adviceObject) throws UnknownAdviceTypeException {if (adviceObject instanceof Advisor) {return (Advisor) adviceObject;}if (!(adviceObject instanceof Advice)) {throw new UnknownAdviceTypeException(adviceObject);}Advice advice = (Advice) adviceObject;if (advice instanceof MethodInterceptor) {// So well-known it doesn't even need an adapter.return new DefaultPointcutAdvisor(advice);}for (AdvisorAdapter adapter : this.adapters) {// Check that it is supported.if (adapter.supportsAdvice(advice)) {return new DefaultPointcutAdvisor(advice);}}throw new UnknownAdviceTypeException(advice);}

重點看看supportAdvice方法,有三個類實現(xiàn)了這個方法。我們隨便抽一個類看看
class AfterReturningAdviceAdapter implements AdvisorAdapter, Serializable {public boolean supportsAdvice(Advice advice) {return (advice instanceof AfterReturningAdvice);}public MethodInterceptor getInterceptor(Advisor advisor) {AfterReturningAdvice advice = (AfterReturningAdvice) advisor.getAdvice();return new AfterReturningAdviceInterceptor(advice);}}
該類的supportsAdvice方法非常簡單,只是判斷了一下advice的類型是不是AfterReturningAdvice。
我們看到這里應(yīng)該有所啟發(fā)。

其實,我們可以這樣做,定義一個接口或者抽象類,里面有個support方法判斷參數(shù)傳的code是否自己可以處理,如果可以處理則走支付邏輯。
public interface IPay {boolean support(String code);void pay();}public class AliaPay implements IPay {public boolean support(String code) {return "alia".equals(code);}public void pay() {System.out.println("===發(fā)起支付寶支付===");}}public class WeixinPay implements IPay {public boolean support(String code) {return "weixin".equals(code);}public void pay() {System.out.println("===發(fā)起微信支付===");}}public class JingDongPay implements IPay {public boolean support(String code) {return "jingdong".equals(code);}public void pay() {System.out.println("===發(fā)起京東支付===");}}
每個支付類都有一個support方法,判斷傳過來的code是否和自己定義的相等。
public class PayService4 implements ApplicationContextAware, InitializingBean {private ApplicationContext applicationContext;private ListpayList = null; public void afterPropertiesSet() throws Exception {if (payList == null) {payList = new ArrayList<>();MapbeansOfType = applicationContext.getBeansOfType(IPay.class); beansOfType.forEach((key, value) -> payList.add(value));}}public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {this.applicationContext = applicationContext;}????public?void?toPay(String?code)?{for (IPay iPay : payList) {if (iPay.support(code)) {iPay.pay();}}}}
這段代碼中先把實現(xiàn)了IPay接口的支付類實例初始化到一個list集合中,返回在調(diào)用支付接口時循環(huán)遍歷這個list集合,如果code跟自己定義的一樣,則調(diào)用當(dāng)前的支付類實例的pay方法。

五、其他的消除if...else的方法
當(dāng)然實際項目開發(fā)中使用if...else判斷的場景非常多,上面只是其中幾種場景。下面再列舉一下,其他常見的場景。
1.根據(jù)不同的數(shù)字返回不同的字符串
public String getMessage(int code) {if (code == 1) {return "成功";} else if (code == -1) {return "失敗";} else if (code == -2) {return "網(wǎng)絡(luò)超時";} else if (code == -3) {return "參數(shù)錯誤";}throw new RuntimeException("code錯誤");}
其實,這種判斷沒有必要,用一個枚舉就可以搞定。
public enum MessageEnum {SUCCESS(1, "成功"),FAIL(-1, "失敗"),TIME_OUT(-2, "網(wǎng)絡(luò)超時"),PARAM_ERROR(-3, "參數(shù)錯誤");private int code;private String message;MessageEnum(int code, String message) {this.code = code;this.message = message;}public int getCode() {return this.code;}public String getMessage() {return this.message;}public static MessageEnum getMessageEnum(int code) {return Arrays.stream(MessageEnum.values()).filter(x -> x.code == code).findFirst().orElse(null);}}
再把調(diào)用方法稍微調(diào)整一下
public String getMessage(int code) {MessageEnum messageEnum = MessageEnum.getMessageEnum(code);return messageEnum.getMessage();}
完美。
2.集合中的判斷
上面的枚舉MessageEnum中的getMessageEnum方法,如果不用java8的語法的話,可能要這樣寫
public static MessageEnum getMessageEnum(int code) {for (MessageEnum messageEnum : MessageEnum.values()) {if (code == messageEnum.code) {return messageEnum;}}return null;}
對于集合中過濾數(shù)據(jù),或者查找方法,java8有更簡單的方法消除if...else判斷。
public static MessageEnum getMessageEnum(int code) {return Arrays.stream(MessageEnum.values()).filter(x -> x.code == code).findFirst().orElse(null);}
3.簡單的判斷
其實有些簡單的if...else完全沒有必要寫,可以用三目運算符代替,比如這種情況:
public String getMessage2(int code) {if(code == 1) {return "成功";}return "失敗";}
改成三目運算符:
public String getMessage2(int code) {return code == 1 ? "成功" : "失敗";}
修改之后代碼更簡潔一些。
4.判斷是否為null
java中自從有了null之后,很多地方都要判斷實例是否為null,不然可能會出現(xiàn)NPE的異常。
public String getMessage2(int code) {return code == 1 ? "成功" : "失敗";}public String getMessage3(int code) {Test test = null;return test.getMessage2(1);}
這里如果不判斷異常的話,就會出現(xiàn)NPE異常。我們只能老老實實加上判斷。
public String getMessage3(int code) {Test test = null;if (test != null) {return test.getMessage2(1);}return null;}
有沒有其他更優(yōu)雅的處理方式呢?
public String getMessage3(int code) {Test test = null;OptionaltestOptional = Optional.of(test); return testOptional.isPresent() ? testOptional.get().getMessage2(1) : null;}
答案是使用Optional
當(dāng)然,還有很多其他的場景可以優(yōu)化if...else,我再這里就不一一介紹了,感興趣的朋友可以給我留言,一起探討和研究一下。
好文章,我在看


