catch current user on backend startup

kelvich · kelvich · commit 68abec257b4d · 2017-11-13T02:17:36.000+03:00
diff --git a/multimaster.c b/multimaster.c
@@ -4487,10 +4487,12 @@ typedef struct MtmGucEntry
 
 static HTAB *MtmGucHash = NULL;
 static dlist_head MtmGucList = DLIST_STATIC_INIT(MtmGucList);
+static inline void MtmGucUpdate(const char *key, char *value);
 
 static void MtmGucInit(void)
 {
 	HASHCTL		hash_ctl;
+	char	   *current_role;
 
 	MemSet(&hash_ctl, 0, sizeof(hash_ctl));
 	hash_ctl.keysize = GUC_KEY_MAXLEN;
@@ -4500,6 +4502,14 @@ static void MtmGucInit(void)
 						MTM_GUC_HASHSIZE,
 						&hash_ctl,
 						HASH_ELEM | HASH_CONTEXT);
+
+	/*
+	 * If current role is not equal to MtmDatabaseUser, than set it bofore
+	 * any other GUC vars.
+	 */
+	current_role = GetConfigOptionByName("session_authorization", NULL, false);
+	if (strcmp(MtmDatabaseUser, current_role) != 0)
+		MtmGucUpdate("session_authorization", current_role);
 }
 
 static void MtmGucDiscard()
@@ -4593,6 +4603,9 @@ char* MtmGucSerialize(void)
 	dlist_iter iter;
 	int nvars = 0;
 
+	if (!MtmGucHash)
+		MtmGucInit();
+
 	serialized_gucs = makeStringInfo();
 
 	dlist_foreach(iter, &MtmGucList)
diff --git a/sql/multimaster.sql b/sql/multimaster.sql
@@ -0,0 +1,10 @@
+create user user1;
+create schema user1;
+alter schema user1 owner to user1;
+
+\c "user=user1 dbname=regression"
+create table user1.test(i int primary key);
+
+\c "user=user1 dbname=regression port=5433"
+select * from test;
+
