本文介绍如何使用Java将MySQL数据库中的企业、部门、职位数据转换为树形JSON结构。 目标是将数据库数据转换成如下所示的层级结构:
[ { "code": "c1", "name": "企业A", "departments": [ { "code": "d1", "name": "财务部", "jobs": [ {"code": "j1", "name": "财务主管"}, {"code": "j2", "name": "会计"} ] }, { "code": "d2", "name": "技术部", "jobs": [ {"code": "j21", "name": "Java"}, {"code": "j22", "name": "JS"} ] } ] }, { "code": "c2", "name": "企业B", "departments": [ { "code": "d20", "name": "销售部", "jobs": [{"code": "j20", "name": "销售员"}] } ] } ]
我们将使用HashMap来高效地构建树形结构。以下代码片段展示了核心转换逻辑:
List<Company> data = new ArrayList<>(); Company curCompany; Department curDepartment; Map<String, Company> companyMap = new HashMap<>(); Map<String, Department> departmentMap = new HashMap<>(); for (Po po : records) { // records 为从数据库查询到的数据列表 // 处理企业 String companyKey = po.getCompanyCode(); curCompany = companyMap.getOrDefault(companyKey, new Company()); curCompany.setCode(po.getCompanyCode()); curCompany.setName(po.getCompanyName()); companyMap.put(companyKey, curCompany); if (!data.contains(curCompany)) { //避免重复添加企业 data.add(curCompany); } // 处理部门 String departmentKey = po.getCompanyCode() + po.getDeptCode(); curDepartment = departmentMap.getOrDefault(departmentKey, new Department()); curDepartment.setCode(po.getDeptCode()); curDepartment.setName(po.getDeptName()); departmentMap.put(departmentKey, curDepartment); curCompany.getDepartments().add(curDepartment); // 处理职位 Job job = new Job(); job.setCode(po.getJobCode()); //假设数据库字段名为JobCode job.setName(po.getJobName()); //假设数据库字段名为JobName curDepartment.getJobs().add(job); }
这段代码首先创建了两个HashMap,分别用于存储企业和部门对象,以方便快速查找和构建父子关系。 然后,它遍历数据库查询结果 records,依次处理企业、部门和职位信息,并将其添加到相应的HashMap中。 最后,data 列表包含了最终的树形结构数据,可以直接转换为JSON格式输出。 注意,代码中假设了数据库表结构和实体类属性名,需要根据实际情况进行调整。 getOrDefault 方法避免了冗余的 containsKey 检查。
最后,可以使用Jackson或Gson等JSON库将 data 列表转换为JSON字符串。