show mwthods for GUC vars

Vladimir Ershov · Vladimir Ershov · commit 1ab15cff947e · 2017-08-25T18:25:39.000+03:00
diff --git a/src/memutils.c b/src/memutils.c
@@ -12,6 +12,11 @@ MemoryContext init_mem_ctx(const char *name)
 								 ALLOCSET_DEFAULT_MAXSIZE);
 }
 
+bool is_worker_context_initialized(void)
+{
+	return SchedulerWorkerContext == NULL ? false: true;
+}
+
 MemoryContext init_worker_mem_ctx(const char *name)
 {
 	AssertState(SchedulerWorkerContext == NULL);
@@ -47,6 +52,15 @@ void *worker_alloc(Size size)
 	return MemoryContextAlloc(SchedulerWorkerContext, size);
 }
 
+void drop_worker_context(void)
+{
+	if(SchedulerWorkerContext)
+	{
+		MemoryContextDelete(SchedulerWorkerContext);
+		SchedulerWorkerContext = NULL;
+	}
+}
+
 void delete_worker_mem_ctx(MemoryContext old)
 {
 	if(!old) old = TopMemoryContext;
diff --git a/src/memutils.h b/src/memutils.h
@@ -16,4 +16,7 @@ char *_mcopy_string(MemoryContext ctx, char *str);
 char *my_copy_string(char *str);
 void check_scheduler_context(void);
 
+bool is_worker_context_initialized(void);
+void drop_worker_context(void);
+
 #endif
diff --git a/src/pgpro_scheduler.c b/src/pgpro_scheduler.c
@@ -47,6 +47,9 @@
 PG_MODULE_MAGIC;
 #endif
 
+static const char *show_scheduler_nodename(void);
+static const char *show_scheduler_database(void);
+
 volatile sig_atomic_t got_sighup = false;
 volatile sig_atomic_t got_sigterm = false;
 
@@ -297,6 +300,41 @@ char *set_schema(const char *name, bool get_old)
 	return current;
 }
 
+static const char *
+show_scheduler_nodename(void)
+{
+	static char nbuf[256];
+	char *nname;
+
+	nname = get_scheduler_nodename(CurrentMemoryContext);
+	snprintf(nbuf, sizeof(nbuf), "%s", nname);
+	pfree(nname);
+	return nbuf;
+}
+
+static const char *
+show_scheduler_database(void)
+{
+	static char nbuf[256];
+	const char *value;
+	bool drop_context = false;
+
+	if(!is_worker_context_initialized())
+	{
+		init_worker_mem_ctx("temp for functions");
+		drop_context = true;
+	}
+
+	value = check_multimaster_database();
+
+	if(!value)
+		value = GetConfigOption("schedule.database", true, false);
+	snprintf(nbuf, sizeof(nbuf), "%s", value);
+
+	if(drop_context) drop_worker_context();
+	return nbuf;
+}
+
 char *get_scheduler_nodename(MemoryContext mem)
 {
 	const char *opt;
@@ -670,7 +708,7 @@ void _PG_init(void)
 		0,
 		NULL,
 		NULL,
-		NULL
+		show_scheduler_database
 	);
 	DefineCustomStringVariable(
 		"schedule.nodename",
@@ -682,7 +720,7 @@ void _PG_init(void)
 		0,
 		NULL,
 		NULL,
-		NULL
+		show_scheduler_nodename
 	);
 	DefineCustomStringVariable(
 		"schedule.transaction_state",
