bring xact events back

kelvich · kelvich · commit 41a090d548f2 · 2017-11-06T18:30:19.000+03:00
diff --git a/src/backend/access/transam/twophase.c b/src/backend/access/transam/twophase.c
@@ -1509,6 +1509,12 @@ FinishPreparedTransaction(const char *gid, bool isCommit)
 	LWLockAcquire(TwoPhaseStateLock, LW_EXCLUSIVE);
 	RemoveGXact(gxact);
 	LWLockRelease(TwoPhaseStateLock);
+
+	if (isCommit)
+		CallXactCallbacks(XACT_EVENT_COMMIT_PREPARED);
+	else
+		CallXactCallbacks(XACT_EVENT_ABORT_PREPARED);
+
 	MyLockedGxact = NULL;
 
 	pfree(buf);
@@ -2408,3 +2414,13 @@ PrepareRedoRemove(TransactionId xid, bool giveWarning)
 
 	return;
 }
+
+
+/*
+ * Return identified of current global transaction
+ */
+const char*
+GetLockedGlobalTransactionId(void)
+{
+	return MyLockedGxact ? MyLockedGxact->gid : NULL;
+}
diff --git a/src/backend/access/transam/xact.c b/src/backend/access/transam/xact.c
@@ -305,7 +305,6 @@ static void AtCommit_Memory(void);
 static void AtStart_Cache(void);
 static void AtStart_Memory(void);
 static void AtStart_ResourceOwner(void);
-static void CallXactCallbacks(XactEvent event);
 static void CallSubXactCallbacks(SubXactEvent event,
 					 SubTransactionId mySubid,
 					 SubTransactionId parentSubid);
@@ -1928,6 +1927,7 @@ StartTransaction(void)
 	 */
 	s->state = TRANS_INPROGRESS;
 
+	CallXactCallbacks(XACT_EVENT_START);
 	ShowTransactionState("StartTransaction");
 }
 
@@ -3339,7 +3339,7 @@ UnregisterXactCallback(XactCallback callback, void *arg)
 	}
 }
 
-static void
+void
 CallXactCallbacks(XactEvent event)
 {
 	XactCallbackItem *item;
diff --git a/src/include/access/twophase.h b/src/include/access/twophase.h
@@ -57,4 +57,7 @@ extern void PrepareRedoAdd(char *buf, XLogRecPtr start_lsn,
 			   XLogRecPtr end_lsn);
 extern void PrepareRedoRemove(TransactionId xid, bool giveWarning);
 extern void restoreTwoPhaseData(void);
+
+extern const char *GetLockedGlobalTransactionId(void);
+
 #endif							/* TWOPHASE_H */
diff --git a/src/include/access/xact.h b/src/include/access/xact.h
@@ -97,14 +97,17 @@ extern int	MyXactFlags;
  */
 typedef enum
 {
+	XACT_EVENT_START,
 	XACT_EVENT_COMMIT,
 	XACT_EVENT_PARALLEL_COMMIT,
 	XACT_EVENT_ABORT,
 	XACT_EVENT_PARALLEL_ABORT,
 	XACT_EVENT_PREPARE,
 	XACT_EVENT_PRE_COMMIT,
 	XACT_EVENT_PARALLEL_PRE_COMMIT,
-	XACT_EVENT_PRE_PREPARE
+	XACT_EVENT_PRE_PREPARE,
+	XACT_EVENT_COMMIT_PREPARED,
+	XACT_EVENT_ABORT_PREPARED
 } XactEvent;
 
 typedef void (*XactCallback) (XactEvent event, void *arg);
@@ -403,5 +406,6 @@ extern void ParseAbortRecord(uint8 info, xl_xact_abort *xlrec, xl_xact_parsed_ab
 extern void EnterParallelMode(void);
 extern void ExitParallelMode(void);
 extern bool IsInParallelMode(void);
+extern void CallXactCallbacks(XactEvent event);
 
 #endif							/* XACT_H */

Original file line number	Diff line number	Diff line change
`@@ -305,7 +305,6 @@ static void AtCommit_Memory(void);`
`305`	`305`	`static void AtStart_Cache(void);`
`306`	`306`	`static void AtStart_Memory(void);`
`307`	`307`	`static void AtStart_ResourceOwner(void);`
`308`		`-static void CallXactCallbacks(XactEvent event);`
`309`	`308`	`static void CallSubXactCallbacks(SubXactEvent event,`
`310`	`309`	`SubTransactionId mySubid,`
`311`	`310`	`SubTransactionId parentSubid);`
`@@ -1928,6 +1927,7 @@ StartTransaction(void)`
`1928`	`1927`	`*/`
`1929`	`1928`	`s->state = TRANS_INPROGRESS;`
`1930`	`1929`
	`1930`	`+ CallXactCallbacks(XACT_EVENT_START);`
`1931`	`1931`	`ShowTransactionState("StartTransaction");`
`1932`	`1932`	`}`
`1933`	`1933`
`@@ -3339,7 +3339,7 @@ UnregisterXactCallback(XactCallback callback, void *arg)`
`3339`	`3339`	`}`
`3340`	`3340`	`}`
`3341`	`3341`
`3342`		`-static void`
	`3342`	`+void`
`3343`	`3343`	`CallXactCallbacks(XactEvent event)`
`3344`	`3344`	`{`
`3345`	`3345`	`XactCallbackItem *item;`