package com.skua.modules.system.controller; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.skua.common.constant.CacheConstant; import com.skua.core.api.vo.LoginUser; import com.skua.core.api.vo.Result; import com.skua.core.api.vo.TransTreeModel; import com.skua.core.api.vo.TreeData; import com.skua.core.aspect.annotation.AutoLog; import com.skua.core.context.BaseContextHandler; import com.skua.core.query.QueryGenerator; import com.skua.core.util.ConvertUtils; import com.skua.core.util.JwtUtil; import com.skua.core.util.TreeUtils; import com.skua.modules.flow.utils.StringUtil; import com.skua.modules.system.entity.SysDepart; import com.skua.modules.system.entity.SysUser; import com.skua.modules.system.model.DepartIdModel; import com.skua.modules.system.model.SysDepartTreeModel; import com.skua.modules.system.service.ISysDepartService; import com.skua.modules.system.service.ISysUserService; import com.skua.modules.system.vo.DepartTree; import com.skua.modules.system.vo.SysDeptUserVO; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang.StringUtils; import org.apache.shiro.SecurityUtils; import org.jeecgframework.poi.excel.ExcelImportUtil; import org.jeecgframework.poi.excel.def.NormalExcelConstants; import org.jeecgframework.poi.excel.entity.ExportParams; import org.jeecgframework.poi.excel.entity.ImportParams; import org.jeecgframework.poi.excel.view.JeecgEntityExcelView; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.cache.annotation.CacheEvict; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import org.springframework.web.servlet.ModelAndView; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.IOException; import java.util.*; import java.util.stream.Collectors; /** * 部门表 前端控制器 */ @Api(tags = "系统机构管理") @RestController @RequestMapping("/sys/sysDepart") @Slf4j public class SysDepartController { @Autowired ISysUserService userService; @Autowired private ISysDepartService sysDepartService; /** * 查询数据 查出所有部门,并以树结构数据格式响应给前端 * * @return */ @RequestMapping(value = "/queryTreeList", method = RequestMethod.GET) public Result<List<SysDepartTreeModel>> queryTreeList(String departType) { Result<List<SysDepartTreeModel>> result = new Result<>(); try { List<SysDepartTreeModel> list = sysDepartService.queryTreeList(departType); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 部门列表 * * @return */ //@ApiOperation(value="部门列表", notes="部门列表-无分页") @RequestMapping(value = "/list", method = RequestMethod.GET) public Result<List<SysDepart>> queryList() { Result<List<SysDepart>> result = new Result<>(); try { List<SysDepart> list = sysDepartService.queryList(); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 添加新数据 添加用户新建的部门对象数据,并保存到数据库 * * @param sysDepart * @return */ @AutoLog(value = "添加部门信息") @RequestMapping(value = "/add", method = RequestMethod.POST) @CacheEvict(value = {CacheConstant.DEPART_INFO_CACHE, CacheConstant.DEPART_IDMODEL_CACHE}, allEntries = true) public Result<SysDepart> add(@RequestBody SysDepart sysDepart, HttpServletRequest request) { Result<SysDepart> result = new Result<SysDepart>(); String username = JwtUtil.getUserNameByToken(request); try { sysDepart.setCreateBy(username); sysDepartService.saveDepartData(sysDepart, username); //清除部门树内存 // FindsDepartsChildrenUtil.clearSysDepartTreeList(); // FindsDepartsChildrenUtil.clearDepartIdModel(); result.success("添加成功!"); } catch (Exception e) { log.error(e.getMessage(), e); result.error500("操作失败"); } return result; } /** * 编辑数据 编辑部门的部分数据,并保存到数据库 * * @param sysDepart * @return */ @AutoLog(value = "编辑部门信息") @RequestMapping(value = "/edit", method = RequestMethod.PUT) @CacheEvict(value = {CacheConstant.DEPART_INFO_CACHE, CacheConstant.DEPART_IDMODEL_CACHE}, allEntries = true) public Result<SysDepart> edit(@RequestBody SysDepart sysDepart, HttpServletRequest request) { String username = JwtUtil.getUserNameByToken(request); sysDepart.setUpdateBy(username); Result<SysDepart> result = new Result<SysDepart>(); SysDepart sysDepartEntity = sysDepartService.getById(sysDepart.getId()); if (sysDepartEntity == null) { result.error500("未找到对应实体"); } else { boolean ok = sysDepartService.updateDepartDataById(sysDepart, username); // TODO 返回false说明什么? if (ok) { //清除部门树内存 //FindsDepartsChildrenUtil.clearSysDepartTreeList(); //FindsDepartsChildrenUtil.clearDepartIdModel(); result.success("修改成功!"); } } return result; } /** * 通过id删除 * * @param id * @return */ @AutoLog(value = "删除部门信息") @RequestMapping(value = "/delete", method = RequestMethod.DELETE) @CacheEvict(value = {CacheConstant.DEPART_INFO_CACHE, CacheConstant.DEPART_IDMODEL_CACHE}, allEntries = true) public Result<SysDepart> delete(@RequestParam(name = "id", required = true) String id) { Result<SysDepart> result = new Result<SysDepart>(); SysDepart sysDepart = sysDepartService.getById(id); if (sysDepart == null) { result.error500("未找到对应实体"); } else { boolean ok = sysDepartService.delete(id); if (ok) { //清除部门树内存 //FindsDepartsChildrenUtil.clearSysDepartTreeList(); // FindsDepartsChildrenUtil.clearDepartIdModel(); result.success("删除成功!"); } } return result; } /** * 批量删除 根据前端请求的多个ID,对数据库执行删除相关部门数据的操作 * * @param ids * @return */ @AutoLog(value = "批量删除部门信息") @RequestMapping(value = "/deleteBatch", method = RequestMethod.DELETE) @CacheEvict(value = {CacheConstant.DEPART_INFO_CACHE, CacheConstant.DEPART_IDMODEL_CACHE}, allEntries = true) public Result<SysDepart> deleteBatch(@RequestParam(name = "ids", required = true) String ids) { Result<SysDepart> result = new Result<SysDepart>(); if (ids == null || "".equals(ids.trim())) { result.error500("参数不识别!"); } else { this.sysDepartService.deleteBatchWithChildren(Arrays.asList(ids.split(","))); result.success("删除成功!"); } return result; } /** * 查询数据 添加或编辑页面对该方法发起请求,以树结构形式加载所有部门的名称,方便用户的操作 * * @return */ @RequestMapping(value = "/queryIdTree", method = RequestMethod.GET) public Result<List<DepartIdModel>> queryIdTree() { Result<List<DepartIdModel>> result = new Result<>(); try { List<DepartIdModel> list = sysDepartService.queryDepartIdTreeList(); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } @RequestMapping(value = "/queryDepartTree", method = RequestMethod.GET) public Result<List<DepartIdModel>> queryDepartTree() { Result<List<DepartIdModel>> result = new Result<>(); try { List<DepartIdModel> list = sysDepartService.queryDepartTree(); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 获取用户所有的厂区列表 * * @return */ @RequestMapping(value = "/factoryList", method = RequestMethod.GET) public Result<List<SysDepart>> factoryList() { Result<List<SysDepart>> result = new Result<>(); try { List<SysDepart> list = getUserFactoryList2(BaseContextHandler.getUserId()); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 获取用户所有的分中心列表 * * @return */ @RequestMapping(value = "/centerAllList", method = RequestMethod.GET) public Result<List<SysDepart>> centerAllList() { Result<List<SysDepart>> result = new Result<>(); try { List<SysDepart> list = sysDepartService.getCenterAllList(); result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } private List<SysDepart> getUserFactoryList2(String userId) { List<SysDepart> list = new ArrayList<>(); list = sysDepartService.getUserFactoryList2(userId); if (list.size() > 0) { return list; } else { return new ArrayList<SysDepart>(); } } /** * <p> * 部门搜索功能方法,根据关键字模糊搜索相关部门 * </p> * * @param keyWord * @return */ @RequestMapping(value = "/searchBy", method = RequestMethod.GET) public Result<List<SysDepartTreeModel>> searchBy(@RequestParam(name = "keyWord", required = true) String keyWord) { Result<List<SysDepartTreeModel>> result = new Result<List<SysDepartTreeModel>>(); try { List<SysDepartTreeModel> treeList = this.sysDepartService.searhBy(keyWord); if (treeList.size() == 0 || treeList == null) { throw new Exception(); } result.setSuccess(true); result.setResult(treeList); return result; } catch (Exception e) { e.fillInStackTrace(); result.setSuccess(false); result.setMessage("查询失败或没有您想要的任何数据!"); return result; } } /** * 导出excel * * @param request * @param sysDepart */ @RequestMapping(value = "/exportXls") public ModelAndView exportXls(SysDepart sysDepart, HttpServletRequest request) { // Step.1 组装查询条件 QueryWrapper<SysDepart> queryWrapper = QueryGenerator.initQueryWrapper(sysDepart, request.getParameterMap()); //Step.2 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); List<SysDepart> pageList = sysDepartService.list(queryWrapper); //按字典排序 Collections.sort(pageList, new Comparator<SysDepart>() { @Override public int compare(SysDepart arg0, SysDepart arg1) { return arg0.getOrgCode().compareTo(arg1.getOrgCode()); } }); //导出文件名称 mv.addObject(NormalExcelConstants.FILE_NAME, "部门列表"); mv.addObject(NormalExcelConstants.CLASS, SysDepart.class); LoginUser user = (LoginUser) SecurityUtils.getSubject().getPrincipal(); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("部门列表数据", "导出人:" + user.getRealname(), "导出信息")); mv.addObject(NormalExcelConstants.DATA_LIST, pageList); return mv; } /** * 通过excel导入数据 * * @param request * @param response * @return */ @RequestMapping(value = "/importExcel", method = RequestMethod.POST) public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; Map<String, MultipartFile> fileMap = multipartRequest.getFileMap(); for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) { MultipartFile file = entity.getValue();// 获取上传文件对象 ImportParams params = new ImportParams(); params.setTitleRows(2); params.setHeadRows(1); params.setNeedSave(true); try { // orgCode编码长度 int codeLength = 3; List<SysDepart> listSysDeparts = ExcelImportUtil.importExcel(file.getInputStream(), SysDepart.class, params); //按长度排序 Collections.sort(listSysDeparts, new Comparator<SysDepart>() { @Override public int compare(SysDepart arg0, SysDepart arg1) { return arg0.getOrgCode().length() - arg1.getOrgCode().length(); } }); for (SysDepart sysDepart : listSysDeparts) { String orgCode = sysDepart.getOrgCode(); if (orgCode.length() > codeLength) { String parentCode = orgCode.substring(0, orgCode.length() - codeLength); QueryWrapper<SysDepart> queryWrapper = new QueryWrapper<SysDepart>(); queryWrapper.eq("org_code", parentCode); try { SysDepart parentDept = sysDepartService.getOne(queryWrapper); if (!parentDept.equals(null)) { sysDepart.setParentId(parentDept.getId()); } else { sysDepart.setParentId(""); } } catch (Exception e) { //没有查找到parentDept } } else { sysDepart.setParentId(""); } sysDepartService.save(sysDepart); } return Result.ok("文件导入成功!数据行数:" + listSysDeparts.size()); } catch (Exception e) { log.error(e.getMessage(), e); return Result.error("文件导入失败:" + e.getMessage()); } finally { try { file.getInputStream().close(); } catch (IOException e) { e.printStackTrace(); } } } return Result.error("文件导入失败!"); } /** * 获取部门人员机构树 * * @return */ @RequestMapping(value = "/queryDeptAndEmpTreeList", method = RequestMethod.GET) public Result<List<TreeData>> queryDeptAndEmpTreeList(String departId) { //根据token获取该用户厂id departId = BaseContextHandler.getRealDepartId(); String orgType = ConvertUtils.getString(BaseContextHandler.get("orgType")); if (!StringUtils.isEmpty(orgType)) { departId = ConvertUtils.getString(BaseContextHandler.getRealDepartId()); } //获取数据列表 List<SysDeptUserVO> list = sysDepartService.queryDeptAndEmpTreeList(departId); //将业务数据转换为构造树所需的数据结构 List<TransTreeModel> transTreeModels = list.stream().map(bo -> convertToTreeModel(bo)).collect(Collectors.toList()); List<TreeData> treeList = TreeUtils.buildTree(transTreeModels, departId); Result<List<TreeData>> result = new Result<>(); try { result.setResult(treeList); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 获取部门人员机构树 * * @return */ @RequestMapping(value = "/queryDeptAndUserTreeList", method = RequestMethod.GET) public Result<List<TreeData>> queryDeptAndUserTreeList(String departId,String userType) { //获取数据列表 List<SysDeptUserVO> list = sysDepartService.queryDeptAndUserTreeList(departId,userType); //将业务数据转换为构造树所需的数据结构 List<TransTreeModel> transTreeModels = list.stream().map(bo -> convertToTreeModel(bo)).collect(Collectors.toList()); List<TreeData> treeList = TreeUtils.buildTree(transTreeModels, departId); Result<List<TreeData>> result = new Result<>(); try { result.setResult(treeList); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 获取部门人员机构树-无数据权限 * * @return */ @RequestMapping(value = "/queryAllDeptAndEmpTreeList", method = RequestMethod.GET) public Result<List<TreeData>> queryAllDeptAndEmpTreeList(String departId) { //根据token获取该用户厂id // departId = BaseContextHandler.getDeparts(); // String orgType = ConvertUtils.getString(BaseContextHandler.get("orgType")); // if(!StringUtils.isEmpty(orgType)) { // departId = ConvertUtils.getString(BaseContextHandler.get("realDepartId")); // } departId = "c6d7cb4deeac411cb3384b1b31278596"; //获取数据列表 List<SysDeptUserVO> list = sysDepartService.queryDeptAndEmpTreeList(departId); //将业务数据转换为构造树所需的数据结构 List<TransTreeModel> transTreeModels = list.stream().map(bo -> convertToTreeModel(bo)).collect(Collectors.toList()); List<TreeData> treeList = TreeUtils.buildTree(transTreeModels, departId); Result<List<TreeData>> result = new Result<>(); try { result.setResult(treeList); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 根据厂ID获取部门人员机构树 * * @param departId * @return */ @GetMapping(value = "/queryDeptUserTreeList") public Result<List<TreeData>> queryDeptUserTreeList(String departId) { List<SysDeptUserVO> list = sysDepartService.queryDeptUserTreeList(departId); //将业务数据转换为构造树所需的数据结构 List<TransTreeModel> transTreeModels = list.stream().map(bo -> convertToTreeModel(bo)).collect(Collectors.toList()); List<TreeData> treeList = TreeUtils.buildTree(transTreeModels, ""); Result<List<TreeData>> result = new Result<>(); try { result.setResult(treeList); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 获取厂区树 * * @return */ @RequestMapping(value = "/queryFactoryTreeList", method = RequestMethod.GET) public Result<List<TreeData>> queryFactoryTreeList() { //获取数据列表 List<SysDeptUserVO> list = sysDepartService.queryFactoryTreeList(); //将业务数据转换为构造树所需的数据结构 List<TransTreeModel> transTreeModels = list.stream().map(bo -> convertToTreeModel(bo)).collect(Collectors.toList()); List<TreeData> treeList = TreeUtils.buildTree(transTreeModels, ""); Result<List<TreeData>> result = new Result<>(); try { result.setResult(treeList); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 将SysDeptUserVO对象数据填充到TransTreeModel中 * * @param sysDeptUserVO * @return */ private TransTreeModel convertToTreeModel(SysDeptUserVO sysDeptUserVO) { TransTreeModel transTreeModel = new TransTreeModel(); transTreeModel.setId(sysDeptUserVO.getId()); transTreeModel.setPid(sysDeptUserVO.getParentId()); transTreeModel.setTitle(sysDeptUserVO.getTitle()); Map<String, String> map = new HashMap<String, String>(); map.put("nodeType", sysDeptUserVO.getNodeType()); transTreeModel.setMap(map); return transTreeModel; } /** * 获取片区列表 * * @return */ @ApiOperation(value = "片区列表", notes = "片区列表") @GetMapping(value = "/getAreaList") public Result<List<SysDepart>> getAreaList() { Result<List<SysDepart>> result = new Result<List<SysDepart>>(); String areaIds = ""; //片区数据权限 Object orgType = BaseContextHandler.get("orgType"); if (!ConvertUtils.isEmpty(orgType)) { if (orgType.toString().equals("1")) { //集团用户 } else if (orgType.toString().equals("2")) { //片区用户 areaIds = "'" + BaseContextHandler.get("realDepartId").toString() + "'"; } } else { //根据厂站id获取片区id String departId = BaseContextHandler.get("realDepartId").toString(); areaIds = "'" + sysDepartService.getAreaIdByDepartId(departId) + "'"; } List<SysDepart> list = sysDepartService.getAreaList(areaIds); result.setSuccess(true); result.setResult(list); return result; } @ApiOperation(value = "实时数据片区列表", notes = "实时数据片区列表") @GetMapping(value = "/getAreaListForMonitor") public Result<List<SysDepart>> getAreaListForMonitor() { Result<List<SysDepart>> result = new Result<List<SysDepart>>(); String areaIds = ""; //片区数据权限 Object orgType = BaseContextHandler.get("orgType"); if (!ConvertUtils.isEmpty(orgType)) { if (orgType.toString().equals("1")) { //集团用户 } else if (orgType.toString().equals("2")) { //片区用户 areaIds = "'" + BaseContextHandler.get("realDepartId").toString() + "'"; } } else { //根据厂站id获取片区id String departId = BaseContextHandler.get("realDepartId").toString(); areaIds = "'" + sysDepartService.getAreaIdByDepartId(departId) + "'"; } List<SysDepart> list = sysDepartService.getAreaListForMonitor(areaIds); result.setSuccess(true); result.setResult(list); return result; } /** * 根据片区ID获取水厂数据 * * @param areaId * @return */ @ApiOperation(value = "根据片区id获取水厂列表", notes = "根据片区id获取水厂列表") @GetMapping(value = "/queryDepartListByAreaId") public Result<List<SysDepart>> queryDepartListByAreaId(String areaId, @RequestParam(required = false) String departType) { List<SysDepart> list = sysDepartService.queryDepartListByAreaId(areaId); Result<List<SysDepart>> result = new Result<>(); try { if (StringUtil.isNotBlank(departType)) { list = list.stream().filter(depart -> Arrays.asList(departType.split(",")).contains(depart.getDepartType())).collect(Collectors.toList()); } result.setResult(list); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } /** * 获取厂区列表 * * @return */ @ApiOperation(value = "厂区列表", notes = "厂区列表") @GetMapping(value = "getFactoryList") public Result<List<SysDepart>> getFactoryList() { Result<List<SysDepart>> result = new Result<List<SysDepart>>(); List<SysDepart> list = sysDepartService.getFactoryList(); result.setSuccess(true); result.setResult(list); return result; } /** * 获取集团列表 * * @return */ @ApiOperation(value = "集团列表", notes = "集团列表") @GetMapping(value = "getJtList") public Result<List<SysDepart>> getJtList() { Result<List<SysDepart>> result = new Result<List<SysDepart>>(); List<SysDepart> list = sysDepartService.getJtList(); result.setSuccess(true); result.setResult(list); return result; } /** * 获取APP端厂区列表 * * @return */ @GetMapping(value = "getAppFactoryList") public Result<List<SysDepart>> getAppFactoryList() { Result<List<SysDepart>> result = new Result<List<SysDepart>>(); List<SysDepart> list = sysDepartService.getFactoryList(); result.setSuccess(true); result.setResult(list); return result; } /** * 获取公司列表 * * @return */ @GetMapping(value = "getCmpyList") public Result<List<SysDepart>> getCmpyList() { Result<List<SysDepart>> result = new Result<List<SysDepart>>(); List<SysDepart> list = sysDepartService.getCmpyList(); result.setSuccess(true); result.setResult(list); return result; } /** * 根据departType得到厂站ID * * @param departType * @return */ @GetMapping(value = "getDepartIds") public Result<String> getDepartIds(String departType) { Result<String> result = new Result<>(); QueryWrapper<SysDepart> sysDepartQueryWrapper = new QueryWrapper<>(); sysDepartQueryWrapper.eq("depart_type", departType); sysDepartQueryWrapper.eq("del_flag", "1"); sysDepartQueryWrapper.select("id"); List<SysDepart> sysDepartList = sysDepartService.list(sysDepartQueryWrapper); StringBuilder factorys = new StringBuilder(); sysDepartList.forEach(sysDepart -> { factorys.append(","); factorys.append(sysDepart.getId()); }); result.setSuccess(true); result.setResult(ConvertUtils.getString(factorys).replaceFirst(",", "")); return result; } /** * 获取所有用户(工作流接口) * @return */ @RequestMapping(value = "/getAllUserForFlow", method = RequestMethod.GET) public Result<List<Map>> getAllUserForFlow() { QueryWrapper<SysUser> queryWrapper = new QueryWrapper<>(); queryWrapper.eq("del_flag", "0"); queryWrapper.orderByDesc("create_time"); List<SysUser> userList = userService.list(queryWrapper); //构造数据: Result<List<Map>> :[{"itemName":"","itemCode":""}] List<Map> data = new LinkedList<>(); for (SysUser sysUser : userList) { Map map = new HashMap(); map.put("itemName", sysUser.getRealname()); map.put("itemCode", sysUser.getId()); data.add(map); } Result<List<Map>> result = new Result<>(); try { result.setResult(data); result.setSuccess(true); } catch (Exception e) { log.error(e.getMessage(), e); } return result; } @ApiOperation("积木报表机构树") @GetMapping(value = "/jimu/tree") public List<DepartTree> jimuDepartTree(DepartTree departTree) { List<DepartTree> list = sysDepartService.getJimuDepartTree(departTree.getPid()); return list; } }