问题:如何优化代码中大量的 if/else ?
- 提前 return。
- 使用
switch ... case
。 - 使用策略模式。
什么是策略模式
Define a family of algorithms,encapsulate each one,and make them interchangeable.
定义一组算法,将每个算法都封装起来,并且使它们之间可以互换。
策略模式使用的就是面向对象的继承和多态机制,从而实现同一行为在不同场景下具备不同实现。在策略模式中,我们创建表示各种策略模式的对象Strategy
和一个行为随着策略对象而改变的Context
对象。
策略模式本质:分离算法,选择实现。
策略模式的实现
- 定义接口:
public interface Strategy {
int doOperation(int num1, int num2) throws Exception;
}
- 接口实现一:两数相加。
public class AddStrategy implements Strategy {
@Override
public int doOperation(int num1, int num2) throws Exception {
return num1 + num2;
}
}
- 接口实现二:两数相减。
public class SubStrategy implements Strategy {
@Override
public int doOperation(int num1, int num2) throws Exception {
return num1 - num2;
}
}
- 接口实现三:两数相乘。
public class MultiplyStrategy implements Strategy {
@Override
public int doOperation(int num1, int num2) throws Exception {
return num1 * num2;
}
}
- 定义
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);
}
}
- 策略模式使用。
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
- 一般情况下逻辑处理:
if(user==null){
// do action 1
}else{
// do action 2
}
- 使用Optional
Optional<User> userOptional = Optional.ofNullable(user);
userOptional.map(action1).orElse(action2);
Optional<User> userOptional = Optional.ofNullable(user);
userIptional.map(action1).orElse(action2);