枚举 & 表驱动

【枚举 & 表驱动】

初级版本

先看一段代码

/**
 * if/else 版本
 */
public void trafficIfElseVersion(String trafficLight) {
    if ("红色".equals(trafficLight)) {
        System.out.println("现在是红灯，请停下");
    } else if ("绿色".equals(trafficLight)) {
        System.out.println("现在是绿灯，请通行");
    } else if ("黄灯".equals(trafficLight)) {
        System.out.println("现在是黄灯，请注意");
    }
}

/**
 * switch 版本
 */
public void trafficSwitchVersion(String trafficLight) {

    switch (trafficLight) {
        case "红色":
            System.out.println("现在是红灯，请停下");
            break;
        case "绿色":
            System.out.println("现在是绿灯，请通行");
            break;
        case "黄色":
            System.out.println("现在是黄灯，请注意");
            break;
        default:
            break;
    }
}

上面两个版本有本质性区别吗？没有，对每个颜色的灯做一些响应，条件要写全

啰嗦进化

再来看一个版本

// map 里封装了条件对应的动作
private static final Map<String, Consumer> TRAFFIC_LIGHT_TO_ACTION = new HashMap<>();

static {
    TRAFFIC_LIGHT_TO_ACTION.put("红色", o -> System.out.println("现在是红灯，请停下"));
    TRAFFIC_LIGHT_TO_ACTION.put("绿色", o -> System.out.println("现在是绿灯，请通行"));
    TRAFFIC_LIGHT_TO_ACTION.put("黄色", o -> System.out.println("现在是黄灯，请注意"));
}

public static void trafficMapVersion(String trafficLight) {
    // 不考虑非法值，只说明方向
    TRAFFIC_LIGHT_TO_ACTION.get(trafficLight).accept(null);
}

这个版本里简单动作（方法）可以这么搞，但还是要硬编码进去，而且对于 trafficLight 这样 String 类型，一直出现在代码中，属于魔法值，复杂的方法可以把各种灯抽象出来类，实例化后作为 value 添加进去

再进化

import lombok.AllArgsConstructor;
import lombok.Getter;

import java.util.Arrays;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

@Getter
@AllArgsConstructor
public enum TrafficLight {

    RED("红灯") {
        @Override
        public void action() {
            System.out.println("现在是红灯，请停下");
        }
    },
    GREEN("绿灯") {
        @Override
        public void action() {
            System.out.println("现在是绿灯，请通行");
        }
    },
    YELLOW("黄灯") {
        @Override
        public void action() {
            System.out.println("现在是黄灯，请注意");
        }
    };

    private String color;

    /**
     * 用于存储 color: Enum
     * 程序构建，不需要手动添加了
     * <p>
     * 在上一个的手动 map 里也可以把 各种灯抽象出类，实例化后和颜色对应，
     * 手动对应也可，反射扫描添加也可
     * 枚举提供了这种功能就直接用了
     */
    private static final Map<String, TrafficLight> COLOR_TO_ENUM =
            Arrays.stream(TrafficLight.values())
                    .collect(Collectors.toMap(TrafficLight::getColor, e -> e));

    /**
     * 提供通过 color 字符拿到实例的方法
     *
     * 这里使用了 Optional 语义上表示如果 code 不存在(非法值)，起一个提示作用
     * 
     * @param color 颜色
     * @return 对应的实例
     */
    public static Optional<TrafficLight> getEnumByColorString(String color) {
        return Optional.ofNullable(COLOR_TO_ENUM.get(color));
    }

    /**
     * 抽象方法
     * 复杂的参数也可以搞
     */
    public abstract void action();

    public static void main(String[] args) {

        // 使用，没有了 if/else 和 switch 主逻辑更加清晰
        // 实际的条件转移到了 map 里的状态
        String color = "红色";
        Optional<TrafficLight> trafficLightOpt = getEnumByColorString(color);
        if (trafficLightOpt.isPresent()) {
            // 这里体现了多态，具体实例去调用自己的实现
            trafficLightOpt.get().action();
        }
    }
}

总结：

• 构建表 (MAP) 代替条件判断
• 抽象方法，多态调用具体实现
• 虽然代码没有少写，但是主逻辑确实清晰了许多，不同实例的不同实现封装到内部，清爽。

姊妹篇 枚举的常用姿势 (JAVA 为例)

枚举 & 表驱动的方式只适用于一些比较固定的情况，如果要结合 Spring 需要注入 service 和 dao 这种对象的时候，就不太合适了，这个时候该怎么办呢？
参见 Spring & 表驱动