package com.ruoyi.system.service.impl;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.common.basic.PageInfo;
|
import com.ruoyi.common.core.domain.entity.TDept;
|
import com.ruoyi.system.mapper.TDeptMapper;
|
import com.ruoyi.system.service.TDeptService;
|
import org.springframework.stereotype.Service;
|
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 点位管理 服务实现类
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2025-05-28
|
*/
|
@Service
|
public class TDeptServiceImpl extends ServiceImpl<TDeptMapper, TDept> implements TDeptService {
|
|
/**
|
* 获取部门树结构
|
* @return 部门树列表
|
*/
|
public List<TDept> selectDeptTreeList() {
|
// 查询所有部门
|
List<TDept> depts = this.list();
|
// 构建树结构
|
return buildDeptTree(depts);
|
}
|
|
/**
|
* 构建部门树结构
|
* @param depts 部门列表
|
* @return 树结构的部门列表
|
*/
|
private List<TDept> buildDeptTree(List<TDept> depts) {
|
return depts.stream()
|
// 筛选出顶级部门（parentId为null或0的部门）
|
.filter(dept -> dept.getParentId() == null || dept.getParentId() == 0)
|
// 为每个顶级部门构建子树
|
.peek(dept -> dept.setChildren(getChildren(dept, depts)))
|
.collect(Collectors.toList());
|
}
|
|
/**
|
* 递归获取部门的子部门
|
* @param root 当前部门
|
* @param depts 所有部门列表
|
* @return 子部门列表
|
*/
|
private List<TDept> getChildren(TDept root, List<TDept> depts) {
|
return depts.stream()
|
// 筛选出当前部门的直接子部门
|
.filter(dept -> dept.getParentId() != null && dept.getParentId().equals(root.getId()))
|
// 为每个子部门递归构建其子树
|
.peek(dept -> dept.setChildren(getChildren(dept, depts)))
|
.collect(Collectors.toList());
|
}
|
}
|
