fix: set copy context to null after deleting it

CrunchyData · Jan 21, 2025 · 2ca5739 · 2ca5739
1 parent 47a0209
commit 2ca5739
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/parquet_copy_hook/copy_to_dest_receiver.rs b/src/parquet_copy_hook/copy_to_dest_receiver.rs
@@ -126,6 +126,8 @@ impl CopyToParquetDestReceiver {
     fn cleanup(&mut self) {
         if !self.per_copy_context.is_null() {
             unsafe { MemoryContextDelete(self.per_copy_context) };
+
+            self.per_copy_context = std::ptr::null_mut();
         }
 
         if !self.parquet_writer_context.is_null() {