fix:限制文件个数

src/main/java/cn/com/poc/knowledge/rest/impl/KnowledgeRestImpl.java

@@ -107,7 +107,15 @@ public class KnowledgeRestImpl implements KnowledgeRest {
         Assert.notNull(dto.getKnowledgeInfoId());
         Assert.notNull(dto.getKdIds());
         Assert.notNull(dto.getSegmentationConfig());
-        return knowledgeService.trainKnowledge(dto.getKnowledgeInfoId(), dto.getKdIds(), dto.getSegmentationConfig());
+        UserBaseEntity userBaseEntity = BlContext.getCurrentUser();
+        MemberEquityInfo equityInfo = memberEquityService.getEquityInfo(userBaseEntity.getUserId());
+        String lock = userBaseEntity.getUserId().toString().intern();
+        synchronized (lock) {
+            if ((dto.getKdIds().size() + equityInfo.getUsedKnowledgeNum()) > equityInfo.getKnowledgeNum()) {
+                throw new I18nMessageException("exception/create.num.limit");
+            }
+            return knowledgeService.trainKnowledge(dto.getKnowledgeInfoId(), dto.getKdIds(), dto.getSegmentationConfig());
+        }
     }
 
     @Override