Add GetSharedBackgroundWorkerHandle()

Nikita Glukhov · Nikita Glukhov · commit 9032918a0129 · 2017-05-02T22:14:02.000+03:00
diff --git a/src/backend/postmaster/bgworker.c b/src/backend/postmaster/bgworker.c
@@ -1028,6 +1028,16 @@ RegisterDynamicBackgroundWorker(BackgroundWorker *worker,
 	return success;
 }
 
+BackgroundWorkerHandle *
+GetSharedBackgroundWorkerHandle(BackgroundWorkerHandle *handle,
+								BackgroundWorkerHandle **shhandle)
+{
+	if (!*shhandle)
+		*shhandle = ShmemAlloc(sizeof(*handle));
+	memcpy(*shhandle, handle, sizeof(*handle));
+	return *shhandle;
+}
+
 /*
  * Get the PID of a dynamically-registered background worker.
  *
diff --git a/src/include/postmaster/bgworker.h b/src/include/postmaster/bgworker.h
@@ -128,6 +128,10 @@ extern BgwHandleStatus
 /* Terminate a bgworker */
 extern void TerminateBackgroundWorker(BackgroundWorkerHandle *handle);
 
+extern BackgroundWorkerHandle *
+			GetSharedBackgroundWorkerHandle(BackgroundWorkerHandle *handle,
+											BackgroundWorkerHandle **shhandle);
+
 /* This is valid in a running worker */
 extern PGDLLIMPORT BackgroundWorker *MyBgworkerEntry;
 
