From 46d4a8583240cbe3de6f1b4a196dd5d06485398a Mon Sep 17 00:00:00 2001 From: skyouc Date: 星期四, 20 三月 2025 16:02:44 +0800 Subject: [PATCH] #新增 1. 添加库们逻辑分区 --- rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/CompanysController.java | 22 +++++++++++++++++++--- 1 files changed, 19 insertions(+), 3 deletions(-) diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/CompanysController.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/CompanysController.java index 0c5d1f2..9e69bb6 100644 --- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/CompanysController.java +++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/CompanysController.java @@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.vincent.rsf.framework.common.Cools; import com.vincent.rsf.framework.common.R; +import com.vincent.rsf.framework.exception.CoolException; import com.vincent.rsf.server.common.utils.ExcelUtil; import com.vincent.rsf.server.common.annotation.OperationLog; import com.vincent.rsf.server.common.domain.BaseParam; @@ -12,6 +13,7 @@ import com.vincent.rsf.server.manager.entity.Companys; import com.vincent.rsf.server.manager.service.CompanysService; import com.vincent.rsf.server.system.controller.BaseController; +import io.swagger.annotations.Api; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.web.bind.annotation.*; @@ -19,6 +21,7 @@ import javax.servlet.http.HttpServletResponse; import java.util.*; +@Api(tags = "寰�鏉ヤ紒涓�") @RestController public class CompanysController extends BaseController { @@ -55,10 +58,17 @@ @OperationLog("Create 瀛楀吀鏁版嵁闆�") @PostMapping("/companys/save") public R save(@RequestBody Companys companys) { + if (Objects.isNull(companys.getName())) { + throw new CoolException("浼佷笟鍚嶇О涓嶈兘涓虹┖锛侊紒"); + } + if (Objects.isNull(companys.getBreifCode())) { + throw new CoolException("浼佷笟鍔╄鐮佷笉鑳戒负绌�!!"); + } + if (Objects.isNull(companys.getType())) { + throw new CoolException("浼佷笟绫诲瀷涓嶈兘涓虹┖锛侊紒"); + } companys.setCreateBy(getLoginUserId()); - companys.setCreateTime(new Date()); companys.setUpdateBy(getLoginUserId()); - companys.setUpdateTime(new Date()); if (!companysService.save(companys)) { return R.error("Save Fail"); } @@ -104,7 +114,13 @@ @PreAuthorize("hasAuthority('manager:companys:list')") @PostMapping("/companys/export") public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception { - ExcelUtil.build(ExcelUtil.create(companysService.list(), Companys.class), response); + List<Companys> companies = new ArrayList<>(); + if (!Objects.isNull(map.get("ids"))) { + companies = companysService.list(new LambdaQueryWrapper<Companys>().in(Companys::getId, map.get("ids"))); + } else { + companies = companysService.list(); + } + ExcelUtil.build(ExcelUtil.create(companies, Companys.class), response); } } -- Gitblit v1.9.1