fix: use closable iterator util to close autoclosable edges rather th…

…an mannually close it
apache · Aug 18, 2023 · 4862a70 · 4862a70
1 parent 2ae0f6c
commit 4862a70
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/hugegraph-core/src/main/java/org/apache/hugegraph/traversal/algorithm/CountTraverser.java b/hugegraph-core/src/main/java/org/apache/hugegraph/traversal/algorithm/CountTraverser.java
@@ -32,6 +32,7 @@
 import org.apache.hugegraph.iterator.FlatMapperIterator;
 import org.apache.hugegraph.structure.HugeEdge;
 import org.apache.hugegraph.util.E;
+import org.apache.tinkerpop.gremlin.structure.util.CloseableIterator;
 
 public class CountTraverser extends HugeTraverser {
 
@@ -93,12 +94,7 @@ public long count(Id source, List<EdgeStep> steps,
 
             return this.count.longValue();
         } finally {
-            if (edges instanceof FlatMapperIterator) {
-                try {
-                    ((FlatMapperIterator<?, ?>)edges).close();
-                } catch (Exception ignored) {
-                }
-            }
+            CloseableIterator.closeIterator(edges);
         }
     }