@@ -3,6 +3,7 @@ use std::sync::LazyLock;
33
44use parquet_copy_hook:: hook:: { init_parquet_copy_hook, ENABLE_PARQUET_COPY_HOOK } ;
55use parquet_copy_hook:: pg_compat:: MarkGUCPrefixReserved ;
6+ use pgrx:: memcxt:: PgMemoryContexts ;
67use pgrx:: pg_sys:: AsPgCStr ;
78use pgrx:: { prelude:: * , GucContext , GucFlags , GucRegistry } ;
89use tokio:: runtime:: Runtime ;
@@ -42,14 +43,16 @@ pub extern "C-unwind" fn _PG_init() {
4243 }
4344
4445 unsafe {
45- GucRegistry :: define_bool_guc (
46- CStr :: from_ptr ( "pg_parquet.enable_copy_hooks" . as_pg_cstr ( ) ) ,
47- CStr :: from_ptr ( "Enable parquet copy hooks" . as_pg_cstr ( ) ) ,
48- CStr :: from_ptr ( "Enable parquet copy hooks" . as_pg_cstr ( ) ) ,
49- & ENABLE_PARQUET_COPY_HOOK ,
50- GucContext :: Userset ,
51- GucFlags :: default ( ) ,
52- )
46+ PgMemoryContexts :: TopMemoryContext . switch_to ( |_| {
47+ GucRegistry :: define_bool_guc (
48+ CStr :: from_ptr ( "pg_parquet.enable_copy_hooks" . as_pg_cstr ( ) ) ,
49+ CStr :: from_ptr ( "Enable parquet copy hooks" . as_pg_cstr ( ) ) ,
50+ CStr :: from_ptr ( "Enable parquet copy hooks" . as_pg_cstr ( ) ) ,
51+ & ENABLE_PARQUET_COPY_HOOK ,
52+ GucContext :: Userset ,
53+ GucFlags :: default ( ) ,
54+ )
55+ } ) ;
5356 } ;
5457
5558 MarkGUCPrefixReserved ( "pg_parquet" ) ;
0 commit comments