Java精简返回给前端的结果集

需求：

前端页面需要展示三级菜单，后端返回结果集。问题在于，JavaBean中的属性值比实际需要的数量多，为了节省后端传递给前端的流量，将不需要的属性值设为null，这样转换为Json字符串的长度会小很多。

难点在于如何将id和parentid设置为null

数据结构要求如下：

[
  {
    name:"一级菜单"
      menus:[ 
         {
          name:"二级菜单"
          menus:[
             {
                name:"三级菜单"
             },
             .......
          ]
       },
       .......
      ]
  },
  .......
]

数据库中数据表：

CREATE TABLE `tb_ad` (
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT 'ID',
  `name` varchar(50) DEFAULT NULL COMMENT '广告名称',
  `position` varchar(50) DEFAULT NULL COMMENT '广告位置',
  `start_time` datetime DEFAULT NULL COMMENT '开始时间',
  `end_time` datetime DEFAULT NULL COMMENT '到期时间',
  `status` char(1) DEFAULT NULL COMMENT '状态',
  `image` varchar(100) DEFAULT NULL COMMENT '图片地址',
  `url` varchar(100) DEFAULT NULL COMMENT 'URL',
  `remarks` varchar(1000) DEFAULT NULL COMMENT '备注',
  PRIMARY KEY (`id`)
)

对应的JavaBean如下：

@Table(name="tb_category")
public class Category implements Serializable{

    @Id
    private Integer id;//分类ID

    private String name;//分类名称

    private Integer goodsNum;//商品数量

    private String isShow;//是否显示

    private String isMenu;//是否导航

    private Integer seq;//排序

    private Integer parentId;//上级ID

    private Integer templateId;//模板ID
    private List<Category> menus;
        //省略get/set

}

方式一，在双层嵌套循环的过程中，在适当位置将id和parentid赋值为null。

优点，代码编写简单，测试容易。缺点，对于非包装类型无能为力，逻辑有点绕。

    public List<Category> findCategoryTree() {
        //1.查找所有需要显示的category
        Example example = new Example(Category.class);
        Example.Criteria criteria = example.createCriteria();
        criteria.andEqualTo("isShow","1");
        example.setOrderByClause("seq");//排序
        List<Category> categories = categoryMapper.selectByExample(example);
        ArrayList<Category> parentList = new ArrayList<>();
        //2.通过递归的方式，给menus赋值
        for (Category category : categories) {
            //一级category
            if(Objects.nonNull(category.getParentId())&&category.getParentId()==0){
                //清除一级目录的parentId
                category.setParentId(null);
                parentList.add(category);
            }
            findChildren(categories,category);
            category.setGoodsNum(null);
            category.setIsMenu(null);
            category.setIsShow(null);
            category.setSeq(null);
            category.setTemplateId(null);
        }
        return parentList;
    }

    private void findChildren(List<Category> categories, Category category){
        List<Category> list=new ArrayList<>();
        for (Category category1 : categories) {
            if(category.getId().equals(category1.getParentId())){
                //到此说明已经找到它的father，那么parentId已经没有用了
                category1.setParentId(null);
                list.add(category1);
            }
        }
        //到此说明已经找到它所有的children，那么id已经没有用了
        category.setId(null);
        if(list.size()>0){
            category.setMenus(list);
        }
    }

方式二，讲树状的结果集转为json字符串，利用正则表达式替换掉不需要的属性，然后再将字符串转为Object。

优点，对于非包装类的属性也适用，代码重复利用性高。缺点，会生成多个字符串常量，占用内存稍高，正则表达式写起来麻烦，不容易测试。

    public Object findCategoryTree() {
        //1.查找所有需要显示的category
        Example example = new Example(Category.class);
        Example.Criteria criteria = example.createCriteria();
        criteria.andEqualTo("isShow","1");
        example.setOrderByClause("seq");//排序
        List<Category> categories = categoryMapper.selectByExample(example);
        ArrayList<Category> parentList = new ArrayList<>();
        //2.通过递归的方式，给menus赋值
        for (Category category : categories) {
            //一级category
            if(Objects.nonNull(category.getParentId())&&category.getParentId()==0){
                //清除一级目录的parentId
                //category.setParentId(null);
                parentList.add(category);
            }
            findChildren(categories,category);
            category.setGoodsNum(null);
            category.setIsMenu(null);
            category.setIsShow(null);
            category.setSeq(null);
            category.setTemplateId(null);
        }
       //3.通过自定义方法将"id","parentId"属性移除
        ArrayList<String> attributes = new ArrayList<>();
        attributes.add("id");
        attributes.add("parentId");
        return AttributeUtils.removeAttribute(parentList,attributes);
    }

    private void findChildren(List<Category> categories, Category category){
        List<Category> list=new ArrayList<>();
        for (Category category1 : categories) {
            if(category.getId().equals(category1.getParentId())){
                //到此说明已经找到它的father，那么parentId已经没有用了
                //category1.setParentId(null);
                list.add(category1);
            }
        }
        //到此说明已经找到它所有的children，那么id已经没有用了
        //category.setId(null);
        if(list.size()>0){
            category.setMenus(list);
        }
    }

自定义工具类

public class AttributeUtils {
    public static <T> Object removeAttribute(List<T> list, List<String> attributeNames) {
        //1.将list转为json字符串
        String jsonString = JSONObject.toJSONString(list);
        //2.删除掉json字符串中相关属性及属性值
        for (String attributeName : attributeNames) {
            //元素在开头
            String regex1 = "[^,]\""+attributeName+"\":*[^}]*?,";
            Pattern p1 = Pattern.compile(regex1);
            Matcher m1 = p1.matcher(jsonString);
            jsonString = m1.replaceAll("{");
            //元素在尾部
           String regex2 = ",\""+attributeName+"\":*[^{]*?}";
            Pattern p2 = Pattern.compile(regex2);
            Matcher m2 = p2.matcher(jsonString);
            jsonString=m2.replaceAll("}");
            //元素在中间
            String regex3 = ",\""+attributeName+"\":*[^{]*?,";
            Pattern p3 = Pattern.compile(regex3);
            Matcher m3 = p3.matcher(jsonString);
            jsonString=m3.replaceAll("");
        }
        Object parse = JSONObject.parse(jsonString);
        return parse;
    }
}