技术总结 | 策略模式

问题：如何优化代码中大量的 if/else ?

1. 提前 return。
2. 使用 switch ... case。
3. 使用策略模式。

什么是策略模式

Define a family of algorithms,encapsulate each one,and make them interchangeable.
定义一组算法，将每个算法都封装起来，并且使它们之间可以互换。

策略模式使用的就是面向对象的继承和多态机制，从而实现同一行为在不同场景下具备不同实现。在策略模式中，我们创建表示各种策略模式的对象Strategy和一个行为随着策略对象而改变的Context对象。
策略模式本质：分离算法，选择实现。

策略模式的实现

1. 定义接口：

public interface Strategy {
    int doOperation(int num1, int num2) throws Exception;
}

2. 接口实现一：两数相加。

public class AddStrategy implements Strategy {
    @Override
    public int doOperation(int num1, int num2) throws Exception {
        return num1 + num2;
    }
}

3. 接口实现二：两数相减。

public class SubStrategy implements Strategy {
    @Override
    public int doOperation(int num1, int num2) throws Exception {
        return num1 - num2;
    }
}

4. 接口实现三：两数相乘。

public class MultiplyStrategy implements Strategy {
    @Override
    public int doOperation(int num1, int num2) throws Exception {
        return num1 * num2;
    }
}

5. 定义Context对象

public class Context {
    private Strategy strategy;
    public Context(Strategy strategy) {
        this.strategy = strategy;
    }
    public int executeStrategy(int num1, int num2) throws Exception {
        return strategy.doOperation(num1, num2);
    }
}

6. 策略模式使用。

public class StrategyTest {
    public static void main(String[] args) throws Exception {
        Context context = new Context(new AddStrategy());
        System.out.println(context.executeStrategy(9, 5));

        context = new Context(new SubStrategy());
        System.out.println(context.executeStrategy(9, 5));

        context = new Context(new MultiplyStrategy());
        System.out.println(context.executeStrategy(9, 5));
    }
}

定义枚举策略模式简化代码

public enum StrategyEnum {
    ADD {
        @Override
        public int calc(int num1, int num2) {
            return num1 + num2;
        }
    },
    SUB {
        @Override
        public int calc(int num1, int num2) {
            return num1 - num2;
        }
    };
    abstract int calc(int num1, int num2);
}

public class StrategyTest {
    public static void main(String[] args) throws Exception {
        int result = StrategyEnum.ADD.calc(1, 2);
        System.out.println(result);

        result = StrategyEnum.SUB.calc(5, 1);
        System.out.println(result);
    }
}

JDK8 新特性：学会使用Optional

1. 一般情况下逻辑处理：

if(user==null){
// do action 1
}else{
// do action 2
}

2. 使用Optional

Optional<User> userOptional = Optional.ofNullable(user);
userOptional.map(action1).orElse(action2);

Optional<User> userOptional = Optional.ofNullable(user);
userIptional.map(action1).orElse(action2);