Java枚举Direction的ofDegrees方法实现：返回枚举或空Optional

原代码存在的核心问题 #

• 错误的对象比较：循环中用dir == obj比较枚举实例与Integer对象，两者类型完全不同，永远不会匹配，导致始终返回null。
• 返回类型不符合要求：方法声明返回Direction，但需求是无匹配时返回空Optional，因此返回类型需改为Optional<Direction>。
• 冗余的Optional包装：输入参数是int基本类型，不可能为null，Optional.ofNullable(degrees)及后续的空检查完全多余。

修正后的代码 #

public enum Direction {
    N(0), NE(45), E(90), SE(135), S(180), SW(225), W(270), NW(315);

    private final int degrees;

    Direction(final int degrees) {
        this.degrees = degrees;
    }

    public static Optional<Direction> ofDegrees(int degrees) {
        for (Direction dir : Direction.values()) {
            if (dir.degrees == degrees) {
                return Optional.of(dir);
            }
        }
        return Optional.empty();
    }
}

关键修正点说明 #

• 返回类型调整：将方法返回类型改为Optional<Direction>，匹配时用Optional.of(dir)返回枚举实例，无匹配时用Optional.empty()返回空Optional，完全满足需求。
• 正确的匹配逻辑：直接比较枚举实例的degrees字段与输入的角度值（均为int类型，用==即可），找到匹配项立即返回。
• 简化冗余代码：移除不必要的Optional包装和空检查，让逻辑更简洁清晰。

可选优化（针对高频调用场景） #

如果该方法会被频繁调用，可提前构建角度到枚举的映射，避免每次遍历所有枚举值：

private static final Map<Integer, Direction> DEGREE_TO_DIRECTION = new HashMap<>();

static {
    for (Direction dir : Direction.values()) {
        DEGREE_TO_DIRECTION.put(dir.degrees, dir);
    }
}

public static Optional<Direction> ofDegrees(int degrees) {
    return Optional.ofNullable(DEGREE_TO_DIRECTION.get(degrees));
}

这种方式借助HashMap的O(1)查找性能，适合高频率调用的场景。

内容的提问来源于stack exchange，提问作者Kostadin Samardjiev