射洪城投
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
mitao
3 天以前 f769d817f8ba2522147c73fa93ccacdc3aab0cb6 
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/CleanToiletSanitizationController.java


@@ -1,9 +1,47 @@


package com.ruoyi.web.controller.api;






import org.springframework.web.bind.annotation.RequestMapping;


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 cn.hutool.json.JSONObject;


import com.ruoyi.common.annotation.Log;


import com.ruoyi.common.basic.PageInfo;


import com.ruoyi.common.core.domain.R;


import com.ruoyi.common.core.exception.ServiceException;


import com.ruoyi.common.enums.BusinessType;


import com.ruoyi.common.utils.WebUtils;


import com.ruoyi.system.export.CleanToiletSanitizationExport;


import com.ruoyi.system.model.CleanSludge;


import com.ruoyi.system.model.CleanToiletSanitization;


import com.ruoyi.system.model.DrDisplacement;


import com.ruoyi.system.query.CleanSludgeListQuery;


import com.ruoyi.system.query.DrDisplacementListQuery;


import com.ruoyi.system.service.CleanSludgeService;


import com.ruoyi.system.service.CleanToiletSanitizationService;


import io.swagger.annotations.Api;


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 org.springframework.web.bind.annotation.RestController;


import javax.annotation.Resource;


import javax.servlet.ServletOutputStream;


import javax.servlet.http.HttpServletResponse;


import java.io.IOException;


import java.io.InputStream;


import java.net.URLEncoder;


import java.time.LocalDate;


import java.time.LocalDateTime;


import java.util.ArrayList;


import java.util.Arrays;


import java.util.List;




/**


 * <p>


@@ -14,7 +52,171 @@


 * @since 2025-10-17


 */


@RestController


@Api(tags = "厕所消杀记录")


@Slf4j


@RequestMapping("/clean-toilet-sanitization")


public class CleanToiletSanitizationController {


    @Resource


    private CleanToiletSanitizationService cleanToiletSanitizationService;




    @ApiOperation(value = "厕所消杀记录分页列表")


    @PostMapping(value = "/pageList")


    public R<PageInfo<CleanToiletSanitization>> pageList(@RequestBody CleanSludgeListQuery query) {


        PageInfo<CleanToiletSanitization> cleanToiletSanitizationPageInfo = cleanToiletSanitizationService.pageList(query);


        return R.ok(cleanToiletSanitizationPageInfo);


    }




    @Log(title = "厕所消杀记录导入", businessType = BusinessType.INSERT)


    @ApiOperation(value = "厕所消杀记录导入")


