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;
    }
}