From dc600db1211ba96302c2f75b4dd224512aad89cf Mon Sep 17 00:00:00 2001 From: puzhibing <393733352@qq.com> Date: 星期五, 19 一月 2024 15:32:09 +0800 Subject: [PATCH] 修改bug --- ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/management/MgtStaffController.java | 24 +++++++++++++++++++----- 1 files changed, 19 insertions(+), 5 deletions(-) diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/management/MgtStaffController.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/management/MgtStaffController.java index ad4926a..29cc75a 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/management/MgtStaffController.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/management/MgtStaffController.java @@ -3,6 +3,7 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.ruoyi.common.core.domain.R; import com.ruoyi.common.core.utils.poi.ExcelUtil; +import com.ruoyi.common.core.web.domain.AjaxResult; import com.ruoyi.common.log.annotation.Log; import com.ruoyi.common.log.enums.BusinessType; import com.ruoyi.common.security.utils.SecurityUtils; @@ -17,14 +18,14 @@ import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.RequestBody; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RequestMethod; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; +import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; import java.util.List; + +import static com.ruoyi.common.core.web.domain.AjaxResult.success; /** * @author jqs34 @@ -80,9 +81,12 @@ Long userId = SecurityUtils.getUserId(); mgtStaffSuggestPageDto.setUserId(userId); Page<MgtStaffSuggestPageVo> page = new Page<>(); - if(page.getSize()>5000){ + if(mgtStaffSuggestPageDto.getPageSize() > 5000){ page.setSize(5000); page.setCurrent(1); + }else{ + page.setSize(mgtStaffSuggestPageDto.getPageSize()); + page.setCurrent(mgtStaffSuggestPageDto.getPageNum()); } List<MgtStaffSuggestPageVo> staffSuggestPageVoList = staffSuggestService.pageMgtStaffSuggest(page,mgtStaffSuggestPageDto); ExcelUtil<MgtStaffSuggestPageVo> util = new ExcelUtil<MgtStaffSuggestPageVo>(MgtStaffSuggestPageVo.class); @@ -147,4 +151,14 @@ sysStaffService.mgtDeleteStaff(mgtBaseGetDto); return R.ok(); } + + @ApiOperation(value = "导入员工数据") + @PostMapping("/importSysStaff") + public AjaxResult importSysStaff(@RequestPart("file") MultipartFile file) throws Exception + { + ExcelUtil<MgtSysStaffImportDto> util = new ExcelUtil<MgtSysStaffImportDto>(MgtSysStaffImportDto.class); + List<MgtSysStaffImportDto> sysStaffImportDtoList = util.importExcel(file.getInputStream()); + String message = sysStaffService.importSysStaff(sysStaffImportDtoList); + return success(message); + } } -- Gitblit v1.7.1