JAVA 枚举 获取上下级逻辑

@Getter
public enum LevelEnum {
    /**
     * 一级节点
     */

    ONE_NODE("ONE_NODE", null),
    /**
     * 二级节点
     */
    TWO_NODE("TWO_NODE", ONE_NODE),
    /**
     * 三级节点
     */
    THREE_NODE("THREE_NODE", TWO_NODE),
    /**
     * 四级节点
     */
    FOUR_NODE("FOUR_NODE", THREE_NODE);

    @EnumValue
    private final String value;
    // 上级节点
    private final LevelEnum parent;
    // 唯一的下级节点
    private LevelEnum child;

    // 构造函数:自动绑定父子关系
    LevelEnum (String value, LevelEnum parent) {
        this.value = value;
        this.parent = parent;
        // 初始时无下级
        this.child = null;

        // 自动为父节点设置当前节点作为其下级
        if (parent != null) {
            parent.child = this;
        }
    }

    // 是否为根节点
    public boolean isRoot() {
        return parent == null;
    }

    // 是否为末端节点(无下级)
    public boolean isLeaf() {
        return child == null;
    }

    // 根据数值查找对应的节点
    public static LevelEnum  fromValue(String value) {
        for (LevelEnum node : values()) {
            if (Objects.equals(node.value, value)) {
                return node;
            }
        }
        throw new IllegalArgumentException("无效数值: " + value);
    }

    // 获取从当前节点到根节点的路径
    public List<LevelEnum > getPathToRoot() {
        List<LevelEnum > path = new ArrayList<>();
        LevelEnum current = this;
        while (current != null) {
            path.add(current);
            current = current.parent;
        }
        // 调整为根节点在前的顺序
        Collections.reverse(path);
        return path;
    }

    // 获取从当前节点到末端节点的路径
    public List<LevelEnum > getPathToLeaf() {
        List<LevelEnum > path = new ArrayList<>();
        LevelEnum current = this;
        while (current != null) {
            path.add(current);
            current = current.child;
        }
        return path;
    }

    /**
     * 获取子节点的值
     *
     * @return 如果子节点不为空,则返回子节点的值;否则返回null
     */
    public String getChildValue() {
        if (child != null) {
            return child.getValue();
        }
        return null;
    }

    // 获取第一级节点
    public static LevelEnum getFirstLevel() {
        for (LevelEnum node : values()) {
            if (node.isRoot()) {
                return node;
            }
        }
        return null;
    }

    public static String getFirstLevelValue() {
        for (LevelEnum node : values()) {
            if (node.isRoot()) {
                return node.getValue();
            }
        }
        return null;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值