<kbd id="afajh"><form id="afajh"></form></kbd>
<strong id="afajh"><dl id="afajh"></dl></strong>
    <del id="afajh"><form id="afajh"></form></del>
        1. <th id="afajh"><progress id="afajh"></progress></th>
          <b id="afajh"><abbr id="afajh"></abbr></b>
          <th id="afajh"><progress id="afajh"></progress></th>

          消除又臭又長的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實例的方法不只一種。

          四、模板方法判斷

          當(dāng)然除了上面介紹的兩種方法之外,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)用當(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;    Optional testOptional = Optional.of(test);    return testOptional.isPresent() ? testOptional.get().getMessage2(1) : null;}

          答案是使用Optional

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


          好文章,我在看

          瀏覽 27
          點贊
          評論
          收藏
          分享

          手機掃一掃分享

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

          手機掃一掃分享

          分享
          舉報
          <kbd id="afajh"><form id="afajh"></form></kbd>
          <strong id="afajh"><dl id="afajh"></dl></strong>
            <del id="afajh"><form id="afajh"></form></del>
                1. <th id="afajh"><progress id="afajh"></progress></th>
                  <b id="afajh"><abbr id="afajh"></abbr></b>
                  <th id="afajh"><progress id="afajh"></progress></th>
                  一区二区操逼豆花口骚逼 | 肏屄网| 国产精品二区高清在线苍井空 | 国产美女内射视频 | 精品成人一区二区三区 |