/**
|
* Copyright (c) 2018 人人开源 All rights reserved.
|
* <p>
|
* https://www.renren.io
|
* <p>
|
* 版权所有，侵权必究！
|
*/
|
|
package com.zt.modules.sys.service;
|
|
import cn.hutool.core.convert.Convert;
|
import com.alibaba.excel.util.StringUtils;
|
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.constant.Status;
|
import com.zt.common.db.query.QueryFilter;
|
import com.zt.common.entity.MapData;
|
import com.zt.common.exception.ErrorCode;
|
import com.zt.common.exception.RenException;
|
import com.zt.common.service.BaseService;
|
import com.zt.common.servlet.Result;
|
import com.zt.common.utils.CacheUtils;
|
import com.zt.common.utils.TreeUtils;
|
import com.zt.core.context.User;
|
import com.zt.core.context.UserContext;
|
import com.zt.core.sys.model.SysDept;
|
import com.zt.core.sys.service.ISysDeptService;
|
import com.zt.modules.sys.dao.SysDeptDao;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.cache.annotation.CacheEvict;
|
import org.springframework.cache.annotation.Cacheable;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* 部门管理
|
*
|
* @author hehz
|
*/
|
@Service
|
public class SysDeptService extends BaseService<SysDeptDao, SysDept> implements ISysDeptService {
|
|
@Autowired
|
private SysUserService sysUserService;
|
|
private List<SysDept> getAll() {
|
// 这里手动缓存，不然在当前方法内调用缓存无效
|
List<SysDept> all = null;//(List<SysDept>) CacheUtils.get(Constant.Cache.DEPT, "all");
|
if (all == null) {
|
all = baseDao.getAll();
|
//CacheUtils.put(Constant.Cache.DEPT, "all", all);
|
}
|
return all.stream().map(dept -> {
|
try {
|
return dept.clone();
|
} catch (CloneNotSupportedException e) {
|
}
|
return null;
|
}).collect(Collectors.toList());
|
}
|
|
public List<SysDept> getTenantAll() {
|
return getAll();
|
// return getAll().stream().filter(dept -> dept.getTenantId().equals(UserContext.getUser().getTenantId()))
|
// .collect(Collectors.toList());
|
}
|
|
/**
|
* 查询当前租户下公司树
|
*
|
* @return
|
*/
|
public List<SysDept> getCompanyTree() {
|
List<SysDept> list = getTenantAll().stream().filter(dept -> dept.isCompany()).collect(Collectors.toList());
|
// 查询所属公司及子公司的数据
|
List<Long> deptIds = this.getWithDescendantIds(UserContext.getUser().getCompanyId());
|
list = list.stream().filter(dept -> deptIds.contains(dept.getId())).collect(Collectors.toList());
|
List<SysDept> listOut = TreeUtils.build(list);
|
//List<SysDept> listOut2 = TreeUtils.toList(listOut);
|
return listOut;
|
}
|
|
/**
|
* 查询当前登录用户公司部门树
|
*
|
* @return
|
*/
|
public List<SysDept> getDeptTreeRegister() {
|
List<SysDept> list = getTenantAll();
|
|
List<SysDept> listArea = list.stream().filter(dept -> "sy".equals(dept.getCode())).collect(Collectors.toList());
|
if (listArea.size() > 0) {
|
Long deptId = listArea.get(0).getId();
|
List<Long> deptIds = this.getWithDescendantIds(deptId);
|
deptIds.remove(0);
|
list = list.stream().filter(dept -> deptIds.contains(dept.getId())).collect(Collectors.toList());
|
}
|
return TreeUtils.build(list);
|
}
|
|
public List<SysDept> getDeptTree() {
|
User user = UserContext.getUser();
|
List<SysDept> list = getTenantAll();
|
|
if (1==0 && !user.isSuperAdmin()) {
|
// 查询所属部门及子部门的数据
|
List<Long> deptIds = this.getWithDescendantIds(user.getDeptId());
|
list = list.stream().filter(dept -> deptIds.contains(dept.getId())).collect(Collectors.toList());
|
}
|
|
return TreeUtils.build(list);
|
}
|
|
/**
|
* 查询公司下部门树
|
*
|
* @param companyId
|
* @return
|
*/
|
public List<SysDept> getDeptList(Long companyId) {
|
QueryFilter queryFilter = new QueryFilter();
|
// 普通管理员，只能查询所属部门及子部门的数据
|
User user = UserContext.getUser();
|
if (!user.isSuperAdmin()) {
|
queryFilter.addParam("deptIds", this.getWithDescendantIds(user.getDeptId()));
|
}
|
// 查询公司下部门列表
|
queryFilter.addParam("isCompany", Constant.Bool.NO);
|
queryFilter.addParam("companyId", companyId);
|
|
// 查询部门列表
|
List<SysDept> entityList = baseDao.getList(queryFilter.getParams());
|
|
return TreeUtils.build(entityList);
|
}
|
|
@Override
|
@Cacheable(value = Constant.Cache.DEPT, key = "'id:' + #id")
|
public SysDept get(Long id) {
|
return baseDao.getById(id);
|
}
|
|
/**
|
* 通过deptId查找所有祖先，按照祖先的层级排序
|
*
|
* @param id
|
* @return
|
*/
|
@Override
|
public List<SysDept> getAncestor(Long id) {
|
return TreeUtils.getAncestor(getTenantAll(), id);
|
}
|
|
/**
|
* 获取自己和所有下级
|
*
|
* @param id
|
* @return
|
*/
|
public List<SysDept> getWithDescendant(Long id) {
|
return TreeUtils.getWithDescendant(getTenantAll(), id);
|
}
|
|
@Override
|
@Cacheable(value = Constant.Cache.DEPT, key = "'withdescendant:' + #id")
|
public List<Long> getWithDescendantIds(Long id) {
|
return this.getWithDescendant(id).stream().map(dept -> dept.getId()).collect(Collectors.toList());
|
}
|
|
|
/**
|
* 获取下级
|
*
|
* @param id
|
* @return
|
*/
|
@Override
|
public List<SysDept> getDescendant(Long id) {
|
return TreeUtils.getDescendant(getTenantAll(), id);
|
}
|
|
/**
|
* 获取某部门的子节点
|
*
|
* @param id
|
* @return
|
*/
|
@Override
|
public List<SysDept> getChildren(Long id) {
|
return getTenantAll().stream().filter(dept -> dept.getPid().equals(id)).collect(Collectors.toList());
|
}
|
|
public List<SysDept> getByTypes(List<String> types) {
|
return getTenantAll().stream().filter(dept -> types.contains(dept.getNature())).collect(Collectors.toList());
|
}
|
|
/**
|
* 增加部门
|
*
|
* @param dept
|
*/
|
@Override
|
@CacheEvict(value = Constant.Cache.DEPT, allEntries = true)
|
public void insert(SysDept dept) {
|
if (dept.getId() == null || dept.getId() == 0) {
|
dept.setId(IdWorker.getId());
|
}
|
dept.setStatus(Status.ENABLE.getValue());
|
resetFullNameAndInnerId(dept, null);
|
|
// 部门类型为公司时，生成id
|
if (dept.isCompany()) {
|
dept.setCompanyId(dept.getId());
|
}
|
|
super.insert(dept);
|
|
// 更新数据范围表
|
// dataScopeService.updateByDeptAdd(dept.getId(), dept.getParentId());
|
}
|
|
/**
|
* 更新部门
|
*
|
* @param dept
|
*/
|
@Override
|
@CacheEvict(value = Constant.Cache.DEPT, allEntries = true)
|
public void update(SysDept dept) {
|
Long id = dept.getId();
|
if (id == 0) {
|
throw new RenException("获取主键失败");
|
}
|
|
SysDept tmp = get(id);
|
TreeUtils.updateValidate(getTenantAll(), dept, tmp);
|
|
if (!tmp.getPid().equals(dept.getPid())) {// 部门层级改变了
|
resetFullNameAndInnerId(dept, null);
|
}
|
|
super.update(dept);
|
|
// if (!tmp.getPid().equals(pid)) {// 部门层级改变了
|
// // 更新数据范围表
|
// dataScopeService.updateByDeptUpdate(dept.getId(),
|
// dept.getParentId());
|
// }
|
}
|
|
/**
|
* 删除部门
|
*
|
* @param id
|
*/
|
@Override
|
@CacheEvict(value = Constant.Cache.DEPT, allEntries = true)
|
@Transactional(rollbackFor = Exception.class)
|
public void delete(Long id) {
|
// 判断是否有子部门
|
List<Long> subList = this.getWithDescendantIds(id);
|
if (subList.size() > 1) {
|
throw new RenException(ErrorCode.DEPT_SUB_DELETE_ERROR);
|
}
|
|
// 判断部门下面是否有用户
|
int count = sysUserService.getByDeptId(id).size();
|
if (count > 0) {
|
throw new RenException(ErrorCode.DEPT_USER_DELETE_ERROR);
|
}
|
|
// 删除
|
this.deleteLogic(id);
|
}
|
|
/**
|
* 修改全称和内部编码
|
*
|
* @param dept
|
* @param parent
|
*/
|
private void resetFullNameAndInnerId(SysDept dept, SysDept parent) {
|
if (parent == null) {
|
parent = get(dept.getPid());
|
}
|
if (parent != null) {// 存在上级
|
dept.setPids(parent.getPids() + "," + String.valueOf(dept.getId()));
|
} else {// 没有上级
|
dept.setPids(String.valueOf(dept.getId()));
|
}
|
resetChildFullNameAndInnerId(dept);
|
}
|
|
/**
|
* 修改子类全称和内部编码
|
*
|
* @param parent
|
*/
|
private void resetChildFullNameAndInnerId(SysDept parent) {
|
List<SysDept> list = getChildren(parent.getId());
|
for (SysDept dept : list) {
|
resetFullNameAndInnerId(dept, parent);
|
this.update(dept);
|
resetChildFullNameAndInnerId(dept);
|
}
|
}
|
|
public List<SysDept> fetchRepairer() {
|
List<SysDept> list = this.getTenantAll();
|
|
String factoryId = "cj".equals(UserContext.getUser().getUnitType()) ? UserContext.getUser().getCompanyId().toString() : "";
|
String regionFlag = UserContext.getUser().getArea();//getSessionValue(UserContext.REGION_FLAG);
|
|
/*
|
如果属于某厂家，就返回此厂家
|
*/
|
if (!StringUtils.isEmpty(factoryId)) {
|
List<SysDept> collect = list
|
.stream()
|
.filter(dept -> {
|
return dept.getId().equals(new Long(factoryId));
|
})
|
.collect(Collectors.toList());
|
return collect;
|
}
|
|
if (!StringUtils.isEmpty(regionFlag)) {
|
List<SysDept> collectRegion = list
|
.stream()
|
.filter(dept -> {
|
return dept.getCode().equals(regionFlag);
|
})
|
.collect(Collectors.toList());
|
|
if (!collectRegion.isEmpty()) {
|
final SysDept regionSysDept = collectRegion.get(0);
|
|
List<SysDept> collect = list
|
.stream()
|
.filter(dept -> {
|
String pids = dept.getPids();
|
String[] pidsArray = pids.split(",");
|
for (String pid : pidsArray) {
|
if (pid.equals(regionSysDept.getId().toString()) && new Integer(2).equals(dept.getNature())) {
|
return true;
|
}
|
}
|
|
return false;
|
})
|
.collect(Collectors.toList());
|
return collect;
|
}
|
}
|
|
List<SysDept> collect = list
|
.stream()
|
// .filter(dept -> (new Integer(1).equals(dept.getNature()) && dept.isCompany())
|
// || (new Integer(2).equals(dept.getNature()) && dept.isCompany() /*&& dept.getPid().toString().equals("1271368584383188993") */)
|
// || "HBDD".equals(dept.getCode()))
|
.filter(dept -> "cj".equals(dept.getNature()))
|
.collect(Collectors.toList());
|
return collect;
|
}
|
|
public Long fetchRepairerId(String factoryName) {
|
List<SysDept> list = this.fetchRepairer();
|
if (list != null && !list.isEmpty()) {
|
for (SysDept sysDept : list) {
|
if (sysDept.getName() != null && sysDept.getName().equals(factoryName)) {
|
return sysDept.getId();
|
}
|
}
|
}
|
return null;
|
}
|
|
public List<SysDept> getListByParent(String pid, String pCode, String code) {
|
HashMap<String, Object> map = new HashMap<>();
|
map.put("pCode", pCode);
|
map.put("code", code);
|
map.put("pId", pid);
|
List<SysDept> list = baseDao.getListByParent(map);
|
return list;
|
}
|
|
public Long getCompanyIdByDeptId(Long deptId) {
|
SysDept sysDept = baseDao.getById(deptId);
|
if (sysDept != null)
|
return sysDept.getCompanyId();
|
else
|
return deptId;
|
}
|
|
public Map<String, String> getMapFactory(Boolean idToName) {
|
List<SysDept> list = this.getTenantAll();
|
|
List<SysDept> collect = list
|
.stream()
|
.filter(dept -> ("cj".equals(dept.getNature()) ))
|
.collect(Collectors.toList());
|
|
Map<String, String> result = new HashMap<>();
|
for (SysDept item : collect) {
|
if (idToName)
|
result.put(item.getId().toString(), item.getName());
|
else
|
result.put(item.getName(), item.getId().toString());
|
}
|
return result;
|
}
|
|
public Map<String, String> getMapTeamFactory(Boolean idToName) {
|
List<SysDept> list = this.getTenantAll();
|
|
List<SysDept> collect = list
|
.stream()
|
.filter(dept -> ("cj".equals(dept.getNature()) ||"td".equals(dept.getNature())))
|
.collect(Collectors.toList());
|
|
Map<String, String> result = new HashMap<>();
|
for (SysDept item : collect) {
|
if (idToName)
|
result.put(item.getId().toString(), item.getName());
|
else
|
result.put(item.getName(), item.getId().toString());
|
}
|
return result;
|
}
|
}
|
