Add mtm.inject_2pc_error function

knizhnik · kelvich · commit 5b20bea963c0 · 2017-11-13T00:15:40.000+03:00
diff --git a/multimaster--1.0.sql b/multimaster--1.0.sql
@@ -65,6 +65,10 @@ CREATE FUNCTION mtm.poll_node(nodeId integer, noWait boolean default FALSE) RETU
 AS 'MODULE_PATHNAME','mtm_poll_node'
 LANGUAGE C;
 
+CREATE FUNCTION mtm.inject_2pc_error(stage integer) RETURNS void
+AS 'MODULE_PATHNAME','mtm_inject_2pc_error'
+LANGUAGE C;
+
 CREATE TABLE IF NOT EXISTS mtm.ddl_log (issued timestamp with time zone not null, query text);
 
 CREATE TABLE IF NOT EXISTS mtm.local_tables(rel_schema text, rel_name text, primary key(rel_schema, rel_name));
diff --git a/multimaster.c b/multimaster.c
@@ -113,6 +113,7 @@ PG_FUNCTION_INFO_V1(mtm_get_cluster_state);
 PG_FUNCTION_INFO_V1(mtm_get_cluster_info);
 PG_FUNCTION_INFO_V1(mtm_make_table_local);
 PG_FUNCTION_INFO_V1(mtm_dump_lock_graph);
+PG_FUNCTION_INFO_V1(mtm_inject_2pc_error);
 
 static Snapshot MtmGetSnapshot(Snapshot snapshot);
 static void MtmInitialize(void);
@@ -687,6 +688,10 @@ MtmPrePrepareTransaction(MtmCurrentTrans* x)
 		return;
 	}
 
+	if (Mtm->inject2PCError == 1) { 
+		Mtm->inject2PCError = 0;
+		elog(ERROR, "ERROR INJECTION for transaction %d (%s)", x->xid, x->gid);
+	}
 	x->xid = GetCurrentTransactionId();
 	Assert(TransactionIdIsValid(x->xid));
 
@@ -741,6 +746,10 @@ MtmPostPrepareTransaction(MtmCurrentTrans* x)
 { 
 	MtmTransState* ts;
 
+	if (Mtm->inject2PCError == 2) { 
+		Mtm->inject2PCError = 0;
+		elog(ERROR, "ERROR INJECTION for transaction %d (%s)", x->xid, x->gid);
+	}
 	MtmLock(LW_EXCLUSIVE);
 	ts = hash_search(MtmXid2State, &x->xid, HASH_FIND, NULL);
 	Assert(ts != NULL);
@@ -782,6 +791,10 @@ MtmPostPrepareTransaction(MtmCurrentTrans* x)
 		MTM_LOG3("%d: Result of vote: %d", MyProcPid, ts->status);
 		MtmUnlock();
 	}
+	if (Mtm->inject2PCError == 3) { 
+		Mtm->inject2PCError = 0;
+		elog(ERROR, "ERROR INJECTION for transaction %d (%s)", x->xid, x->gid);
+	}
 }
 
 
@@ -1483,6 +1496,7 @@ static void MtmInitialize()
 		Mtm->gcCount = 0;
 		Mtm->nConfigChanges = 0;
 		Mtm->localTablesHashLoaded = false;
+		Mtm->inject2PCError = 0;
 		for (i = 0; i < MtmNodes; i++) {
 			Mtm->nodes[i].oldestSnapshot = 0;
 			Mtm->nodes[i].transDelay = 0;
@@ -2426,12 +2440,13 @@ mtm_get_cluster_info(PG_FUNCTION_ARGS)
 		usrfctx = (MtmGetClusterInfoCtx*)palloc(sizeof(MtmGetNodeStateCtx));
 		get_call_result_type(fcinfo, NULL, &desc);
 		funcctx->attinmeta = TupleDescGetAttInMetadata(desc);
-		usrfctx->nodeId = 1;
+		usrfctx->nodeId = 0;
 		funcctx->user_fctx = usrfctx;
 		MemoryContextSwitchTo(oldcontext);      
     }
     funcctx = SRF_PERCALL_SETUP();	
 	usrfctx = (MtmGetClusterInfoCtx*)funcctx->user_fctx;
+	while (++usrfctx->nodeId <= Mtm->nAllNodes && BIT_CHECK(Mtm->disabledNodeMask, usrfctx->nodeId-1));
 	if (usrfctx->nodeId > Mtm->nAllNodes) {
 		SRF_RETURN_DONE(funcctx);      
 	}	
@@ -2527,6 +2542,12 @@ Datum mtm_dump_lock_graph(PG_FUNCTION_ARGS)
 	return CStringGetTextDatum(s->data);
 }
 
+Datum mtm_inject_2pc_error(PG_FUNCTION_ARGS)
+{
+	Mtm->inject2PCError = PG_GETARG_INT32(0);
+    PG_RETURN_VOID();
+}
+
 /*
  * -------------------------------------------
  * Broadcast utulity statements
@@ -2794,9 +2815,12 @@ static void MtmProcessUtility(Node *parsetree, const char *queryString,
 					}
 					break;
 				case TRANS_STMT_PREPARE:
+					elog(ERROR, "Two phase commit is not supported by multimaster");
+					break;
 				case TRANS_STMT_COMMIT_PREPARED:
 				case TRANS_STMT_ROLLBACK_PREPARED:
-					elog(ERROR, "Two phase commit is not supported by multimaster");
+  				    skipCommand = true;
+					break;
 				default:
 					break;
 				}
diff --git a/multimaster.h b/multimaster.h
@@ -165,6 +165,7 @@ typedef struct
 	nodemask_t reconnectMask; 	       /* Mask of nodes connection to which has to be reestablished by sender */
 
 	bool   localTablesHashLoaded;      /* Whether data from local_tables table is loaded in shared memory hash table */
+	int    inject2PCError;             /* Simulate error during 2PC commit at this node */
     int    nLiveNodes;                 /* Number of active nodes */
     int    nAllNodes;                  /* Total numbber of nodes */
     int    nReceivers;                 /* Number of initialized logical receivers (used to determine moment when Mtm intialization is completed */
