優(yōu)化if-else代碼的八種方案
前言
代碼中如果if-else比較多,閱讀起來(lái)比較困難,維護(hù)起來(lái)也比較困難,很容易出bug,接下來(lái),本文將介紹優(yōu)化if-else代碼的八種方案。
優(yōu)化方案一:提前return,去除不必要的else
如果if-else代碼塊包含return語(yǔ)句,可以考慮通過(guò)提前return,把多余else干掉,使代碼更加優(yōu)雅。
優(yōu)化前:
- if (condition) {
- //doSomething
- } else {
- return;
- }
優(yōu)化后:
- if(!condition){
- return;
- }
- //doSomething
優(yōu)化方案二:使用條件三目運(yùn)算符
使用條件三目運(yùn)算符可以簡(jiǎn)化某些if-else,使代碼更加簡(jiǎn)潔,更具有可讀性。
優(yōu)化前:
- int price;
- if (condition) {
- price = 80;
- } else {
- price = 100;
- }
優(yōu)化后:
- int price = condition ? 80 : 100;
優(yōu)化方案三:使用枚舉
在某些時(shí)候,使用枚舉也可以?xún)?yōu)化if-else邏輯分支,按個(gè)人理解,它也可以看作一種表驅(qū)動(dòng)方法。
優(yōu)化前:
- String OrderStatusDes;
- if (orderStatus == 0) {
- OrderStatusDes = "訂單未支付";
- } elseif ( OrderStatus == 1 ) {
- OrderStatusDes = "訂單已支付";
- } elseif ( OrderStatus == 2 ) {
- OrderStatusDes = "已發(fā)貨";
- }
- ...
優(yōu)化后:
先定義一個(gè)枚舉
- public enum OrderStatusEnum {
- UN_PAID(0, "訂單未支付"),
- PAIDED(1, "訂單已支付"),
- SENDED(2, "已發(fā)貨"),
- ;
- private int index;
- private String desc;
- public int getIndex() {
- return index;
- }
- public String getDesc() {
- return desc;
- }
- OrderStatusEnum(int index, String desc) {
- this.index = index;
- this.desc = desc;
- }
- OrderStatusEnum of(int orderStatus) {
- for (OrderStatusEnum temp : OrderStatusEnum.values()) {
- if (temp.getIndex() == orderStatus) {
- return temp;
- }
- }
- return null;
- }
- }
有了枚舉之后,以上if-else邏輯分支,可以?xún)?yōu)化為一行代碼
- String OrderStatusDes = OrderStatusEnum.0f(orderStatus).getDesc();
優(yōu)化方案四:合并條件表達(dá)式
如果有一系列條件返回一樣的結(jié)果,可以將它們合并為一個(gè)條件表達(dá)式,讓邏輯更加清晰。
優(yōu)化前
- double getVipDiscount () {
- if (age < 18) {
- return 0.8;
- }
- if ("深圳".equals(city)) {
- return 0.8;
- }
- if (isStudent) {
- return 0.8;
- }
- //do somethig
- }
優(yōu)化后
- double getVipDiscount(){
- if (age< 18 || "深圳".equals(city)||isStudent){
- return 0.8;
- }
- //doSomthing
- }
優(yōu)化方案五:使用 Optional
有時(shí)候if-else比較多,是因?yàn)榉强张袛鄬?dǎo)致的,這時(shí)候你可以使用java8的Optional進(jìn)行優(yōu)化。
優(yōu)化前:
- String str = "jay@huaxiao";
- if (str != null) {
- System.out.println(str);
- } else {
- System.out.println("Null");
- }
優(yōu)化后:
- Optional<String> strOptional = Optional.of("jay@huaxiao");
- strOptional.ifPresentOrElse(System.out::println, () -> System.out.println("Null"));
優(yōu)化方案六:表驅(qū)動(dòng)法
表驅(qū)動(dòng)法,又稱(chēng)之為表驅(qū)動(dòng)、表驅(qū)動(dòng)方法。表驅(qū)動(dòng)方法是一種使你可以在表中查找信息,而不必用很多的邏輯語(yǔ)句(if或case)來(lái)把它們找出來(lái)的方法。以下的demo,把map抽象成表,在map中查找信息,而省去不必要的邏輯語(yǔ)句。
優(yōu)化前:
- if (param.equals(value1)) {
- doAction1(someParams);
- } else if (param.equals(value2)) {
- doAction2(someParams);
- } else if (param.equals(value3)) {
- doAction3(someParams);
- }
- // ...
優(yōu)化后:
- Map<?, Function<?> action> actionMappings = new HashMap<>();
- // 這里泛型 ? 是為方便演示,實(shí)際可替換為你需要的類(lèi)型
- // 初始化
- actionMappings.put(value1, (someParams) -> {
- doAction1(someParams)
- });
- actionMappings.put(value2, (someParams) -> {
- doAction2(someParams)
- });
- actionMappings.put(value3, (someParams) -> {
- doAction3(someParams)
- });
- // 省略多余邏輯語(yǔ)句
- actionMappings.get(param).apply(someParams);
優(yōu)化方案七:優(yōu)化邏輯結(jié)構(gòu),讓正常流程走主干
優(yōu)化前:
- public double getAdjustedCapital () {
- if (_capital <= 0.0) {
- return 0.0;
- }
- if (_intRate > 0 && _duration > 0) {
- return (_income / _duration) * ADJ_FACTOR;
- }
- return 0.0;
- }
優(yōu)化后:
- public double getAdjustedCapital () {
- if (_capital <= 0.0) {
- return 0.0;
- }
- if (_intRate <= 0 || _duration <= 0) {
- return 0.0;
- }
- return (_income / _duration) * ADJ_FACTOR;
- }
將條件反轉(zhuǎn)使異常情況先退出,讓正常流程維持在主干流程,可以讓代碼結(jié)構(gòu)更加清晰。
優(yōu)化方案八:策略模式+工廠(chǎng)方法消除if else
假設(shè)需求為,根據(jù)不同勛章類(lèi)型,處理相對(duì)應(yīng)的勛章服務(wù),優(yōu)化前有以下代碼:
- String medalType = "guest";
- if ("guest".equals(medalType)) {
- System.out.println("嘉賓勛章");
- } else if ("vip".equals(medalType)) {
- System.out.println("會(huì)員勛章");
- } else if ("guard".equals(medalType)) {
- System.out.println("展示守護(hù)勛章");
- }
- ...
首先,我們把每個(gè)條件邏輯代碼塊,抽象成一個(gè)公共的接口,可以得到以下代碼:
- //勛章接口
- public interface IMedalService {
- void showMedal ();
- String getMedalType ();
- }
我們根據(jù)每個(gè)邏輯條件,定義相對(duì)應(yīng)的策略實(shí)現(xiàn)類(lèi),可得以下代碼:
- //守護(hù)勛章策略實(shí)現(xiàn)類(lèi)
- public class GuardMedalServiceImpl implements IMedalService {
- @Override
- public void showMedal() {
- System.out.println("展示守護(hù)勛章");
- }
- @Override
- public String getMedalType() {
- return "guard";
- }
- }
- //嘉賓勛章策略實(shí)現(xiàn)類(lèi)
- public class GuestMedalServiceImpl implements IMedalService {
- @Override
- public void showMedal() {
- System.out.println("嘉賓勛章");
- }
- @Override
- public String getMedalType() {
- return "guest";
- }
- }
- //VIP勛章策略實(shí)現(xiàn)類(lèi)
- public class VipMedalServiceImpl implements IMedalService {
- @Override
- public void showMedal() {
- System.out.println("會(huì)員勛章");
- }
- @Override
- public String getMedalType() {
- return "vip";
- }
- }
接下來(lái),我們?cè)俣x策略工廠(chǎng)類(lèi),用來(lái)管理這些勛章實(shí)現(xiàn)策略類(lèi),如下:
- //勛章服務(wù)工產(chǎn)類(lèi)
- public class MedalServicesFactory {
- private static final Map<String, IMedalService> map = new HashMap<>();
- static {
- map.put("guard", new GuardMedalServiceImpl());
- map.put("vip", new VipMedalServiceImpl());
- map.put("guest", new GuestMedalServiceImpl());
- }
- public static IMedalService getMedalService(String medalType) {
- return map.get(medalType);
- }
- }
- ...
使用了策略+工廠(chǎng)模式之后,代碼變得簡(jiǎn)潔多了,如下:
- public class Test {
- public static void main(String[] args) {
- String medalType = "guest";
- IMedalService medalService = MedalServicesFactory.getMedalService(medalType);
- medalService.showMedal();
- }
- }