package cn.lili.modules.permission.serviceimpl;
|
|
import cn.lili.common.enums.ResultCode;
|
import cn.lili.common.exception.ServiceException;
|
import cn.lili.modules.permission.entity.dos.AdminUser;
|
import cn.lili.modules.permission.entity.dos.Department;
|
import cn.lili.modules.permission.entity.vo.DepartmentVO;
|
import cn.lili.modules.permission.mapper.DepartmentMapper;
|
import cn.lili.modules.permission.service.AdminUserService;
|
import cn.lili.modules.permission.service.DepartmentRoleService;
|
import cn.lili.modules.permission.service.DepartmentService;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 部门业务层实现
|
*
|
* @author Chopper
|
* @since 2020/11/17 3:47 下午
|
*/
|
@Slf4j
|
@Service
|
public class DepartmentServiceImpl extends ServiceImpl<DepartmentMapper, Department> implements DepartmentService {
|
|
/**
|
* 管理员
|
*/
|
@Autowired
|
private AdminUserService adminUserService;
|
/**
|
* 部门角色
|
*/
|
@Autowired
|
private DepartmentRoleService departmentRoleService;
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void deleteByIds(List<String> ids) {
|
QueryWrapper<AdminUser> queryWrapper = new QueryWrapper<>();
|
queryWrapper.in("department_id", ids);
|
if (adminUserService.count(queryWrapper) > 0) {
|
throw new ServiceException(ResultCode.PERMISSION_DEPARTMENT_DELETE_ERROR);
|
}
|
this.removeByIds(ids);
|
departmentRoleService.deleteByDepartment(ids);
|
}
|
|
@Override
|
public List<DepartmentVO> tree(QueryWrapper<Department> initWrapper) {
|
try {
|
List<Department> departments = this.list(initWrapper);
|
|
List<DepartmentVO> all = new ArrayList<>();
|
departments.forEach(item -> all.add(new DepartmentVO(item)));
|
|
List<DepartmentVO> tree = new ArrayList<>();
|
all.forEach(item -> {
|
if ("0".equals(item.getParentId())) {
|
initChild(item, all);
|
tree.add(item);
|
}
|
});
|
|
return tree;
|
} catch (Exception e) {
|
log.error("部门业务错误", e);
|
return null;
|
}
|
}
|
|
|
/**
|
* 递归初始化子树
|
*
|
* @param tree 树结构
|
* @param departmentVOS 数据库对象集合
|
*/
|
private void initChild(DepartmentVO tree, List<DepartmentVO> departmentVOS) {
|
departmentVOS.stream()
|
.filter(item -> (item.getParentId().equals(tree.getId())))
|
.forEach(child -> {
|
DepartmentVO childTree = new DepartmentVO(child);
|
initChild(childTree, departmentVOS);
|
tree.getChildren().add(childTree);
|
});
|
}
|
|
|
}
|
