From 6a1069e62dbabbe2e479276aef2151a0f4d0bf26 Mon Sep 17 00:00:00 2001
From: yanghuadong <dong@honeycomb.cn>
Date: Fri, 17 Nov 2023 09:10:00 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D/getApis=E5=9C=A8redis?=
 =?UTF-8?q?=E6=A8=A1=E5=BC=8F=E4=B8=8B=E5=BE=88=E6=85=A2=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore                                         | 6 ++++++
 src/main/java/cn/langpy/kotime/data/RedisBase.java | 9 +++++----
 2 files changed, 11 insertions(+), 4 deletions(-)
 create mode 100644 .gitignore

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..614d133
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,6 @@
+*.iml
+.idea/
+rebel.xml
+target/
+.txlcn
+*.log
diff --git a/src/main/java/cn/langpy/kotime/data/RedisBase.java b/src/main/java/cn/langpy/kotime/data/RedisBase.java
index 1ab2462..0fca248 100644
--- a/src/main/java/cn/langpy/kotime/data/RedisBase.java
+++ b/src/main/java/cn/langpy/kotime/data/RedisBase.java
@@ -219,12 +219,12 @@ public class RedisBase implements GraphService {
         List<MethodInfo> methodInfos = new ArrayList<>();
         List<MethodNode> smethodNodes = searchList(methodPre, MethodNode.class);
 
+        List<MethodRelation> methodRelations = searchList(methodRelationPre, MethodRelation.class);
+        Map<String, MethodRelation> methodRelationMap = methodRelations.stream().collect(Collectors.toMap(k -> k.getTargetId(), v -> v, (p1, p2) -> p1));
         for (MethodNode methodNode : smethodNodes) {
             if (MethodType.Controller == methodNode.getMethodType()) {
                 String id = methodNode.getId();
-                List<MethodRelation> smethodRelations = searchList(methodRelationPre, MethodRelation.class);
-
-                Optional<MethodRelation> relations = smethodRelations.stream().filter(methodRelation -> methodRelation.getTargetId().equals(id)).findFirst();
+                Optional<MethodRelation> relations = Optional.ofNullable(methodRelationMap.get(id));
                 MethodRelation relation = null;
                 if (relations.isPresent()) {
                     relation = relations.get();
@@ -266,10 +266,11 @@ public class RedisBase implements GraphService {
         List<MethodInfo> methodInfos = new ArrayList<>();
         List<MethodNode> smethodNodes = searchList(methodPre, MethodNode.class);
         List<MethodRelation> methodRelationList = searchList(methodRelationPre, MethodRelation.class);
+        Map<String, MethodRelation> methodRelationMap = methodRelationList.stream().collect(Collectors.toMap(k -> k.getTargetId(), v -> v, (p1, p2) -> p1));
         for (MethodNode methodNode : smethodNodes) {
             if (methodNode.getName().toLowerCase().contains(question.toLowerCase())) {
                 String id = methodNode.getId();
-                Optional<MethodRelation> relations = methodRelationList.stream().filter(methodRelation -> methodRelation.getTargetId().equals(id)).findFirst();
+                Optional<MethodRelation> relations = Optional.ofNullable(methodRelationMap.get(id));
                 MethodRelation relation = null;
                 if (relations.isPresent()) {
                     relation = relations.get();