Set statement timestamp in apply worker

This ensures that triggers can see an up-to-date timestamp.

Reported-by: Konstantin Evteev <[email protected]>

1 files changed, 3 insertions, 0 deletions

diff --git a/src/backend/replication/logical/worker.c b/src/backend/replication/logical/worker.c
index 97d2dff0dd..21a4fea821 100644
--- a/src/backend/replication/logical/worker.c
+++ b/src/backend/replication/logical/worker.c
@@ -157,12 +157,15 @@ ensure_transaction(void)
 {
 	if (IsTransactionState())
 	{
+		SetCurrentStatementStartTimestamp();
+
 		if (CurrentMemoryContext != ApplyMessageContext)
 			MemoryContextSwitchTo(ApplyMessageContext);
 
 		return false;
 	}
 
+	SetCurrentStatementStartTimestamp();
 	StartTransactionCommand();
 
 	maybe_reread_subscription();