[BUG][JNI] Trigger MemoryBuffer.onClosed after memory is freed

java/src/main/java/ai/rapids/cudf/ColumnVector.java

@@ -1,6 +1,6 @@
 /*
  *
- *  Copyright (c) 2019-2023, NVIDIA CORPORATION.
+ *  Copyright (c) 2019-2024, NVIDIA CORPORATION.
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
  *  you may not use this file except in compliance with the License.
@@ -260,15 +260,18 @@ public void noWarnLeakExpected() {
   public synchronized void close() {
     refCount--;
     offHeap.delRef();
-    if (eventHandler != null) {
-      eventHandler.onClosed(this, refCount);
-    }
-    if (refCount == 0) {
-      super.close();
-      offHeap.clean(false);
-    } else if (refCount < 0) {
-      offHeap.logRefCountDebug("double free " + this);
-      throw new IllegalStateException("Close called too many times " + this);
+    try {
+      if (refCount == 0) {
+        super.close();
+        offHeap.clean(false);
+      } else if (refCount < 0) {
+        offHeap.logRefCountDebug("double free " + this);
+        throw new IllegalStateException("Close called too many times " + this);
+      }
+    } finally {
+      if (eventHandler != null) {
+        eventHandler.onClosed(this, refCount);
+      }
     }
   }
 

java/src/main/java/ai/rapids/cudf/HostColumnVector.java

@@ -150,17 +150,20 @@ public void noWarnLeakExpected() {
   public synchronized void close() {
     refCount--;
     offHeap.delRef();
-    if (eventHandler != null) {
-      eventHandler.onClosed(this, refCount);
-    }
-    if (refCount == 0) {
-      offHeap.clean(false);
-      for( HostColumnVectorCore child : children) {
-        child.close();
+    try {
+      if (refCount == 0) {
+        offHeap.clean(false);
+        for (HostColumnVectorCore child : children) {
+          child.close();
+        }
+      } else if (refCount < 0) {
+        offHeap.logRefCountDebug("double free " + this);
+        throw new IllegalStateException("Close called too many times " + this);
+      }
+    } finally {
+      if (eventHandler != null) {
+        eventHandler.onClosed(this, refCount);
       }
-    } else if (refCount < 0) {
-      offHeap.logRefCountDebug("double free " + this);
-      throw new IllegalStateException("Close called too many times " + this);
     }
   }
 

java/src/main/java/ai/rapids/cudf/MemoryBuffer.java

@@ -1,6 +1,6 @@
 /*
  *
- *  Copyright (c) 2019-2022, NVIDIA CORPORATION.
+ *  Copyright (c) 2019-2024, NVIDIA CORPORATION.
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
  *  you may not use this file except in compliance with the License.
@@ -240,15 +240,18 @@ public synchronized void close() {
     if (cleaner != null) {
       refCount--;
       cleaner.delRef();
-      if (eventHandler != null) {
-        eventHandler.onClosed(refCount);
-      }
-      if (refCount == 0) {
-        cleaner.clean(false);
-        closed = true;
-      } else if (refCount < 0) {
-        cleaner.logRefCountDebug("double free " + this);
-        throw new IllegalStateException("Close called too many times " + this);
+      try {
+        if (refCount == 0) {
+          cleaner.clean(false);
+          closed = true;
+        } else if (refCount < 0) {
+          cleaner.logRefCountDebug("double free " + this);
+          throw new IllegalStateException("Close called too many times " + this);
+        }
+      } finally {
+        if (eventHandler != null) {
+          eventHandler.onClosed(refCount);
+        }
       }
     }
   }