Add access hook

pgrx/src/hooks.rs

@@ -176,6 +176,24 @@ pub trait PgHooks {
         prev_hook(pstate, query, jumble_state)
     }
 
+    fn object_access_hook(
+        &mut self,
+        access: pg_sys::ObjectAccessType,
+        class_id: pg_sys::Oid,
+        object_id: pg_sys::Oid,
+        sub_id: ::std::os::raw::c_int,
+        arg: *mut ::std::os::raw::c_void,
+        prev_hook: fn(
+            access: pg_sys::ObjectAccessType,
+            class_id: pg_sys::Oid,
+            object_id: pg_sys::Oid,
+            sub_id: ::std::os::raw::c_int,
+            arg: *mut ::std::os::raw::c_void,
+        ) -> HookResult<()>,
+    ) -> HookResult<()> {
+        prev_hook(access, class_id, object_id, sub_id, arg)
+    }
+
     /// Called when the transaction aborts
     fn abort(&mut self) {}
 
@@ -194,6 +212,7 @@ struct Hooks {
     prev_process_utility_hook: pg_sys::ProcessUtility_hook_type,
     prev_planner_hook: pg_sys::planner_hook_type,
     prev_post_parse_analyze_hook: pg_sys::post_parse_analyze_hook_type,
+    object_access_hook: pg_sys::object_access_hook_type
 }
 
 static mut HOOKS: Option<Hooks> = None;
@@ -237,6 +256,9 @@ pub unsafe fn register_hook(hook: &'static mut (dyn PgHooks)) {
         prev_post_parse_analyze_hook: pg_sys::post_parse_analyze_hook
             .replace(pgrx_post_parse_analyze),
         prev_emit_log_hook: pg_sys::emit_log_hook.replace(pgrx_emit_log),
+        object_access_hook: pg_sys::object_access_hook_type
+            .replace(pgrx_object_access_hook)
+            .or(Some(pgrx_standard_object_access_hook_wrapper)),
     });
 
     #[pg_guard]