教你如何非常帥氣的改寫“if-else”
掃描二維碼
隨時(shí)隨地手機(jī)看文章
if-else,這是個(gè)再正常不過的coding習(xí)慣,當(dāng)我們代碼量小的時(shí)候用來做條件判斷是再簡(jiǎn)單不過的了。但對(duì)于優(yōu)秀程序員來說,這卻不是好代碼。不信你往下看…
1. 衛(wèi)語句提前return
假設(shè)有如下代碼
if (condition) { // do something } else { return xxx; }
通過對(duì)判斷條件取反,代碼在邏輯表達(dá)上會(huì)更加清晰
if (!condition) { return xxx; } // do something
2. 使用Optional簡(jiǎn)化if判空
2.1 簡(jiǎn)化1級(jí)判空
假設(shè)有如下代碼
if (input != null) { // return value 1 } else { // return value 2 }
使用Optional后
return Optional.ofNullable(input).map(value1).orElse(value2);
2.2 簡(jiǎn)化多級(jí)判空
假設(shè)有如下代碼
if (input != null && input.getUser() != null && input.getUser().getName() != null) { // do action 1 } else { // do action 2 }
使用Optional后
Optional.ofNullable(input) .map(Input::getUser) .map(User::getName) .map(action1) .orElse(action2);
對(duì)于沒有else的場(chǎng)景,使用ifPresent即可
if (input != null && input.getUser() != null && input.getUser.getName() != null) { // do action }
Optional.ofNullable(input) .map(Input::getUser) .map(User::getName) .ifPresent(action);
3. 策略模式
假設(shè)有如下代碼:
if ("dog".equals(petType)) { // 處理dog } else if ("cat".equals(petType)) { // 處理cat } else if ("pig".equals(petType)) { // 處理pig } else if ("rabbit".equals(petType)) { // 處理rabbit } else { throw new UnsupportedOperationException();
這就是不要根據(jù)不同的參數(shù)類型走不同的代碼邏輯,這種場(chǎng)景很常見,他還會(huì)以switch-case的方式出現(xiàn):
switch (petType) { case "dog": // 處理dog break; case "cat": // 處理cat break; case "pig": // 處理pig break; case "rabbit": // 處理rabbit break; default: throw new UnsupportedOperationException(); }
不同的代碼邏輯就代表了不同的策略,我們可以通過如下幾個(gè)方式改寫。
3.1 多態(tài)
public interface Strategy { void invoke(); // 處理各個(gè)邏輯 }
public class DogStrategy implements Strategy { public void invoke() { // 處理dog } }
public class CatStrategy implements Strategy { public void invoke() { // 處理cat }
public class PigStrategy implements Strategy { public void invoke() { // 處理pig } }
public class RabbitStrategy implements Strategy { public void invoke() { // 處理rabbit } }
具體的策略對(duì)象可以放在一個(gè)Map中,優(yōu)化后的實(shí)現(xiàn)類似如下
Strategy strategy = map.get(petType); stratefy.invoke();
關(guān)于如何存放到Map中也兩個(gè)可以參考的方式。
3.1.1 靜態(tài)表
Map<String, Strategy> map = ImmutableMap.<String, Strategy>builder() .put("dog", new DogStrategy()) .put("cat", new CatStrategy()) .put("pig", new PigStrategy()) .put("rabbit", new RabbitStrategy()) .build();
3.1.2 Spring托管下的動(dòng)態(tài)注冊(cè)
定義一個(gè)注冊(cè)中心用于接受注冊(cè)信息
public enum StrategyMapping { INSTANCE; private final Map<String, Classextends Strategy>> map = new ConcurrentHashMap<>(); public void register(String type, Classextends Strategy> clazz) { map.put(type, clazz); } public Strategy getStrategy(String type) { Classextends Strategy> clazz = map.get(type); if (clazz == null) { throw new UnregisteredException(); } return SpringContextHolder.getBean(clazz); } }
將每個(gè)Strategy交由Spring管理,并在構(gòu)造后注冊(cè)
public class DogStrategy implements Strategy { public void init() { StrategyMapping.INSTANCE.register("dog", this.getClass()); } public void invoke() { // 處理dog } }
使用方式就變成了
Strategy strategy = StrategyMapping.INSTANCE.getStrategy(petType); stratefy.invoke();
3.2 枚舉
采用多態(tài)會(huì)額外產(chǎn)生很多策略類,如果我們已經(jīng)預(yù)先將petType定義成了枚舉,就會(huì)發(fā)現(xiàn)可以把Strategy中的invoke()方法放到枚舉中,從而完成了一種映射關(guān)系。
public enum PetType { DOG { public void invoke() { // 處理dog } }, CAT { public void invoke() { // 處理cat } }, PIG { public void invoke() { // 處理pig } }, RABBIT { public void invoke() { // 處理rabbit } }; public abstract void invoke(); }
這樣在調(diào)用時(shí)的代碼就類似如下:
PetType petType = PetType.valueOf(type.toUpperCase(Locale.ROOT)); petType.invoke();
3.3 函數(shù)式簡(jiǎn)化策略
同樣面對(duì)多態(tài)會(huì)額外產(chǎn)生很多策略類的問題,除了枚舉我們還可以使用函數(shù)式的方式來改寫,這里有個(gè)前提最好是策略的內(nèi)容不會(huì)過于復(fù)雜,不然在代碼的可讀性上會(huì)比較差。同樣我們會(huì)有一個(gè)map靜態(tài)表,不過map里面存放的是lambda
Map<String, Runnable> map = ImmutableMap.<String, Runnable>builder() .put("dog", () -> { // 處理dog }) .put("cat", () -> { // 處理cat }) .put("pig", () -> { // 處理pig }) .put("rabbit", () -> { // 處理rabbit }) .build();
使用方式則變成了
Runnable task = map.get(petType); task.run();
本文來自華為云社區(qū),作者:技術(shù)火炬手
免責(zé)聲明:本文內(nèi)容由21ic獲得授權(quán)后發(fā)布,版權(quán)歸原作者所有,本平臺(tái)僅提供信息存儲(chǔ)服務(wù)。文章僅代表作者個(gè)人觀點(diǎn),不代表本平臺(tái)立場(chǎng),如有問題,請(qǐng)聯(lián)系我們,謝謝!