From 3494e6a256eb950b520df5a8d9efbb3ba1eb9f37 Mon Sep 17 00:00:00 2001
From: yuance <182865460@qq.com>
Date: Mon, 11 May 2026 15:56:22 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E4=BB=8E=E5=AD=97=E5=85=B8?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E8=8E=B7=E5=8F=96=E7=9B=AE=E5=BD=95=E5=88=97?=
 =?UTF-8?q?=E8=A1=A8=EF=BC=88=E6=9A=82=E5=AD=98-=E6=9C=AC=E5=9C=B0?=
 =?UTF-8?q?=E6=97=A0=E6=B3=95=E6=B5=8B=E8=AF=95=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../oa/service/impl/OaCompanyServiceImpl.java | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/main/java/com/gxwebsoft/oa/service/impl/OaCompanyServiceImpl.java b/src/main/java/com/gxwebsoft/oa/service/impl/OaCompanyServiceImpl.java
index be4b937..efe647d 100644
--- a/src/main/java/com/gxwebsoft/oa/service/impl/OaCompanyServiceImpl.java
+++ b/src/main/java/com/gxwebsoft/oa/service/impl/OaCompanyServiceImpl.java
@@ -11,6 +11,11 @@ import cn.hutool.core.util.StrUtil;
 
 import com.gxwebsoft.oa.entity.OaCompany;
 import com.gxwebsoft.oa.param.OaCompanyParam;
+import com.gxwebsoft.common.system.entity.Dict;
+import com.gxwebsoft.common.system.entity.DictData;
+import com.gxwebsoft.common.system.param.DictDataParam;
+import com.gxwebsoft.common.system.service.DictService;
+import com.gxwebsoft.common.system.service.DictDataService;
 import com.gxwebsoft.ai.config.KnowledgeBaseConfig;
 import com.gxwebsoft.ai.entity.AiCloudDoc;
 import com.gxwebsoft.ai.entity.AiCloudFile;
@@ -53,6 +58,12 @@ public class OaCompanyServiceImpl extends ServiceImpl<OaCompanyMapper, OaCompany
 
     @Autowired
     private KnowledgeBaseClientFactory clientFactory;
+
+    @Autowired
+    private DictService dictService;
+
+    @Autowired
+    private DictDataService dictDataService;
 	
     @Override
     public PageResult<OaCompany> pageRel(OaCompanyParam param) {
@@ -117,6 +128,25 @@ public class OaCompanyServiceImpl extends ServiceImpl<OaCompanyMapper, OaCompany
             "13.财务数据", "14.以前年度审计"
         );
         
+        // 从字典数据获取目录列表
+        try {
+            // 根据dictCode获取dict
+            Dict dict = dictService.getOne(new LambdaQueryWrapper<Dict>().eq(Dict::getDictCode, "documentCategory"));
+            if (dict != null) {
+                // 根据dictId获取所有字典数据
+                DictDataParam param = new DictDataParam();
+                param.setDictId(dict.getDictId());
+                List<DictData> dictDataList = dictDataService.listRel(param);
+                // 提取dictDataCode作为目录名称
+                directoryNames = dictDataList.stream()
+                    .map(DictData::getDictDataCode)
+                    .collect(Collectors.toList());
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+            // 如果获取字典数据失败，使用默认列表
+        }
+        
         List<AiCloudDoc> directories = new ArrayList<>();
         for (int i = 0; i < directoryNames.size(); i++) {
             AiCloudDoc doc = new AiCloudDoc();