Update QueryResults.java

Fix bug: gremlin语句执行超时将导致内存泄漏statement未释放 fix apache#1639
z7658329 · Nov 10, 2021 · fea140e · fea140e
1 parent 927a991
commit fea140e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/query/QueryResults.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/query/QueryResults.java
@@ -170,7 +170,11 @@ private Set<Id> queryIds() {
     }
 
     public static <T> ListIterator<T> toList(Iterator<T> iterator) {
-        return new ListIterator<>(Query.DEFAULT_CAPACITY, iterator);
+        try {
+            return new ListIterator<>(Query.DEFAULT_CAPACITY, iterator);
+        } finally {
+            CloseableIterator.closeIterator(iterator);
+        }
     }
 
     public static <T> void fillList(Iterator<T> iterator, List<T> list) {