[PGPRO-5845] OutputPluginCallbacks::begin_prepare_cb

Andrei Krichinin · Andrei Krichinin · commit db9f74221109 · 2022-03-18T15:10:01.000+05:00
diff --git a/src/pglogical_output.c b/src/pglogical_output.c
@@ -69,6 +69,7 @@ static void pg_decode_begin_txn(LogicalDecodingContext *ctx,
 static void pg_decode_commit_txn(LogicalDecodingContext *ctx,
 					 ReorderBufferTXN *txn, XLogRecPtr commit_lsn);
 
+static void pg_decode_begin_prepare_txn(LogicalDecodingContext *ctx, ReorderBufferTXN *txn);
 static void pg_decode_prepare_txn(LogicalDecodingContext *ctx, ReorderBufferTXN *txn,
 					  XLogRecPtr lsn);
 static void pg_decode_commit_prepared_txn(LogicalDecodingContext *ctx, ReorderBufferTXN *txn,
@@ -135,6 +136,7 @@ _PG_output_plugin_init(OutputPluginCallbacks *cb)
 	cb->abort_cb = pg_decode_abort_txn;
 
 	cb->filter_prepare_cb = pg_filter_prepare;
+	cb->begin_prepare_cb = pg_decode_begin_prepare_txn;
 	cb->prepare_cb = pg_decode_prepare_txn;
 	cb->commit_prepared_cb = pg_decode_commit_prepared_txn;
 	cb->rollback_prepared_cb = pg_decode_abort_prepared_txn;
@@ -482,6 +484,46 @@ pg_decode_commit_txn(LogicalDecodingContext *ctx, ReorderBufferTXN *txn,
 	}
 }
 
+void
+pg_decode_begin_prepare_txn(LogicalDecodingContext *ctx, ReorderBufferTXN *txn)
+{
+	PGLogicalOutputData *data = (PGLogicalOutputData *) ctx->output_plugin_private;
+	bool		send_replication_origin = data->forward_changeset_origins;
+
+	if (!startup_message_sent)
+		send_startup_message(ctx, data, false /* can't be last message */ );
+
+	/* If the record didn't originate locally, send origin info */
+	send_replication_origin &= txn->origin_id != InvalidRepOriginId;
+
+	if (data->api)
+	{
+		MtmOutputPluginPrepareWrite(ctx, !send_replication_origin, true);
+		data->api->write_begin(ctx->out, data, txn);
+
+		if (send_replication_origin)
+		{
+			char	   *origin;
+
+			/* Message boundary */
+			MtmOutputPluginWrite(ctx, false, false);
+			MtmOutputPluginPrepareWrite(ctx, true, false);
+
+			/*
+			 * XXX: which behaviour we want here?
+			 *
+			 * Alternatives: - don't send origin message if origin name not
+			 * found (that's what we do now) - throw error - that will break
+			 * replication, not good - send some special "unknown" origin
+			 */
+			if (data->api->write_origin &&
+				replorigin_by_oid(txn->origin_id, true, &origin))
+				data->api->write_origin(ctx->out, origin, txn->origin_lsn);
+		}
+		MtmOutputPluginWrite(ctx, true, false);
+	}
+}
+
 void
 pg_decode_prepare_txn(LogicalDecodingContext *ctx, ReorderBufferTXN *txn,
 					  XLogRecPtr lsn)
diff --git a/src/syncpoint.c b/src/syncpoint.c
@@ -381,20 +381,20 @@ GetRecoveryHorizon(int sender_node_id)
 
 	sql = psprintf("select mtm.get_recovery_horizon(%d)", sender_node_id);
 	rc = SPI_execute(sql, true, 0);
-	mtm_log(LOG, "----> GetRecoveryHorizont: sql '%s'", sql);
+	mtm_log(LOG, "----> GetRecoveryHorizon: sql '%s'", sql);
 
 	if (rc != SPI_OK_SELECT && SPI_processed != 1)
 		mtm_log(ERROR, "mtm.get_recovery_horizon failed: rc=%d, SPI_processed=" UINT64_FORMAT,
 				rc, SPI_processed);
 
 	tup = SPI_tuptable->vals[0];
-	mtm_log(LOG, "----> GetRecoveryHorizont: %d", SPI_tuptable->tupdesc->natts);
+	mtm_log(LOG, "----> GetRecoveryHorizon: %d", SPI_tuptable->tupdesc->natts);
 
 	type = SPI_gettype(SPI_tuptable->tupdesc, 1);
-	mtm_log(LOG, "----> GetRecoveryHorizont: type %s", type ? type: "NULL");
+	mtm_log(LOG, "----> GetRecoveryHorizon: type %s", type ? type: "NULL");
 
 	value = SPI_getvalue(tup, SPI_tuptable->tupdesc, 1);
-	mtm_log(LOG, "----> GetRecoveryHorizont: value %s", value ? value: "NULL");
+	mtm_log(LOG, "----> GetRecoveryHorizon: value %s", value ? value: "NULL");
 	
 	horizon = DatumGetLSN(SPI_getbinval(tup, SPI_tuptable->tupdesc, 1, &isnull));
 	if (isnull)
