From b8a4c06e6893d1d4fb3fc36e9d433f82309684ac Mon Sep 17 00:00:00 2001
From: yuance <182865460@qq.com>
Date: Thu, 9 Oct 2025 10:58:15 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=81=E4=B8=9A=E4=BF=A1=E6=81=AF=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=E8=B5=84=E6=96=99=E5=BA=93=E5=BA=93IDs=E5=AD=97?=
 =?UTF-8?q?=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/gxwebsoft/oa/controller/OaCompanyController.java | 4 ++++
 src/main/java/com/gxwebsoft/oa/entity/OaCompany.java          | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/src/main/java/com/gxwebsoft/oa/controller/OaCompanyController.java b/src/main/java/com/gxwebsoft/oa/controller/OaCompanyController.java
index 865e778..3a7aaad 100644
--- a/src/main/java/com/gxwebsoft/oa/controller/OaCompanyController.java
+++ b/src/main/java/com/gxwebsoft/oa/controller/OaCompanyController.java
@@ -11,6 +11,7 @@ import com.gxwebsoft.common.core.web.ApiResult;
 import com.gxwebsoft.common.core.web.PageResult;
 import com.gxwebsoft.common.core.web.PageParam;
 import com.gxwebsoft.common.core.web.BatchParam;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.gxwebsoft.ai.service.KnowledgeBaseService;
 import com.gxwebsoft.common.core.annotation.OperationLog;
 import io.swagger.v3.oas.annotations.tags.Tag;
@@ -72,6 +73,9 @@ public class OaCompanyController extends BaseController {
     	if(StrUtil.isEmpty(oaCompany.getCompanyCode())) {
     		return fail("单位唯一标识不能为空");
     	}
+    	if(oaCompanyService.count(new LambdaQueryWrapper<OaCompany>().eq(OaCompany::getCompanyCode, oaCompany.getCompanyCode()))>0) {
+    		return fail("单位唯一标识已存在");
+    	}
         if (oaCompanyService.save(oaCompany)) {
         	try {
         		//查询知识库
diff --git a/src/main/java/com/gxwebsoft/oa/entity/OaCompany.java b/src/main/java/com/gxwebsoft/oa/entity/OaCompany.java
index b9cf4f3..699a220 100644
--- a/src/main/java/com/gxwebsoft/oa/entity/OaCompany.java
+++ b/src/main/java/com/gxwebsoft/oa/entity/OaCompany.java
@@ -189,6 +189,9 @@ public class OaCompany implements Serializable {
     @Schema(description = "知识库ID")
     private String kbId;
 
+    @Schema(description = "资料库库IDs")
+    private String libraryIds; //英文逗号分割
+
     @Schema(description = "创建时间")
     @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
     private LocalDateTime createTime;