Fix some bugs

src/backend/distributed/shared_library_init.c

@@ -481,6 +481,7 @@ _PG_init(void)
 #endif
 
 	InitializeMaintenanceDaemon();
+	InitializeMaintenanceDaemonForAdminDB();
 
 	/* initialize coordinated transaction management */
 	InitializeTransactionManagement();

src/backend/distributed/utils/maintenanced.c

@@ -134,18 +134,19 @@ static void WarnMaintenanceDaemonNotStarted(void);
 void
 InitializeMaintenanceDaemon(void)
 {
-	elog(LOG, "InitializeMaintenanceDaemon");
 	prev_shmem_startup_hook = shmem_startup_hook;
 	shmem_startup_hook = MaintenanceDaemonShmemInit;
 }
 
+
+/*
+ * InitializeMaintenanceDaemonForAdminDB is called in _PG_Init
+ * at which stage we are not in a transaction or have databaseOid
+ */
 void
 InitializeMaintenanceDaemonForAdminDB(void)
 {
-	elog(LOG, "InitializeMaintenanceDaemonForAdmin");
-	
 	BackgroundWorker worker;
-        BackgroundWorkerHandle *handle = NULL;
 
 	memset(&worker, 0, sizeof(worker));
 
@@ -169,19 +170,12 @@ InitializeMaintenanceDaemonForAdminDB(void)
 	strcpy_s(worker.bgw_function_name, sizeof(worker.bgw_library_name),
 			 "CitusMaintenanceDaemonMain");
 
-        worker.bgw_main_arg = (Datum)0;
-        worker.bgw_notify_pid = MyProcPid;
-
-        if (!RegisterDynamicBackgroundWorker(&worker, &handle)) {
-		elog(LOG, "RegisterDynamicBackgroundWorker failed for admin");
-	}
-
-	pid_t pid;
-        WaitForBackgroundWorkerStartup(handle, &pid);
-
-        pfree(handle);
+	worker.bgw_main_arg = (Datum) 0;
 
+	RegisterBackgroundWorker(&worker);
 }
+
+
 /*
  * InitializeMaintenanceDaemonBackend, called at backend start and
  * configuration changes, is responsible for starting a per-database
@@ -190,7 +184,6 @@ InitializeMaintenanceDaemonForAdminDB(void)
 void
 InitializeMaintenanceDaemonBackend(void)
 {
-	elog(LOG, "InitializeMaintenanceDaemonBackend");
 	Oid extensionOwner = CitusExtensionOwner();
 	bool found;
 
@@ -322,7 +315,6 @@ WarnMaintenanceDaemonNotStarted(void)
 void
 CitusMaintenanceDaemonMain(Datum main_arg)
 {
-	elog(LOG, "CitusMaintenanceDaemonMain");
 	Oid databaseOid = DatumGetObjectId(main_arg);
 	TimestampTz nextStatsCollectionTime USED_WITH_LIBCURL_ONLY =
 		TimestampTzPlusMilliseconds(GetCurrentTimestamp(), 60 * 1000);
@@ -342,14 +334,13 @@ CitusMaintenanceDaemonMain(Datum main_arg)
 
 	if (databaseOid == 0)
 	{
-
 		/* TODO : Get the admin database name from GUC contro_db*/
-		char* databaseName = "postgres";
+		char *databaseName = "postgres";
 
 		BackgroundWorkerInitializeConnection(databaseName, NULL, 0);
 
-		// Now we have a valid MyDatabaseId.
-		// Insert the daemon instance to the hash table.
+		/* Now we have a valid MyDatabaseId. */
+		/* Insert the daemon instance to the hash table. */
 		bool found;
 
 		LWLockAcquire(&MaintenanceDaemonControl->lock, LW_EXCLUSIVE);
@@ -371,14 +362,11 @@ CitusMaintenanceDaemonMain(Datum main_arg)
 			/* ensure the values in MaintenanceDaemonDBData are zero */
 			memset(((char *) myDbData) + sizeof(Oid), 0,
 				   sizeof(MaintenanceDaemonDBData) - sizeof(Oid));
-
-        	     myDbData->userOid = 0;
-                     myDbData->workerPid = 0;
-	             myDbData->triggerNodeMetadataSync = false;
 		}
 
 		before_shmem_exit(MaintenanceDaemonShmemExit, ObjectIdGetDatum(MyDatabaseId));
 		databaseOid = MyDatabaseId;
+		myDbData->userOid = GetSessionUserId();
 	}
 	else
 	{
@@ -398,12 +386,12 @@ CitusMaintenanceDaemonMain(Datum main_arg)
 
 			LWLockRelease(&MaintenanceDaemonControl->lock);
 			proc_exit(0);
-        	
 		}
 		before_shmem_exit(MaintenanceDaemonShmemExit, main_arg);
 
 		BackgroundWorkerInitializeConnectionByOid(databaseOid, myDbData->userOid, 0);
 	}
+
 	/*
 	 * We do metadata sync in a separate background worker. We need its
 	 * handle to be able to check its status.
@@ -945,7 +933,6 @@ MaintenanceDaemonShmemSize(void)
 void
 MaintenanceDaemonShmemInit(void)
 {
-	elog(LOG, "MaintenanceDaemonShmemInit");
 	bool alreadyInitialized = false;
 	HASHCTL hashInfo;
 
@@ -997,7 +984,6 @@ MaintenanceDaemonShmemInit(void)
 static void
 MaintenanceDaemonShmemExit(int code, Datum arg)
 {
-	elog(LOG, "MaintenanceDaemonShmemExit");
 	Oid databaseOid = DatumGetObjectId(arg);
 
 	LWLockAcquire(&MaintenanceDaemonControl->lock, LW_EXCLUSIVE);
@@ -1108,7 +1094,6 @@ LockCitusExtension(void)
 void
 StopMaintenanceDaemon(Oid databaseId)
 {
-	elog(LOG, "StopMaintenanceDaemon");
 	bool found = false;
 	pid_t workerPid = 0;
 
@@ -1140,7 +1125,6 @@ StopMaintenanceDaemon(Oid databaseId)
 void
 TriggerNodeMetadataSync(Oid databaseId)
 {
-	elog(LOG, "TriggerNodeMetadataSync");
 	bool found = false;
 
 	LWLockAcquire(&MaintenanceDaemonControl->lock, LW_EXCLUSIVE);
@@ -1168,7 +1152,6 @@ TriggerNodeMetadataSync(Oid databaseId)
 static bool
 MetadataSyncTriggeredCheckAndReset(MaintenanceDaemonDBData *dbData)
 {
-	elog(LOG, "MetadataSyncTriggeredCheckAndReset");
 	LWLockAcquire(&MaintenanceDaemonControl->lock, LW_EXCLUSIVE);
 
 	bool metadataSyncTriggered = dbData->triggerNodeMetadataSync;