From 2bc8cf25a3d777a590fd54bfef0303f98bc225a2 Mon Sep 17 00:00:00 2001
From: xueqingkun <xueqingkun@126.com>
Date: Mon, 18 Nov 2024 11:54:38 +0800
Subject: [PATCH] =?UTF-8?q?1.=20=E4=BF=AE=E5=A4=8D=E8=B0=B1=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E8=B0=83=E8=AF=95=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../police/service/impl/ModelIndexServiceImpl.java    | 11 +++++++++++
 .../com/supervision/police/vo/GraphDebugResVO.java    |  3 +++
 2 files changed, 14 insertions(+)

diff --git a/src/main/java/com/supervision/police/service/impl/ModelIndexServiceImpl.java b/src/main/java/com/supervision/police/service/impl/ModelIndexServiceImpl.java
index 006c370..31609df 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelIndexServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelIndexServiceImpl.java
@@ -3,6 +3,7 @@ package com.supervision.police.service.impl;
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.lang.Assert;
 import cn.hutool.core.lang.Pair;
+import cn.hutool.core.map.MapUtil;
 import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.json.JSONUtil;
@@ -49,6 +50,7 @@ import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.io.Serializable;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -451,6 +453,15 @@ public class ModelIndexServiceImpl extends ServiceImpl<ModelIndexMapper, ModelIn
         graphDebugResVO.setGraphRelList(pair.getKey());
         graphDebugResVO.setGraphNodeList(pair.getValue());
 
+        List<Map<String, Object>> collect = graphDebugResVO.getGraphRelList().stream().collect(Collectors.groupingBy(WebRelDTO::getName, Collectors.counting()))
+                .entrySet().stream().map(entry -> {
+                    Map<String, Object> map = new HashMap<>();
+                    map.put("name", entry.getKey());
+                    map.put("count", entry.getValue());
+                    return map;
+                }).collect(Collectors.toList());
+        graphDebugResVO.setGraphReCountList(collect);
+
         // 尝试转换为图形数据
         return graphDebugResVO;
     }
diff --git a/src/main/java/com/supervision/police/vo/GraphDebugResVO.java b/src/main/java/com/supervision/police/vo/GraphDebugResVO.java
index f7435ce..7259538 100644
--- a/src/main/java/com/supervision/police/vo/GraphDebugResVO.java
+++ b/src/main/java/com/supervision/police/vo/GraphDebugResVO.java
@@ -31,6 +31,9 @@ public class GraphDebugResVO {
      */
     private List<WebRelDTO> graphRelList;
 
+
+    private List<Map<String, Object>> graphReCountList;
+
     /**
      * 图谱节点数据 name/entityName/id
      */