保洁巡检
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
无关风月
2025-07-03 6a9685e29ff409516a1ad078eaacadaf73843f9b 
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java


@@ -1,32 +1,47 @@


package com.ruoyi.web.controller.api;






import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


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.dto.LocationTypeDTO;


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.query.LocationTypeListQuery;


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;




/**


@@ -37,6 +52,7 @@


 * @author xiaochen


 * @since 2025-05-28


 */


@Slf4j


@Api(tags = "点位管理")


@RestController


@RequestMapping("/t-location")


@@ -50,13 +66,15 @@


    @Resource


    private TCleanerService cleanerService;


    @Resource


    private TTaskService tTaskService;


    private TTaskCleanService tTaskCleanService;


    @Resource


    private TTaskDetailService tTaskDetailService;


    @Resource


    private TDeptService deptService;


    @Resource


    private ISysUserService sysUserService;


    @Resource


    private TokenService tokenService;




    @ApiOperation(value = "点位管理分页列表")


    @PostMapping(value = "/pageList")


@@ -109,7 +127,7 @@


            }


        }


        PageInfo<LocationListTaskVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());


        List<TTask> list = tTaskService.lambdaQuery()


        List<TTask> list = tTaskCleanService.lambdaQuery()


                .eq(TTask::getLocationId, query.getId())


                .between(StringUtils.hasLength(query.getStartTime()), TTask::getImplementTime, query.getStartTime(), query.getEndTime())


                .eq(query.getStatus() != null, TTask::getStatus, query.getStatus())


@@ -167,6 +185,64 @@


    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 = "新增点位")


    @PostMapping(value = "/add")


@@ -194,5 +270,108 @@


        locationService.removeBatchByIds(Arrays.asList(split));


        return R.ok();


    }




    @ApiOperation(value = "点位导入模板下载")


    @GetMapping("/import-template")


    public void importTemplate() {


        List<TLocationImportExcel> locationImportExcels = new ArrayList<>();


        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();


    }




}