fix: add vacuum_delay_point (tensorchord#31)

Signed-off-by: usamoi <usamoi@outlook.com>

src/algorithm/vacuum.rs

@@ -2,7 +2,7 @@ use crate::algorithm::tuples::*;
 use crate::postgres::Relation;
 use base::search::Pointer;
 
-pub fn vacuum(relation: Relation, callback: impl Fn(Pointer) -> bool) {
+pub fn vacuum(relation: Relation, delay: impl Fn(), callback: impl Fn(Pointer) -> bool) {
     // step 1: vacuum height_0_tuple
     {
         let h1_firsts = {
@@ -42,6 +42,7 @@ pub fn vacuum(relation: Relation, callback: impl Fn(Pointer) -> bool) {
         for first in h0_firsts {
             let mut current = first;
             while current != u32::MAX {
+                delay();
                 let mut h0_guard = relation.write(current);
                 for i in 1..=h0_guard.get().len() {
                     let h0_tuple = h0_guard
@@ -97,6 +98,7 @@ pub fn vacuum(relation: Relation, callback: impl Fn(Pointer) -> bool) {
             meta_tuple.vectors_first
         };
         while current != u32::MAX {
+            delay();
             let read = relation.read(current);
             let flag = 'flag: {
                 for i in 1..=read.get().len() {

src/index/am.rs

@@ -732,7 +732,13 @@ pub unsafe extern "C" fn ambulkdelete(
     }
     let callback = callback.unwrap();
     let callback = |p: Pointer| unsafe { callback(&mut pointer_to_ctid(p), callback_state) };
-    algorithm::vacuum::vacuum(unsafe { Relation::new((*info).index) }, callback);
+    algorithm::vacuum::vacuum(
+        unsafe { Relation::new((*info).index) },
+        || unsafe {
+            pgrx::pg_sys::vacuum_delay_point();
+        },
+        callback,
+    );
     stats
 }