java Enum 枚举级联，两层父子级

1，在实际开发中，可能会遇到各种奇怪的业务需求。现在将遇到的其中一种情况分享出来看看。
固定内容的两个枚举，存在关联性，父子关联。希望输出 树形结构数据。

• 包装对象

package model.enums;

import lombok.Data;

import java.util.ArrayList;
import java.util.List;

@Data
public class ItemEum {


    private  List<ItemEum> ch =new ArrayList<>();
    private Integer code;
    private String remark;
    private String name;

    @Override
    public String toString() {
        return "ItemEum{" +
                "ch=" + ch +
                ", code=" + code +
                ", remark='" + remark + '\'' +
                ", name='" + name + '\'' +
                '}';
    }
}

• 父枚举

package model.enums;


import com.alibaba.fastjson.JSONObject;

import java.util.Arrays;



public enum MessageTypeEnum {

    ORDER(1, "订单提醒","ORDER"),
    REFUND(2, "退款提醒","REFUND"),
    AFTERSALE(3, "售后提醒","AFTERSALE"),
    COUPON(4, "优惠券提醒","COUPON"),
    ACTIVITY(5, "活动通知","ACTIVITY"),
    COMMENT(6, "评论通知","COMMENT");


    private Integer code;
    private String remark;
    private String name;

    public Integer getCode() {
        return code;
    }

    public void setCode(Integer code) {
        this.code = code;
    }

    public String getRemark() {
        return remark;
    }

    public void setRemark(String remark) {
        this.remark = remark;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    MessageTypeEnum(Integer code, String remark, String name) {
        this.code = code;
        this.remark = remark;
        this.name = name;
    }

    public static Boolean getMessageTyFlagByName(String Name) {
        return  Arrays.stream(MessageTypeEnum.values()).anyMatch(i->i.getName().equals(Name));
    }

    public static ItemEum getItemEumByName(String Name) {
        ItemEum itemEum  = new ItemEum();
        Arrays.stream(MessageTypeEnum.values()).forEach(i->{
             if(i.getName().equals(Name)){
                 itemEum.setRemark(i.getRemark());
                 itemEum.setName(Name);
                 itemEum.setCode(i.getCode());
             }
        });
        return itemEum;
    }


    public static void main(String[] args) {

        System.out.println(JSONObject.toJSON(ChildTypeEnum.getChildTypeEnumList()));
    }
}

• 子枚举

package model.enums;


import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

/**
 * zzx
 */
public enum ChildTypeEnum {

    ORDER(10, "订单提醒",MessageTypeEnum.ORDER.name()),
    ORDER2(20, "订单提醒2",MessageTypeEnum.ORDER.name()),

    REFUND(30, "退款提醒",MessageTypeEnum.REFUND.name()),
    REFUND2(40, "退款提醒2",MessageTypeEnum.REFUND.name()),

    AFTERSALE(50, "售后提醒",MessageTypeEnum.AFTERSALE.name()),
    AFTERSALE2(60, "售后提醒2",MessageTypeEnum.AFTERSALE.name()),

    COUPON(70, "优惠券提醒",MessageTypeEnum.COUPON.name()),
    COUPON2(80, "优惠券提醒2",MessageTypeEnum.COUPON.name()),

    ACTIVITY(90, "活动通知",MessageTypeEnum.ACTIVITY.name()),
    ACTIVITY2(100, "活动通知2",MessageTypeEnum.ACTIVITY.name()),

    COMMENT(110, "评论通知",MessageTypeEnum.COMMENT.name()),
    COMMENT2(120, "评论通知2",MessageTypeEnum.COMMENT.name());


    private Integer code;
    private String remark;
    private String name;

    public Integer getCode() {
        return code;
    }

    public void setCode(Integer code) {
        this.code = code;
    }

    public String getRemark() {
        return remark;
    }

    public void setRemark(String remark) {
        this.remark = remark;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    ChildTypeEnum(Integer code, String remark, String name) {
        this.code = code;
        this.remark = remark;
        this.name = name;
    }

    public static ItemEum getChildItemEum(Integer code){
        ItemEum itemEum  = new ItemEum();
        Arrays.stream(ChildTypeEnum.values()).forEach(i->{
            if(i.getCode().equals(code)){
                itemEum.setRemark(i.getRemark());
                itemEum.setName(i.getName());
                itemEum.setCode(i.getCode());
            }
        });
        return itemEum;
    }
    public static List<ItemEum> getChildTypeEnumList() {
        final List<ItemEum> li = new ArrayList<>();
          Arrays.stream(ChildTypeEnum.values()).forEach(s->{
            if(MessageTypeEnum.getMessageTyFlagByName(s.getName())){
              // 获取一级权益
                if(!li.stream().anyMatch(an->an.getName().equals(s.getName()))){
                    li.add(MessageTypeEnum.getItemEumByName(s.getName()));
                }
                //组装二级权益
                li.stream().forEach(l->{
                    if(l.getName().equals(s.getName())){
                       l.getCh().add(getChildItemEum(s.getCode()));
                    }
                });
            }
        });
        return li.stream().sorted(Comparator.comparing(ItemEum::getName)
                .thenComparing(ItemEum::getCode)
                .thenComparing(ItemEum::getRemark)).collect(Collectors.toList());
    }

}

执行main 方法后，得到的数据结构如下：

[{
    "code": 1,
    "ch": [{
        "code": 10,
        "ch": [],
        "name": "ORDER",
        "remark": "订单提醒"
    }, {
        "code": 20,
        "ch": [],
        "name": "ORDER",
        "remark": "订单提醒2"
    }],
    "name": "ORDER",
    "remark": "订单提醒"
}, {
    "code": 2,
    "ch": [{
        "code": 30,
        "ch": [],
        "name": "REFUND",
        "remark": "退款提醒"
    }, {
        "code": 40,
        "ch": [],
        "name": "REFUND",
        "remark": "退款提醒2"
    }],
    "name": "REFUND",
    "remark": "退款提醒"
}, {
    "code": 3,
    "ch": [{
        "code": 50,
        "ch": [],
        "name": "AFTERSALE",
        "remark": "售后提醒"
    }, {
        "code": 60,
        "ch": [],
        "name": "AFTERSALE",
        "remark": "售后提醒2"
    }],
    "name": "AFTERSALE",
    "remark": "售后提醒"
}, {
    "code": 4,
    "ch": [{
        "code": 70,
        "ch": [],
        "name": "COUPON",
        "remark": "优惠券提醒"
    }, {
        "code": 80,
        "ch": [],
        "name": "COUPON",
        "remark": "优惠券提醒2"
    }],
    "name": "COUPON",
    "remark": "优惠券提醒"
}, {
    "code": 5,
    "ch": [{
        "code": 90,
        "ch": [],
        "name": "ACTIVITY",
        "remark": "活动通知"
    }, {
        "code": 100,
        "ch": [],
        "name": "ACTIVITY",
        "remark": "活动通知2"
    }],
    "name": "ACTIVITY",
    "remark": "活动通知"
}, {
    "code": 6,
    "ch": [{
        "code": 110,
        "ch": [],
        "name": "COMMENT",
        "remark": "评论通知"
    }, {
        "code": 120,
        "ch": [],
        "name": "COMMENT",
        "remark": "评论通知2"
    }],
    "name": "COMMENT",
    "remark": "评论通知"
}]