feat: add util function fence_vector_index

src/index/functions.rs

@@ -1,7 +1,41 @@
+use super::utils::from_oid_to_handle;
+use crate::error::check_client;
+use crate::ipc::client;
+use base::index::StatError;
+use pgrx::pg_sys::Oid;
+
 #[pgrx::pg_extern(volatile, strict, parallel_safe)]
 fn _vectors_pgvectors_upgrade() {
     if crate::bgworker::is_started() {
         return;
     }
     let _ = std::fs::remove_dir_all("pg_vectors");
 }
+
+#[pgrx::pg_extern(volatile, strict, parallel_safe)]
+fn _vectors_fence_vector_index(oid: Oid) {
+    let handle = from_oid_to_handle(oid);
+    let mut rpc = check_client(client());
+    loop {
+        pgrx::check_for_interrupts!();
+        match rpc.stat(handle) {
+            Ok(s) => {
+                if !s.indexing {
+                    break;
+                }
+            }
+            Err(StatError::NotExist) => pgrx::error!("internal error"),
+        }
+        unsafe {
+            pgrx::pg_sys::WaitLatch(
+                pgrx::pg_sys::MyLatch,
+                (pgrx::pg_sys::WL_LATCH_SET
+                    | pgrx::pg_sys::WL_TIMEOUT
+                    | pgrx::pg_sys::WL_EXIT_ON_PM_DEATH) as _,
+                1000,
+                pgrx::pg_sys::WaitEventTimeout_WAIT_EVENT_PG_SLEEP,
+            );
+            pgrx::pg_sys::ResetLatch(pgrx::pg_sys::MyLatch);
+        }
+    }
+}

src/sql/finalize.sql

@@ -612,6 +612,9 @@ $$;
 CREATE FUNCTION alter_vector_index("index" OID, "key" TEXT, "value" TEXT) RETURNS void
 STRICT LANGUAGE c AS 'MODULE_PATHNAME', '_vectors_alter_vector_index_wrapper';
 
+CREATE FUNCTION fence_vector_index(oid) RETURNS void
+STRICT PARALLEL SAFE LANGUAGE c AS 'MODULE_PATHNAME', '_vectors_fence_vector_index_wrapper';
+
 CREATE FUNCTION vector_dims(vector) RETURNS INT
 STRICT PARALLEL SAFE LANGUAGE c AS 'MODULE_PATHNAME', '_vectors_vecf32_dims_wrapper';