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

        共 9935字,需瀏覽 20分鐘

         ·

        2021-01-05 11:34


        作者蘇三說技術(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 {

        @Override public void pay() { System.out.println("===發(fā)起支付寶支付==="); }}

        @Servicepublic class WeixinPay implements IPay {
        @Override public void pay() { System.out.println("===發(fā)起微信支付==="); }}

        @Servicepublic class JingDongPay implements IPay {
        @Override public void pay() { System.out.println("===發(fā)起京東支付==="); }}


        @Servicepublic class PayService {
        @Autowired private AliaPay aliaPay; @Autowired private WeixinPay weixinPay; @Autowired private 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)系存在了,就可以不用判斷了。

        我們先定義一個注解。

        @Retention(RetentionPolicy.RUNTIME)@Target(ElementType.TYPE)public @interface PayCode {
        String value();
        String name();}

        然后在所有的支付類上都加上注解

        @PayCode(value = "alia", name = "支付寶支付")@Servicepublic class AliaPay implements IPay {

        @Override public void pay() { System.out.println("===發(fā)起支付寶支付==="); }}


        @PayCode(value = "weixin", name = "微信支付")@Servicepublic class WeixinPay implements IPay {
        @Override public void pay() { System.out.println("===發(fā)起微信支付==="); }}

        @PayCode(value = "jingdong", name = "京東支付")@Servicepublic class JingDongPay implements IPay {
        @Override public void pay() { System.out.println("===發(fā)起京東支付==="); }}


        然后增加最關(guān)鍵的類:


        @Servicepublic class PayService2 implements ApplicationListener {
        private static Map<String, IPay> payMap = null;
        @Override 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";
        @Override public 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實例的方法不只一種。

        四、模板方法判斷

        當然除了上面介紹的兩種方法之外,spring的源碼實現(xiàn)中也告訴我們另外一種思路,解決if...else問題。

        我們先一起看看spring AOP的部分源碼,看一下DefaultAdvisorAdapterRegistrywrap方法

          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 {
        @Override public boolean supportsAdvice(Advice advice) { return (advice instanceof AfterReturningAdvice); }
        @Override 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();}

        @Servicepublic class AliaPay implements IPay {
        @Override public boolean support(String code) { return "alia".equals(code); }
        @Override public void pay() { System.out.println("===發(fā)起支付寶支付==="); }}

        @Servicepublic class WeixinPay implements IPay {

        @Override public boolean support(String code) { return "weixin".equals(code); }
        @Override public void pay() { System.out.println("===發(fā)起微信支付==="); }}

        @Servicepublic class JingDongPay implements IPay { @Override public boolean support(String code) { return "jingdong".equals(code); }
        @Override public void pay() { System.out.println("===發(fā)起京東支付==="); }}

        每個支付類都有一個support方法,判斷傳過來的code是否和自己定義的相等。

        @Servicepublic class PayService4 implements ApplicationContextAware, InitializingBean {
        private ApplicationContext applicationContext;
        private List payList = null;
        @Override public void afterPropertiesSet() throws Exception { if (payList == null) { payList = new ArrayList<>(); Map beansOfType = applicationContext.getBeansOfType(IPay.class);
        beansOfType.forEach((key, value) -> payList.add(value)); } }
        @Override 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)用當前的支付類實例的pay方法。

        五、其他的消除if...else的方法

        當然實際項目開發(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;    Optional testOptional = Optional.of(test);    return testOptional.isPresent() ? testOptional.get().getMessage2(1) : null;}

        答案是使用Optional

        當然,還有很多其他的場景可以優(yōu)化if...else,我再這里就不一一介紹了,感興趣的朋友可以給我留言,一起探討和研究一下。


        好文章,我在看

        瀏覽 27
        點贊
        評論
        收藏
        分享

        手機掃一掃分享

        分享
        舉報
        評論
        圖片
        表情
        推薦
        點贊
        評論
        收藏
        分享

        手機掃一掃分享

        分享
        舉報
          
          

            1. 欧美精品有码 | Ts视频在线观看一区二区三区 | 欧美在线视频免费观看 | 女上男下吃奶100动态图 | 变态残暴的av |