如何用Java将MySQL企业、部门、职位数据转换为树形JSON结构？

本文介绍如何使用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 data = new ArrayList<>();
Company curCompany;
Department curDepartment;
Map companyMap = new HashMap<>();
Map 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字符串。