feat: 文档知识库 分片操作限流

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

 package cn.com.poc.knowledge.rest.impl;
 
+import cn.com.poc.common.annotation.RedisLimit;
 import cn.com.poc.knowledge.aggregate.QAKnowledgeService;
 import cn.com.poc.knowledge.convert.QAKnowledgeChunkConvert;
 import cn.com.poc.knowledge.dto.QAKnowledgeChunkDto;
@@ -26,21 +27,25 @@ public class QAKnowledgeRestImpl implements QAKnowledgeRest {
     private QAKnowledgeService qaKnowledgeService;
 
     @Override
+    @RedisLimit(key = "OPEN_KNOWLEDGE_CHUNK", timeout = 1, currentUser = true, count = 1)
     public void openKnowledgeChunk(UpsertChunkInfoDto dto) {
         qaKnowledgeService.openKnowledgeChunk(dto.getKdId(), dto.getChunkRelationId(), dto.getIsOpen());
     }
 
     @Override
+    @RedisLimit(key = "DELETE_KNOWLEDGE_CHUNK", timeout = 1, currentUser = true, count = 1)
     public void deleteKnowledgeChunk(UpsertChunkInfoDto dto) {
         qaKnowledgeService.deleteKnowledgeChunk(dto.getKdId(), dto.getChunkRelationId());
     }
 
     @Override
+    @RedisLimit(key = "UPDATE_KNOWLEDGE_CHUNK_DOC", timeout = 1, currentUser = true, count = 1)
     public void updateKnowledgeChunkDoc(UpsertChunkInfoDto dto) {
         qaKnowledgeService.updateKnowledgeChunkDoc(dto.getKdId(), dto.getChunkRelationId(), dto.getStructId(), dto.getChunkContent());
     }
 
     @Override
+    @RedisLimit(key = "BATCH_UPDATE_KNOWLEDGE_CHUNK_DOC", timeout = 1, currentUser = true, count = 1)
     public void batchUpdateKnowledgeChunkDoc(UpsertChunkInfoDto dto) {
         qaKnowledgeService.updateKnowledgeChunkDoc(dto.getKdId(), dto.getChunkRelationId(), dto.getChunkInfos());
     }
@@ -56,11 +61,13 @@ public class QAKnowledgeRestImpl implements QAKnowledgeRest {
     }
 
     @Override
+    @RedisLimit(key = "UPDATE_KNOWLEDGE_CHUNK", timeout = 1, currentUser = true, count = 1)
     public void updateKnowledgeChunk(Integer kdId, Long structId, String structName, String isIndex) {
         qaKnowledgeService.updateKnowledgeStruct(kdId, structId, structName, isIndex);
     }
 
     @Override
+    @RedisLimit(key = "OPEN_KNOWLEDGE_CHUNK", timeout = 1, currentUser = true, count = 1)
     public void batchDeleteKnowledgeChunks(Integer kdId, List<String> chunkRelationIds) {
         for (String chunkRelationId : chunkRelationIds) {
             qaKnowledgeService.deleteKnowledgeChunk(kdId, chunkRelationId);
@@ -68,6 +75,7 @@ public class QAKnowledgeRestImpl implements QAKnowledgeRest {
     }
 
     @Override
+    @RedisLimit(key = "OPEN_KNOWLEDGE_CHUNK", timeout = 1, currentUser = true, count = 1)
     public void updateChunkSort(Integer kdId, String chunkRelationId, Integer sort) {
         qaKnowledgeService.updateKnowledgeChunkSort(kdId, chunkRelationId, sort);
     }