Close iterator and add missing locking.

linkedfactory · Jul 18, 2024 · 03afdf1 · 03afdf1
1 parent 1fe5e46
commit 03afdf1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/...nkedfactory.core/src/main/java/io/github/linkedfactory/core/kvin/parquet/KvinParquet.java b/...nkedfactory.core/src/main/java/io/github/linkedfactory/core/kvin/parquet/KvinParquet.java
@@ -991,7 +991,11 @@ public boolean hasNext() {
 						currentTuple = getFirstTuple(URIs.createURI(previousTuple.getItem()), previousTuple.getItemId(),
 								previousTuple.getPropertyId() + 1, null);
 					}
-					return currentTuple != null;
+					if (currentTuple == null) {
+						close();
+						return false;
+					}
+					return true;
 				}
 
 				@Override

diff --git a/...ory.core/src/main/java/io/github/linkedfactory/core/kvin/partitioned/KvinPartitioned.java b/...ory.core/src/main/java/io/github/linkedfactory/core/kvin/partitioned/KvinPartitioned.java
@@ -151,7 +151,12 @@ public void put(KvinTuple... tuples) {
 
 	@Override
 	public void put(Iterable<KvinTuple> tuples) {
-		hotStore.put(tuples);
+		Lock readLock = readLock();
+		try {
+			hotStore.put(tuples);
+		} finally {
+			readLock.release();
+		}
 	}
 
 	public void createNewHotDataStore() throws IOException {