diff --git a/src/com/engine/organization/entity/chart/CompanyTreePO.java b/src/com/engine/organization/entity/chart/CompanyTreePO.java index 54918bac..ce287e19 100644 --- a/src/com/engine/organization/entity/chart/CompanyTreePO.java +++ b/src/com/engine/organization/entity/chart/CompanyTreePO.java @@ -23,6 +23,7 @@ public class CompanyTreePO { private String key; private boolean disabled; private int level; + private int depth; public boolean getIsLeaf() { return isLeaf; diff --git a/src/com/engine/organization/service/impl/ChartServiceImpl.java b/src/com/engine/organization/service/impl/ChartServiceImpl.java index 89b91174..eda5af50 100644 --- a/src/com/engine/organization/service/impl/ChartServiceImpl.java +++ b/src/com/engine/organization/service/impl/ChartServiceImpl.java @@ -554,6 +554,7 @@ public class ChartServiceImpl extends Service implements ChartService { private void getChildDepartment(CompanyTreePO parentDepartment, String sql, List departmentTree) { RecordSet rs = new RecordSet(); rs.executeQuery(sql, parentDepartment.getId()); + int depth=1; while (rs.next()) { CompanyTreePO build = CompanyTreePO.builder() .id(rs.getString("id")) @@ -566,7 +567,9 @@ public class ChartServiceImpl extends Service implements ChartService { .build(); getChildDepartment(build, sql, departmentTree); parentDepartment.setLeaf(false); + depth = build.getDepth()+1; } + parentDepartment.setDepth(depth); departmentTree.add(parentDepartment); }