diff --git a/framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseRelationshipEdgeMapper.xml b/framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseRelationshipEdgeMapper.xml
index f073f1c778..e616291808 100644
--- a/framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseRelationshipEdgeMapper.xml
+++ b/framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseRelationshipEdgeMapper.xml
@@ -3,15 +3,19 @@
 <mapper namespace="io.metersphere.base.mapper.ext.BaseRelationshipEdgeMapper">
 
   <select id="getGraphIdsByNodeIds" resultType="java.lang.String">
-    select distinct graph_id
+    select graph_id
     from relationship_edge
     where source_id in
     <foreach collection="ids" item="id" open="(" close=")" separator=",">
       #{id}
     </foreach>
-    or target_id in
+    union
+    select graph_id
+    from relationship_edge
+    where target_id in
     <foreach collection="ids" item="id" open="(" close=")" separator=",">
       #{id}
     </foreach>
   </select>
+
 </mapper>
diff --git a/framework/sdk-parent/sdk/src/main/java/io/metersphere/service/RelationshipEdgeService.java b/framework/sdk-parent/sdk/src/main/java/io/metersphere/service/RelationshipEdgeService.java
index 5d9e34c05c..f558b17820 100644
--- a/framework/sdk-parent/sdk/src/main/java/io/metersphere/service/RelationshipEdgeService.java
+++ b/framework/sdk-parent/sdk/src/main/java/io/metersphere/service/RelationshipEdgeService.java
@@ -293,9 +293,11 @@ public class RelationshipEdgeService {
         if (CollectionUtils.isEmpty(graphIds)) {
             return new ArrayList<>();
         }
+
         RelationshipEdgeExample example = new RelationshipEdgeExample();
         example.createCriteria()
-                .andGraphIdIn(graphIds);
+                .andGraphIdIn(graphIds.stream()
+                        .distinct().collect(Collectors.toList()));
 
         return relationshipEdgeMapper.selectByExample(example);
     }
@@ -319,7 +321,7 @@ public class RelationshipEdgeService {
         markSet.add(id);
         visitedSet.add(id);
 
-        ArrayList<String> nextLevelNodes = new ArrayList();
+        ArrayList<String> nextLevelNodes = new ArrayList<>();
         for (RelationshipEdge relationshipEdge : edges) {
             if (id.equals(relationshipEdge.getSourceId())) {
                 nextLevelNodes.add(relationshipEdge.getTargetId());