restart walsender on counterpart disable too

Author: kelvich

src/include/pglogical_output.h

@@ -48,6 +48,7 @@ typedef struct MtmDecoderPrivate
 	bool	is_recovery;
 	bool	recovery_done;
 	int		recovery_count;
+	int		counterpart_disable_count;
 	int64	session_id;
 	MtmConfig *cfg;
 } MtmDecoderPrivate;

src/pglogical_output.c

@@ -478,6 +478,17 @@ pg_decode_caughtup(LogicalDecodingContext *ctx)
 		abort();					/* keep the compiler quiet */
 	}
 
+
+	if (hooks_data->counterpart_disable_count != MtmGetNodeDisableCount(MtmReplicationNodeId))
+	{
+		mtm_log(LOG, "exiting due to disabled counterpart");
+		if (whereToSendOutput == DestRemote)
+			whereToSendOutput = DestNone;
+
+		proc_exit(0);
+		abort();					/* keep the compiler quiet */
+	}
+
 	/*
 	 * MtmOutputPluginPrepareWrite send some bytes to downstream,
 	 * so we must avoid calling it in normal (non-recovery) situation.

src/pglogical_proto.c

@@ -722,6 +722,7 @@ MtmReplicationStartupHook(struct PGLogicalStartupHookArgs* args)
 	hooks_data->is_recovery = false;
 	hooks_data->cfg = MtmLoadConfig();
 	hooks_data->recovery_count = MtmGetRecoveryCount();
+	hooks_data->counterpart_disable_count = MtmGetNodeDisableCount(MtmReplicationNodeId);
 
 	foreach(param, args->in_params)
 	{