Add base_init_startup_hook

pashkinelfe · Feb 29, 2024 · 6f3d57c · 6f3d57c
1 parent e379850
commit 6f3d57c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/backend/utils/init/postinit.c b/src/backend/utils/init/postinit.c
@@ -81,7 +81,7 @@ static void ClientCheckTimeoutHandler(void);
 static bool ThereIsAtLeastOneRole(void);
 static void process_startup_options(Port *port, bool am_superuser);
 static void process_settings(Oid databaseid, Oid roleid);
-
+base_init_startup_hook_type base_init_startup_hook = NULL;
 
 /*** InitPostgres support ***/
 
@@ -641,6 +641,9 @@ BaseInit(void)
 	 */
 	InitFileAccess();
 
+	if (base_init_startup_hook)
+		base_init_startup_hook();
+
 	/*
 	 * Initialize statistics reporting. This needs to happen early to ensure
 	 * that pgstat's shutdown callback runs after the shutdown callbacks of

diff --git a/src/include/postmaster/postmaster.h b/src/include/postmaster/postmaster.h
@@ -59,6 +59,10 @@ extern int	MaxLivePostmasterChildren(void);
 
 extern bool PostmasterMarkPIDForWorkerNotify(int);
 
+typedef void (*base_init_startup_hook_type)(void);
+
+extern PGDLLIMPORT base_init_startup_hook_type base_init_startup_hook;
+
 #ifdef EXEC_BACKEND
 extern pid_t postmaster_forkexec(int argc, char *argv[]);
 extern void SubPostmasterMain(int argc, char *argv[]) pg_attribute_noreturn();