From 9364cf6824389a3264dc2cdb2bc3b5857ce76ad5 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期四, 03 七月 2025 23:38:54 +0800
Subject: [PATCH] bug修改
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java | 198 +++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 191 insertions(+), 7 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java
index fcdcfb1..5c0dab0 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java
@@ -1,29 +1,47 @@
package com.ruoyi.web.controller.api;
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.ExcelImportUtil;
+import cn.afterturn.easypoi.excel.entity.ExportParams;
+import cn.afterturn.easypoi.excel.entity.ImportParams;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.core.exception.ServiceException;
import com.ruoyi.common.enums.BusinessType;
+import com.ruoyi.common.utils.WebUtils;
import com.ruoyi.common.utils.bean.BeanUtils;
+import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.LocationDTO;
+import com.ruoyi.system.importExcel.TLocationImportExcel;
import com.ruoyi.system.model.*;
import com.ruoyi.system.query.LocationListQuery;
import com.ruoyi.system.query.LocationListTaskQuery;
import com.ruoyi.system.service.*;
-import com.ruoyi.system.vo.system.LocationListTaskVO;
-import com.ruoyi.system.vo.system.LocationListVO;
+import com.ruoyi.system.utils.CustomerImportFailedData;
+import com.ruoyi.system.vo.system.*;
+import com.sun.org.apache.bcel.internal.generic.NEW;
import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiOperation;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.springframework.transaction.annotation.Propagation;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Objects;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URLEncoder;
+import java.util.*;
import java.util.stream.Collectors;
/**
@@ -34,6 +52,7 @@
* @author xiaochen
* @since 2025-05-28
*/
+@Slf4j
@Api(tags = "点位管理")
@RestController
@RequestMapping("/t-location")
@@ -54,6 +73,8 @@
private TDeptService deptService;
@Resource
private ISysUserService sysUserService;
+ @Resource
+ private TokenService tokenService;
@ApiOperation(value = "点位管理分页列表")
@PostMapping(value = "/pageList")
@@ -157,12 +178,70 @@
res.add(temp);
}
pageInfo.setRecords(res);
- return R.ok();
+ return R.ok(pageInfo);
}
@ApiOperation(value = "根据片区id查询保洁员列表 不分页")
@GetMapping(value = "/listCleaner")
public R<List<TCleaner>> listCleaner(@RequestParam String id) {
return R.ok(cleanerService.lambdaQuery().eq(TCleaner::getProjectId,id).list());
+ }
+ @ApiOperation(value = "根据保洁员id查询点位列表 不分页")
+ @GetMapping(value = "/listLocation")
+ public R<List<TLocation>> listLocation(@RequestParam String id) {
+ return R.ok(locationService.lambdaQuery().eq(TLocation::getLocationCleaner,id).list());
+ }
+
+ @ApiOperation(value = "查询部门 项目部列表 不分页")
+ @GetMapping(value = "/listDepts")
+ public R<List<DeptNoLimitVO>> listDepts() {
+ List<TProjectDept> parent = projectDeptService.lambdaQuery().eq(TProjectDept::getParentId, "0").list();
+ List<TProjectDept> child = projectDeptService.lambdaQuery().ne(TProjectDept::getParentId, "0").list();
+ List<TDept> depts = deptService.list();
+ List<DeptNoLimitVO> res = new ArrayList<>();
+ DeptNoLimitVO deptNoLimitVO1 = new DeptNoLimitVO();
+ deptNoLimitVO1.setDeptName("项目部");
+ DeptNoLimitVO deptNoLimitVO2 = new DeptNoLimitVO();
+ deptNoLimitVO2.setDeptName("公司");
+ res.add(deptNoLimitVO1);
+ res.add(deptNoLimitVO2);
+ for (DeptNoLimitVO re : res) {
+ if (re.getDeptName().equals("项目部")){
+ List<DeptNoLimitParentVO> deptNoLimitVOS = new ArrayList<>();
+ for (TProjectDept tProjectDept : parent) {
+ DeptNoLimitParentVO deptNoLimitVO = new DeptNoLimitParentVO();
+ deptNoLimitVO.setDeptName(tProjectDept.getProjectName());
+ deptNoLimitVO.setDeptId(tProjectDept.getId());
+ List<TProjectDept> collect = child.stream().filter(e -> e.getParentId().equals(tProjectDept.getId())).collect(Collectors.toList());
+ List<DeptNoLimitChildVO> deptNoLimitChildVOS = new ArrayList<>();
+ for (TProjectDept projectDept : collect) {
+ DeptNoLimitChildVO deptNoLimitChildVO = new DeptNoLimitChildVO();
+ deptNoLimitChildVO.setDeptName(projectDept.getProjectName());
+ deptNoLimitChildVO.setDeptId(projectDept.getId());
+ deptNoLimitChildVOS.add(deptNoLimitChildVO);
+ }
+ deptNoLimitVO.setDeptChild(deptNoLimitChildVOS);
+ deptNoLimitVOS.add(deptNoLimitVO);
+ }
+ re.setDeptChild(deptNoLimitVOS);
+ }else {
+ List<DeptNoLimitParentVO> deptNoLimitVOS = new ArrayList<>();
+ for (TDept tDept : depts) {
+ DeptNoLimitParentVO deptNoLimitVO = new DeptNoLimitParentVO();
+ deptNoLimitVO.setDeptName(tDept.getDeptName());
+ deptNoLimitVO.setDeptId(tDept.getId());
+ deptNoLimitVOS.add(deptNoLimitVO);
+ }
+ re.setDeptChild(deptNoLimitVOS);
+ }
+
+ }
+ return R.ok(res);
+ }
+ @ApiOperation(value = "根据部门/项目部id查询巡检员列表 不分页")
+ @GetMapping(value = "/listUsers")
+ public R<List<SysUser>> listUsers(@RequestParam String id) {
+ List<SysUser> res = sysUserService.selectUserByDeptId(id);
+ return R.ok(res);
}
@Log(title = "新增点位", businessType = BusinessType.INSERT)
@ApiOperation(value = "新增点位")
@@ -191,5 +270,110 @@
locationService.removeBatchByIds(Arrays.asList(split));
return R.ok();
}
+
+ @ApiOperation(value = "点位导入模板下载")
+ @GetMapping("/import-template")
+ public void importTemplate() {
+ List<TLocationImportExcel> locationImportExcels = new ArrayList<>();
+ TLocationImportExcel tLocationImportExcel = new TLocationImportExcel();
+ locationImportExcels.add(tLocationImportExcel);
+ Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TLocationImportExcel.class, locationImportExcels);
+ HttpServletResponse response = WebUtils.response();
+ ServletOutputStream outputStream = null;
+ try {
+ String fileName = URLEncoder.encode("点位导入模板.xls", "utf-8");
+ response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
+ response.setHeader("content-Type", "application/vnd.ms-excel");
+ response.setHeader("Pragma", "no-cache");
+ response.setHeader("Cache-Control", "no-cache");
+ outputStream = response.getOutputStream();
+ workbook.write(outputStream);
+ } catch (IOException e) {
+ e.printStackTrace();
+ System.out.println("点位导入模板下载失败!");
+ } finally {
+ try {
+ outputStream.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+ }
+
+ @Log(title = "点位导入", businessType = BusinessType.INSERT)
+ @ApiOperation(value = "点位导入")
+ @Transactional(rollbackFor = Exception.class,propagation = Propagation.REQUIRES_NEW)
+ @PostMapping("/importLocation")
+ public R<String> importLocation(@RequestPart("file") MultipartFile file) {
+ ImportParams params = new ImportParams();
+ params.setTitleRows(2); // 标题行数
+ params.setHeadRows(1); //表头行数
+ InputStream inputStream = null;
+// List<CustomerImportFailedData> failedData = new ArrayList<>();
+ List<TLocationImportExcel> locationExcelList;
+ try {
+ inputStream = file.getInputStream();
+ locationExcelList = ExcelImportUtil.importExcel(inputStream,
+ TLocationImportExcel.class, params);
+ } catch (Exception e) {
+ e.printStackTrace();
+ log.error("点位导入失败:{}", e.getMessage());
+ throw new ServiceException("点位导入失败!");
+ } finally {
+ try {
+ inputStream.close();
+ } catch (IOException e) {
+ throw new ServiceException(e.getMessage());
+ }
+ }
+ if (CollectionUtils.isEmpty(locationExcelList)) {
+ throw new ServiceException("点位数据为空!");
+ }
+
+// List<TLocation> saveOrUpdateList = new ArrayList<>();
+ // 查询所有类型点位
+ List<TLocationType> locationTypes = locationTypeService.list();
+ // 查询所有用户
+ List<SysUser> users = sysUserService.selectAllList();
+ // 查询所有部门
+ List<TDept> deptList = deptService.list();
+ // 查询所有保洁员
+ List<TCleaner> cleaners = cleanerService.list();
+
+ for (TLocationImportExcel locationExcel : locationExcelList) {
+ System.err.println(locationExcel);
+ TLocation location = new TLocation();
+ location.setLocationCode(locationExcel.getLocationCode());
+ location.setLocationName(locationExcel.getLocationName());
+ TLocationType tLocationType = locationTypes.stream().filter(locationType -> locationType.getLocationName().equals(locationExcel.getLocationType())).findFirst().orElse(null);
+ if (Objects.nonNull(tLocationType)){
+ location.setLocationType(tLocationType.getId());
+ }
+ location.setLocationAddress(locationExcel.getLocationAddress());
+ String[] addressLonLat = locationExcel.getLocationAddressLonLat().split("/");
+ location.setLocationLon(addressLonLat[0]);
+ location.setLocationLat(addressLonLat[1]);
+ location.setLocationAddressEnd(locationExcel.getLocationAddressEnd());
+ String[] addressEndLonLat = locationExcel.getLocationAddressEndLonLat().split("/");
+ location.setLocationLonEnd(addressEndLonLat[0]);
+ location.setLocationLatEnd(addressEndLonLat[1]);
+ TDept tDept = deptList.stream().filter(dept -> dept.getCode().equals(locationExcel.getDeptCode())).findFirst().orElse(null);
+ if(Objects.nonNull(tDept)){
+ location.setProjectId(tDept.getId());
+ }
+ SysUser sysUser = users.stream().filter(user -> user.getNickName().equals(locationExcel.getNickName())).findFirst().orElse(null);
+ if(Objects.nonNull(sysUser)){
+ location.setLocationLeader(String.valueOf(sysUser.getUserId()));
+ }
+ cleaners.stream().filter(cleaner -> cleaner.getCleanerCode().equals(locationExcel.getCleanerCodeClear())
+ && cleaner.getDeptCode().equals(locationExcel.getDeptCodeClear())
+ && cleaner.getProjectCode().equals(locationExcel.getProjectCodeClear())).findFirst().orElse(null);
+ location.setLocationCleaner(locationExcel.getProjectCodeClear());
+ locationService.save(location);
+ }
+
+ return R.ok();
+ }
+
}
--
Gitblit v1.7.1