徹底消滅if-else,你需要這8種方案!
優(yōu)化方案 1:提前 return,去除不必要的 else
如果 if-else 代碼塊包含 return 語(yǔ)句,可以考慮通過提前 return,把多余 else 干掉,使代碼更加優(yōu)雅。
優(yōu)化前:
- if(condition){
- //doSomething
- }else{
- return ;
- }
優(yōu)化后:
- if(!condition){
- return ;
- }
- //doSomething
優(yōu)化方案 2:使用條件三目運(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)化方案 3:使用枚舉
在某些時(shí)候,使用枚舉也可以優(yōu)化 if-else 邏輯分支,按個(gè)人理解,它也可以看作一種表驅(qū)動(dòng)方法。
優(yōu)化前:
- String OrderStatusDes;
- if(orderStatus==0){
- OrderStatusDes ="訂單未支付";
- }else if(OrderStatus==1){
- OrderStatusDes ="訂單已支付";
- }else if(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 邏輯分支,可以優(yōu)化為一行代碼:
- String OrderStatusDes = OrderStatusEnum.0f(orderStatus).getDesc();
優(yōu)化方案 4:合并條件表達(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)化方案 5:使用 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)化方案 6:表驅(qū)動(dòng)法
表驅(qū)動(dòng)法,又稱之為表驅(qū)動(dòng)、表驅(qū)動(dòng)方法。表驅(qū)動(dòng)方法是一種使你可以在表中查找信息,而不必用很多的邏輯語(yǔ)句(if 或 case)來把它們找出來的方法。
以下的 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í)際可替換為你需要的類型
- // 初始化
- 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)化方案 7:優(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)化方案 8:策略模式+工廠方法消除 if else
假設(shè)需求為,根據(jù)不同勛章類型,處理相對(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();
- }
我們根據(jù)每個(gè)邏輯條件,定義相對(duì)應(yīng)的策略實(shí)現(xiàn)類,可得以下代碼:
- //守護(hù)勛章策略實(shí)現(xiàn)類
- public class GuardMedalServiceImpl implements IMedalService {
- @Override
- public void showMedal() {
- System.out.println("展示守護(hù)勛章");
- }
- }
- //嘉賓勛章策略實(shí)現(xiàn)類
- public class GuestMedalServiceImpl implements IMedalService {
- @Override
- public void showMedal() {
- System.out.println("嘉賓勛章");
- }
- }
- //VIP勛章策略實(shí)現(xiàn)類
- public class VipMedalServiceImpl implements IMedalService {
- @Override
- public void showMedal() {
- System.out.println("會(huì)員勛章");
- }
- }
接下來,我們?cè)俣x策略工廠類,用來管理這些勛章實(shí)現(xiàn)策略類,如下:
- //勛章服務(wù)工產(chǎn)類
- 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);
- }
- }
使用了策略+工廠模式之后,代碼變得簡(jiǎn)潔多了,如下:
- public class Test {
- public static void main(String[] args) {
- String medalType = "guest";
- IMedalService medalService = MedalServicesFactory.getMedalService(medalType);
- medalService.showMedal();
- }
- }