opensource/mateclaw
2
0
Fork 0
mirror of https://gitee.com/mateos/mateclaw.git synced 2026-09-14 19:45:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(wiki): show live page count in knowledge base list and detail

Browse Source
This commit is contained in:
matevip 2026-05-17 17:55:29 +08:00
parent 401c413a77
commit 84d809f52b
1 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
mateclaw-server/src/main/java/vip/mate/wiki/controller/WikiController.java
View File

@ -62,7 +62,7 @@ public class WikiController {
public R<List<WikiKnowledgeBaseEntity>> listKBs( public R<List<WikiKnowledgeBaseEntity>> listKBs(
@RequestHeader(value = "X-Workspace-Id", required = false) Long workspaceId) { @RequestHeader(value = "X-Workspace-Id", required = false) Long workspaceId) {
long wsId = workspaceId != null ? workspaceId : 1L; long wsId = workspaceId != null ? workspaceId : 1L;
return R.ok(kbService.listByWorkspace(wsId)); return R.ok(withLivePageCount(kbService.listByWorkspace(wsId)));
} }
@RequireWorkspaceRole("viewer") @RequireWorkspaceRole("viewer")
@ -73,7 +73,26 @@ public class WikiController {
verifyKBWorkspace(id, workspaceId); verifyKBWorkspace(id, workspaceId);
WikiKnowledgeBaseEntity kb = kbService.getById(id); WikiKnowledgeBaseEntity kb = kbService.getById(id);
if (kb == null) return R.fail("Knowledge base not found"); if (kb == null) return R.fail("Knowledge base not found");
return R.ok(kb); return R.ok(withLivePageCount(kb));
}
/**
* Overlay the live page count onto knowledge bases before returning them.
* The {@code pageCount} column is denormalized and only refreshed by the
* processing pipeline, so system-page generation (overview/log) and other
* out-of-band mutations leave it stale. Recomputing on read keeps the count
* the UI shows consistent with the page list.
*/
private List<WikiKnowledgeBaseEntity> withLivePageCount(List<WikiKnowledgeBaseEntity> kbs) {
kbs.forEach(this::withLivePageCount);
return kbs;
}
private WikiKnowledgeBaseEntity withLivePageCount(WikiKnowledgeBaseEntity kb) {
if (kb != null && kb.getId() != null) {
kb.setPageCount(pageService.countByKbId(kb.getId()));
}
return kb;
} }
@RequireWorkspaceRole("viewer") @RequireWorkspaceRole("viewer")
@ -84,9 +103,9 @@ public class WikiController {
long wsId = workspaceId != null ? workspaceId : 1L; long wsId = workspaceId != null ? workspaceId : 1L;
// 按 agent 查询后，过滤出属于当前 workspace 的知识库 // 按 agent 查询后，过滤出属于当前 workspace 的知识库
List<WikiKnowledgeBaseEntity> kbs = kbService.listByAgentId(agentId); List<WikiKnowledgeBaseEntity> kbs = kbService.listByAgentId(agentId);
return R.ok(kbs.stream() return R.ok(withLivePageCount(kbs.stream()
.filter(kb -> kb.getWorkspaceId() == null || kb.getWorkspaceId().equals(wsId)) .filter(kb -> kb.getWorkspaceId() == null || kb.getWorkspaceId().equals(wsId))
.toList()); .collect(java.util.stream.Collectors.toList())));
} }
@RequireWorkspaceRole("member") @RequireWorkspaceRole("member")