java 递归 树形

1. //菜单树形结构
2. public JSONArray treeMenuList(JSONArray menuList, int parentId) {
3. JSONArray childMenu = new JSONArray();
4. for (Object object : menuList) {
5. JSONObject jsonMenu = JSONObject.fromObject(object);
6. int menuId = jsonMenu.getInt("id");
7. int pid = jsonMenu.getInt("parentId");
8. if (parentId == pid) {
9. JSONArray c_node = treeMenuList(menuList, menuId);
10. jsonMenu.put("childNode", c_node);
11. childMenu.add(jsonMenu);
12. }
13. }
14. return childMenu;
15. }
16. public static void main(String args[]) {
17. MenuCacheService menuCacheService = new MenuCacheService();
18. JSONArray jsonArray = new JSONArray();
19. Menu menu1 = new Menu();
20. menu1.setId(1l);
21. menu1.setParentId(0);
22. menu1.setLevel(0);
23. Menu menu2 = new Menu();
24. menu2.setId(2l);
25. menu2.setParentId(0);
26. menu2.setLevel(0);
27. Menu menu3 = new Menu();
28. menu3.setId(3l);
29. menu3.setParentId(2);
30. menu3.setLevel(1);
31. Menu menu4 = new Menu();
32. menu4.setId(4l);
33. menu4.setParentId(2);
34. menu4.setLevel(1);
35. Menu menu5 = new Menu();
36. menu5.setId(5l);
37. menu5.setParentId(4);
38. menu5.setLevel(2);
39. Menu menu6 = new Menu();
40. menu6.setId(6l);
41. menu6.setParentId(1);
42. menu6.setLevel(1);
43. jsonArray.add(menu1);
44. jsonArray.add(menu2);
45. jsonArray.add(menu3);
46. jsonArray.add(menu4);
47. jsonArray.add(menu5);
48. jsonArray.add(menu6);
49. System.out.print(menuCacheService.treeMenuList(jsonArray,0));
50. }