spliterator-bugfix: Spliterator fix

ydb-platform · Mar 4, 2024 · c5d0adc · c5d0adc
1 parent 7003407
commit c5d0adc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/repository-ydb-v1/src/main/java/tech/ydb/yoj/repository/ydb/YdbSpliterator.java b/repository-ydb-v1/src/main/java/tech/ydb/yoj/repository/ydb/YdbSpliterator.java
@@ -119,6 +119,10 @@ private QueueValue<V> poll() {
     @Override
     // (stream thread) called from stream engine for get new value.
     public boolean tryAdvance(Consumer<? super V> action) {
+        if (closed) {
+            throw new IllegalStateException("Can't use closed YdbSpliterator");
+        }
+
         // Stream API can call tryAdvance() once again even if tryAdvance() returned false
         if (endData) {
             return false;

diff --git a/repository-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/YdbSpliterator.java b/repository-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/YdbSpliterator.java
@@ -120,6 +120,10 @@ private QueueValue<V> poll() {
     @Override
     // (stream thread) called from stream engine for get new value.
     public boolean tryAdvance(Consumer<? super V> action) {
+        if (closed) {
+            throw new IllegalStateException("Can't use closed YdbSpliterator");
+        }
+
         // Stream API can call tryAdvance() once again even if tryAdvance() returned false
         if (endData) {
             return false;