pg_rollback_prepared_xacts() function

kelvich · kelvich · commit 444c1a389f04 · 2016-07-27T20:03:20.000+03:00
diff --git a/src/backend/access/transam/twophase.c b/src/backend/access/transam/twophase.c
@@ -2267,3 +2267,30 @@ GetLockedGlobalTransactionId(void)
 {
 	return MyLockedGxact ? MyLockedGxact->gid : NULL;
 }
+
+int
+FinishAllPreparedTransactions(bool isCommit)
+{
+	int i, count = 0;
+
+	for (i = 0; i < TwoPhaseState->numPrepXacts; i++)
+	{
+		GlobalTransaction gxact = TwoPhaseState->prepXacts[i];
+
+		if (gxact->valid)
+		{
+			FinishPreparedTransaction(gxact->gid, isCommit);
+			count++;
+		}
+	}
+
+	return count;
+}
+
+Datum
+pg_rollback_prepared_xacts(PG_FUNCTION_ARGS)
+{
+	int count;
+	count = FinishAllPreparedTransactions(0);
+	PG_RETURN_INT32(count);
+}
diff --git a/src/include/access/twophase.h b/src/include/access/twophase.h
@@ -61,4 +61,6 @@ extern void FinishPreparedTransaction(const char *gid, bool isCommit);
 
 extern const char *GetLockedGlobalTransactionId(void);
 
+extern int FinishAllPreparedTransactions(bool isCommit);
+
 #endif   /* TWOPHASE_H */
diff --git a/src/include/catalog/pg_proc.h b/src/include/catalog/pg_proc.h
@@ -3052,6 +3052,8 @@ DATA(insert OID = 2561 (  pg_blocking_pids PGNSP PGUID 12 1 0 0 0 f f f f t f v
 DESCR("get array of PIDs of sessions blocking specified backend PID");
 DATA(insert OID = 1065 (  pg_prepared_xact PGNSP PGUID 12 1 1000 0 0 f f f f t t v s 0 0 2249 "" "{28,25,1184,26,26}" "{o,o,o,o,o}" "{transaction,gid,prepared,ownerid,dbid}" _null_ _null_ pg_prepared_xact _null_ _null_ _null_ ));
 DESCR("view two-phase transactions");
+DATA(insert OID = 4014 (   pg_rollback_prepared_xacts PGNSP PGUID 12 1 0 0 0 f f f f t f s u 0 0 23 "" _null_ _null_ _null_ _null_ _null_ pg_rollback_prepared_xacts _null_ _null_ _null_ ));
+DESCR("rollback all two-phase transactions");
 DATA(insert OID = 3819 (  pg_get_multixact_members PGNSP PGUID 12 1 1000 0 0 f f f f t t v s 1 0 2249 "28" "{28,28,25}" "{i,o,o}" "{multixid,xid,mode}" _null_ _null_ pg_get_multixact_members _null_ _null_ _null_ ));
 DESCR("view members of a multixactid");
 
