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