    @Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRES_NEW)


    @PostMapping("/import")


    public R<String> importCleaner(@RequestPart("file") MultipartFile file) {


        ImportParams params = new ImportParams();


        params.setHeadRows(1); //表头行数


        InputStream inputStream = null;


        List<CleanToiletSanitizationExport> locationExcelList;


        try {


            inputStream = file.getInputStream();


            locationExcelList = ExcelImportUtil.importExcel(inputStream,


                    CleanToiletSanitizationExport.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<CleanToiletSanitization> cleanToiletSanitizations = new ArrayList<>();


        for (CleanToiletSanitizationExport locationExcel : locationExcelList) {


            CleanToiletSanitization cleanToiletSanitization = new CleanToiletSanitization();


            String recordDate = locationExcel.getRecordDate();


            LocalDate localDate = LocalDate.parse(recordDate);


            cleanToiletSanitization.setRecordDate(localDate);


            cleanToiletSanitization.setIsCleaned(StringUtils.hasLength(locationExcel.getIsCleaned()) && locationExcel.getIsCleaned().equals("是"));


            cleanToiletSanitization.setIsDisinfected(StringUtils.hasLength(locationExcel.getIsDisinfected()) && locationExcel.getIsDisinfected().equals("是"));


            cleanToiletSanitization.setCleaner(locationExcel.getCleaner());


            cleanToiletSanitization.setInspectionTime(locationExcel.getInspectionTime());


            cleanToiletSanitization.setInspector(locationExcel.getInspector());


            cleanToiletSanitization.setInspectionSituation(locationExcel.getInspectionSituation());


            cleanToiletSanitizations.add(cleanToiletSanitization);


        }


        cleanToiletSanitizationService.saveBatch(cleanToiletSanitizations);


        return R.ok();


    }




    @ApiOperation(value = "厕所消杀记录导出")


    @Log(title = "厕所消杀记录导出", businessType = BusinessType.OTHER)


    @PostMapping(value = "/export")


    public void export(@RequestBody CleanSludgeListQuery query) {


        List<CleanToiletSanitization> list = cleanToiletSanitizationService.lambdaQuery()


                .like(StringUtils.hasLength(query.getRecordDate()), CleanToiletSanitization::getRecordDate, query.getRecordDate())


                .list();




        List<CleanToiletSanitizationExport> cleanToiletSanitizationExports = new ArrayList<>();




        int i = 1;


        for (CleanToiletSanitization cleanToiletSanitization : list) {


            CleanToiletSanitizationExport cleanToiletSanitizationExport = new CleanToiletSanitizationExport();


            cleanToiletSanitizationExport.setNumber(i);


            LocalDate recordDate = cleanToiletSanitization.getRecordDate();


            cleanToiletSanitizationExport.setRecordDate(recordDate.toString());


            cleanToiletSanitizationExport.setIsCleaned(cleanToiletSanitization.getIsCleaned() ? "是" : "否");


            cleanToiletSanitizationExport.setIsDisinfected(cleanToiletSanitization.getIsDisinfected() ? "是" : "否");


            cleanToiletSanitizationExport.setCleaner(cleanToiletSanitization.getCleaner());


            cleanToiletSanitizationExport.setInspectionTime(cleanToiletSanitization.getInspectionTime());


            cleanToiletSanitizationExport.setInspector(cleanToiletSanitization.getInspector());


            cleanToiletSanitizationExport.setInspectionSituation(cleanToiletSanitization.getInspectionSituation());


            cleanToiletSanitizationExports.add(cleanToiletSanitizationExport);


            i++;


        }


        //1.获取excel模板


        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), CleanToiletSanitizationExport.class, cleanToiletSanitizationExports);


        HttpServletResponse response = WebUtils.response();


        response.setContentType("application/vnd.ms-excel");


        response.setCharacterEncoding("utf-8");


        ServletOutputStream outputStream = null;


        try {


            String fileName = URLEncoder.encode("厕所消杀记录.xls", "utf-8");


            response.setHeader("Content-dispodition", "attachment;filename=" + fileName);


            outputStream = response.getOutputStream();


            workbook.write(outputStream);


        } catch (IOException e) {


            e.printStackTrace();


        } finally {


            try {


                outputStream.close();


            } catch (IOException e) {


                e.printStackTrace();


            }


        }


    }




    @ApiOperation(value = "厕所消杀记录导入模板下载")


    @GetMapping("/download")


    public void download() {


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


        CleanToiletSanitizationExport tLocationImportExcel = new CleanToiletSanitizationExport();


        locationImportExcels.add(tLocationImportExcel);


        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), CleanToiletSanitizationExport.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();


            }


        }


    }




    @ApiOperation(value = "添加")


    @Transactional


    @Log(title = "厕所消杀记录-添加", businessType = BusinessType.INSERT)


    @PostMapping(value = "/add")


    public R<Boolean> save(@RequestBody CleanToiletSanitization entity) {


        cleanToiletSanitizationService.save(entity);


        return R.ok();


    }




    @ApiOperation(value = "修改")


    @Transactional


    @Log(title = "厕所消杀记录-修改", businessType = BusinessType.UPDATE)


    @PostMapping(value = "/edit")


    public R<Boolean> edit(@RequestBody CleanToiletSanitization entity) {


        cleanToiletSanitizationService.updateById(entity);


        return R.ok();


    }




    @ApiOperation(value = "详情")


    @GetMapping(value = "/detail")


    public R<CleanToiletSanitization> detail(String id) {


        return R.ok(cleanToiletSanitizationService.getById(id));


    }




    @Log(title = "厕所消杀记录-删除", businessType = BusinessType.DELETE)


    @Transactional


    @ApiOperation(value = "厕所消杀记录-删除")


    @DeleteMapping(value = "/delete")


    public R delete(@RequestParam String ids) {


        cleanToiletSanitizationService.removeBatchByIds(Arrays.asList(ids.split(",")));


        return R.ok();


    }


}