POST_PREPARE xact event and expose gid during PRE_PREPARE.

arssher · arssher · commit 1896708751ae · 2018-10-19T08:46:36.000+03:00
diff --git a/src/backend/access/transam/xact.c b/src/backend/access/transam/xact.c
@@ -2463,6 +2463,8 @@ PrepareTransaction(void)
 	s->state = TRANS_DEFAULT;
 
 	RESUME_INTERRUPTS();
+
+	CallXactCallbacks(XACT_EVENT_POST_PREPARE);
 }
 
 
@@ -5848,3 +5850,10 @@ xact_redo(XLogReaderState *record)
 	else
 		elog(PANIC, "xact_redo: unknown op code %u", info);
 }
+
+/* Get gid for ongoing PREPARE */
+const char *
+GetPrepareGid(void)
+{
+	return prepareGID;
+}
diff --git a/src/include/access/xact.h b/src/include/access/xact.h
@@ -111,7 +111,8 @@ typedef enum
 	XACT_EVENT_PREPARE,
 	XACT_EVENT_PRE_COMMIT,
 	XACT_EVENT_PARALLEL_PRE_COMMIT,
-	XACT_EVENT_PRE_PREPARE
+	XACT_EVENT_PRE_PREPARE,
+	XACT_EVENT_POST_PREPARE
 } XactEvent;
 
 typedef void (*XactCallback) (XactEvent event, void *arg);
@@ -399,6 +400,7 @@ extern void RegisterXactCallback(XactCallback callback, void *arg);
 extern void UnregisterXactCallback(XactCallback callback, void *arg);
 extern void RegisterSubXactCallback(SubXactCallback callback, void *arg);
 extern void UnregisterSubXactCallback(SubXactCallback callback, void *arg);
+extern const char *GetPrepareGid(void);
 
 extern int	xactGetCommittedChildren(TransactionId **ptr);
 
