如何优化代码中大量的if/else

訾博ZiBo

发布于 2025-01-06 18:09:43

3000

代码可运行

文章被收录于专栏：全栈开发工程师全栈开发工程师

运行总次数：0

代码可运行

一、如何优化代码中大量的if/else

不是所有的if/else和switch/case都需要优化，当我们发现有“痛点”或者“闻到代码有坏味道”再来优化才是最好的，不然你可能会写了一个从不扩展的可扩展代码，所有的优化都是为了更好的迭代项目，更好的服务于业务，而不是为了优化而优化——深夜里的程序员

1、判断条件取反，提前return

package com.zibo.ifelse;

// 判断条件取反，提前return
public class Method01 {
    public static void main(String[] args) {
        boolean good = true;
        // 原始写法
        System.out.println(doSth(good));
        // 优化写法：判断条件取反，提前return
        System.out.println(doSthNew(good));
    }

    // 原始写法
    private static String doSth(boolean good) {
        if (good) {
            // do something
            return "你是个好人！热烈欢迎！";
        } else {
            return null;
        }
    }

    // 优化写法：判断条件取反，提前return
    private static String doSthNew(boolean good) {
        if(!good){
            return null;
        }
        // do something
        return "你是个好人！热烈欢迎！";
    }
}

2、策略模式

package com.zibo.ifelse;

import java.util.HashMap;
import java.util.Map;

// 策略模式
public class Method02 {
    public static void main(String[] args) {
        String strategy = "大哥";
        // 原始写法
        doSth(strategy);
        // 多态写法
        doSthNew1(strategy);
        // 枚举写法
        doSthNew2(strategy);
    }

    // 原始写法
    private static void doSth(String strategy) {
        if(strategy.equals("大哥")){
            System.out.println("大哥威猛！……");
        }else if(strategy.equals("二哥")){
            System.out.println("二哥豪爽！……");
        }else if(strategy.equals("三哥")){
            System.out.println("二哥帅气！……");
        }
    }

    // 多态写法
    private static void doSthNew1(String strategy){
        Map<String,Strategy> map = new HashMap<>();
        map.put("大哥",new A());
        map.put("二哥",new B());
        map.put("三哥",new C());
        Strategy s = map.get(strategy);
        s.run();
    }

    // 枚举写法
    private static void doSthNew2(String strategy){
        EStrategy.valueOf(strategy).run();
    }
}
// 多态写法
interface Strategy{
    void run();
}

class A implements Strategy{
    @Override
    public void run() {
        System.out.println("大哥威猛！……");
    }
}

class B implements Strategy{
    @Override
    public void run() {
        System.out.println("二哥豪爽！……");
    }
}

class C implements Strategy{
    @Override
    public void run() {
        System.out.println("二哥帅气！……");
    }
}

// 枚举写法
enum EStrategy{
    大哥 {
        @Override
        void run() {
            System.out.println("大哥威猛！……");
        }
    }, 二哥 {
        @Override
        void run() {
            System.out.println("二哥豪爽！……");
        }
    }, 三哥 {
        @Override
        void run() {
            System.out.println("二哥帅气！……");
        }
    };

    abstract void run();
}

3、使用Optional

package com.zibo.ifelse;

import java.util.Optional;
import java.util.function.Function;

// 使用 Optional
public class Method03 {
    public static void main(String[] args) {
        String str = null;
        // 原始写法
        doSth(str);
        // 使用 Optional
        doSthNew(str);
    }

    // 原始写法
    private static void doSth(String str) {
        if(str == null){
            System.out.println("你在耍我！");
        }else {
            System.out.println("好的，大哥！");
        }
    }

    // 使用 Optional
    private static void doSthNew(String str) {
        Optional<String> userOptional = Optional.ofNullable(str);
        Function<String, String> function = string -> "好的，大哥！";
        System.out.println(userOptional.map(function).orElse("你在耍我！"));
    }
}

4、表驱动法

表驱动法：使用查询代替逻辑语句

package com.zibo.ifelse;

// 表驱动法：数组小技巧
public class Method04 {
    public static void main(String[] args) {
        // 原始写法
        System.out.println(doSth(1));
        // 数组优化
        System.out.println(doSthNew(1));
    }

    // 原始写法
    private static int doSth(int month) {
        if (month == 1)  return 31;
        if (month == 2)  return 29;
        if (month == 3)  return 31;
        if (month == 4)  return 30;
        if (month == 5)  return 31;
        if (month == 6)  return 30;
        if (month == 7)  return 31;
        if (month == 8)  return 31;
        if (month == 9)  return 30;
        if (month == 10)  return 31;
        if (month == 11)  return 30;
        if (month == 12)  return 31;
        return 0;
    }

    // 数组优化
    private static int doSthNew(int month) {
        int[] monthDays = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        return monthDays[--month];
    }
}

5、反射

package com.zibo.ifelse;

import java.lang.reflect.Method;

// 反射
public class Method05 {
    public static void main(String[] args) throws Exception {
        // 原始写法
        doSth("doOne");
        // 反射写法
        doSthNew("doOne");
    }

    // 原始写法
    private static void doSth(String ds) {
        if("doOne".equals(ds)){
            new DoSth().doOne();
        }else if("doTwo".equals(ds)){
            new DoSth().doTwo();
        }else if("doThree".equals(ds)){
            new DoSth().doThree();
        }else {
            System.out.println("啥也不干！");
        }
    }

    // 反射写法
    private static void doSthNew(String ds) throws Exception {
        Method method = DoSth.class.getDeclaredMethod(ds);
        DoSth doSth = DoSth.class.newInstance();
        method.invoke(doSth);
    }
}
class DoSth{
    public void doOne(){
        System.out.println("One");
    }

    public void doTwo(){
        System.out.println("Two");
    }

    public void doThree(){
        System.out.println("Three");
    }
}

6、方法引用

package com.zibo.ifelse;

import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;

// 方法引用
public class Method05 {
    public static void main(String[] args) {
        // 原始写法
        doSth("doOne");
        // 方法引用
        doSthNew("doOne");
    }

    // 原始写法
    private static void doSth(String ds) {
        if("doOne".equals(ds)){
            new DoSth().doOne();
        }else if("doTwo".equals(ds)){
            new DoSth().doTwo();
        }else if("doThree".equals(ds)){
            new DoSth().doThree();
        }else {
            System.out.println("啥也不干！");
        }
    }

    // 方法引用
    private static void doSthNew(String ds){
        Map<String, Consumer<DoSth>> functionMap = new HashMap<>();
        functionMap.put("doOne", DoSth::doOne);
        functionMap.put("doTwo", DoSth::doTwo);
        functionMap.put("doThree", DoSth::doThree);
        functionMap.get(ds).accept(new DoSth());
    }
}
class DoSth{
    public void doOne(){
        System.out.println("One");
    }

    public void doTwo(){
        System.out.println("Two");
    }

    public void doThree(){
        System.out.println("Three");
    }
}