send GUC variables to nodes

kelvich · kelvich · commit 26b59ca331f6 · 2017-11-13T00:15:27.000+03:00
diff --git a/multimaster.c b/multimaster.c
@@ -139,7 +139,7 @@ HTAB* MtmXid2State;
 static HTAB* MtmGid2State;
 static HTAB* MtmLocalTables;
 
-static bool  MtmIsRecoverySession;
+static bool MtmIsRecoverySession;
 
 static MtmCurrentTrans MtmTx;
 
@@ -198,6 +198,9 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 							 ProcessUtilityContext context, ParamListInfo params,
 							 DestReceiver *dest, char *completionTag);
 
+static StringInfo	MtmGUCBuffer;
+static bool			MtmGUCBufferAllocated = false;
+
 /*
  * -------------------------------------------
  * Synchronize access to MTM structures.
@@ -2238,6 +2241,12 @@ static void MtmBroadcastUtilityStmt(char const* sql, bool ignoreError)
 	{ 
 		if (conns[i]) 
 		{
+			if (MtmGUCBufferAllocated && !MtmRunUtilityStmt(conns[i], MtmGUCBuffer->data, &utility_errmsg) && !ignoreError)
+			{
+				errorMsg = "Failed to set GUC variables at node %d";
+				failedNode = i;
+				break;
+			}
 			if (!MtmRunUtilityStmt(conns[i], "BEGIN TRANSACTION", &utility_errmsg) && !ignoreError)
 			{
 				errorMsg = "Failed to start transaction at node %d";
@@ -2249,7 +2258,10 @@ static void MtmBroadcastUtilityStmt(char const* sql, bool ignoreError)
 				if (i + 1 == MtmNodeId)
 					errorMsg = utility_errmsg;
 				else
+				{
+					elog(ERROR, utility_errmsg);
 					errorMsg = "Failed to run command at node %d";
+				}
 
 				failedNode = i;
 				break;
@@ -2407,7 +2419,6 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 		case T_FetchStmt:
 		case T_DoStmt:
 		case T_CreateTableSpaceStmt:
-		case T_DropTableSpaceStmt:
 		case T_AlterTableSpaceOptionsStmt:
 		case T_TruncateStmt:
 		case T_CommentStmt: /* XXX: we could replicate these */;
@@ -2416,9 +2427,9 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 		case T_ExecuteStmt:
 		case T_DeallocateStmt:
 		case T_GrantStmt: /* XXX: we could replicate some of these these */;
-		case T_GrantRoleStmt:
-		case T_AlterDatabaseStmt:
-		case T_AlterDatabaseSetStmt:
+		//case T_GrantRoleStmt:
+		//case T_AlterDatabaseStmt:
+		//case T_AlterDatabaseSetStmt:
 		case T_NotifyStmt:
 		case T_ListenStmt:
 		case T_UnlistenStmt:
@@ -2427,22 +2438,46 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 		case T_VacuumStmt:
 		case T_ExplainStmt:
 		case T_AlterSystemStmt:
-		case T_VariableSetStmt:
 		case T_VariableShowStmt:
 		case T_DiscardStmt:
-		case T_CreateEventTrigStmt:
-		case T_AlterEventTrigStmt:
-		case T_CreateRoleStmt:
-		case T_AlterRoleStmt:
-		case T_AlterRoleSetStmt:
-		case T_DropRoleStmt:
+		//case T_CreateEventTrigStmt:
+		//case T_AlterEventTrigStmt:
+		//case T_CreateRoleStmt:
+		//case T_AlterRoleStmt:
+		//case T_AlterRoleSetStmt:
+		//case T_DropRoleStmt:
 		case T_ReassignOwnedStmt:
 		case T_LockStmt:
-		case T_ConstraintsSetStmt:
+		//case T_ConstraintsSetStmt:
 		case T_CheckPointStmt:
 		case T_ReindexStmt:
 		    skipCommand = true;
 			break;
+		case T_VariableSetStmt:
+			{
+				//VariableSetStmt *stmt = (VariableSetStmt *) parsetree;
+
+				if (!MtmGUCBufferAllocated)
+				{
+					MemoryContext oldcontext;
+
+					oldcontext = MemoryContextSwitchTo(TopMemoryContext);
+					MtmGUCBuffer = makeStringInfo();
+					MemoryContextSwitchTo(oldcontext);
+					MtmGUCBufferAllocated = true;
+				}
+
+				//appendStringInfoString(MtmGUCBuffer, "SET ");
+				//appendStringInfoString(MtmGUCBuffer, stmt->name);
+				//appendStringInfoString(MtmGUCBuffer, " TO ");
+				//appendStringInfoString(MtmGUCBuffer, ExtractSetVariableArgs(stmt));
+				//appendStringInfoString(MtmGUCBuffer, "; ");
+
+				appendStringInfoString(MtmGUCBuffer, queryString);
+
+				skipCommand = true;
+			}
+			break;
 		case T_CreateStmt:
 			{
 				/* Do not replicate temp tables */
@@ -2490,7 +2525,6 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 				}
 			}
 			break;
-		case T_CreateSchemaStmt:
 	    default:
 			skipCommand = false;
 			break;
