Move deallocator to the right place

Col-E · May 21, 2022 · 0c34623 · 0c34623
1 parent 59711d0
commit 0c34623
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/main/java/software/coley/llzip/util/BufferData.java b/src/main/java/software/coley/llzip/util/BufferData.java
@@ -103,10 +103,10 @@ public void close() {
 				if (cleaned)
 					return;
 				cleaned = true;
-			}
-			ByteBuffer buffer = this.buffer;
-			if (buffer.isDirect()) {
-				CleanerUtil.invokeCleaner(buffer);
+				ByteBuffer buffer = this.buffer;
+				if (buffer.isDirect()) {
+					CleanerUtil.invokeCleaner(buffer);
+				}
 			}
 		}
 	}

diff --git a/src/main/java/software/coley/llzip/util/UnsafeMappedFile.java b/src/main/java/software/coley/llzip/util/UnsafeMappedFile.java
@@ -116,11 +116,11 @@ public void close() {
 				if (cleaned)
 					return;
 				cleaned = true;
+				Runnable deallocator = this.deallocator;
+				if (deallocator != null)
+					deallocator.run();
 			}
 		}
-		Runnable deallocator = this.deallocator;
-		if (deallocator != null)
-			deallocator.run();
 	}
 
 	private void ensureOpen() {