在項目中我們經常會通過一個父級的ID來獲取所有子級的數據,下面是常用的實現方式。
數據庫表結構數據如下:
Pojo類:
public class TDistrict {
private Integer districtId;
private Integer districtType;
private String districtName;
private Integer parentId;
private String districtLevStatus;
private Date creationTime;
private String code;
//getter、setter
}
Java代碼實現:
public class AreaUtils {
/**
* 根據當前節點獲得本節點及所有的子節點
* @param list : 數據庫所有的區域集合
* @param areaId : 要查詢的節點
* @param res : 返回的結果集
* @return
*/
public static List<TDistrict> areaR(List<TDistrict> list,Integer areaId,List<TDistrict> res) {
//根據areaId獲取這個區域的信息
TDistrict areaInfo = list.stream().filter(area -> (area.getDistrictId().intValue() == areaId.intValue())).collect(Collectors.toList()).get(0);
//將父節點添加到結果集
res.add(areaInfo);
//沒有子節點
if(areaInfo.getDistrictType() == 2) {
return res;
}
//遍歷子節點
for (int i = 0; i < list.size(); i++) {
if(list.get(i).getParentId().intValue() == areaId.intValue()) {
Integer subAreaId = list.get(i).getDistrictId();
areaR(list,subAreaId,res);
}
}
return res;
}
}