fix: 课题组重复去重

boyang
1 parent b19689c2
Showing 1 changed file with 25 additions and 5 deletions
src/pages/ResearchGroup/ResearchGroupAccess/components/AddModal.tsx
@@ -135,12 +135,34 @@ const AddModal = forwardRef&lt;AddModalRef, AddModalProps&gt;((props, ref) =&gt; {
         // 提取所有相同groupName的不同companyName
         const companySet = new Set<string>();
         const companyIdMap = new Map<string, string>();
+        const importItemsMap = new Map<string, any>(); // 存储createByName="import"的项
-        // 假设接口返回数据包含列表项，每项有companyName和id
+        // 首先遍历所有非import的项
         res.data.data.forEach((item: any) => {
-          if (item.groupName === groupName && item.companyName) {
+          if (
+            item.groupName === groupName &&
+            item.companyName &&
+            item.createByName !== 'import'
+          ) {
             companySet.add(item.companyName);
-            companyIdMap.set(item.companyName, item.id); // 保存id用于提交
+            companyIdMap.set(item.companyName, item.id);
+          } else if (
+            item.groupName === groupName &&
+            item.companyName &&
+            item.createByName === 'import'
+          ) {
+            // 如果是import项，则存储该companyName的第一个项
+            if (!importItemsMap.has(item.companyName)) {
+              importItemsMap.set(item.companyName, item);
+            }
+          }
+        });
+
+        // 如果有companyName只存在于import项中，加入第一个import项
+        importItemsMap.forEach((item, companyName) => {
+          if (!companySet.has(companyName)) {
+            companySet.add(companyName);
+            companyIdMap.set(companyName, item.id);
           }
         });
@@ -150,9 +172,7 @@ const AddModal = forwardRef&lt;AddModalRef, AddModalProps&gt;((props, ref) =&gt; {
           value: name,
           id: companyIdMap.get(name) || '',
         }));
-
         setCompanyOptions(companies);
-
         // 如果只有一个选项，自动选中
         if (companies.length === 1) {
           form.setFieldsValue({