Add on_remote_server_up to ReplicationDataHandler

synapse/replication/tcp/client.py

@@ -119,3 +119,6 @@ class ReplicationDataHandler:
 
     async def on_position(self, stream_name: str, token: int):
         self.store.process_replication_rows(stream_name, token, [])
+
+    def on_remote_server_up(self, server: str):
+        """Called when get a new REMOTE_SERVER_UP command."""

synapse/replication/tcp/handler.py

@@ -165,6 +165,7 @@ class ReplicationCommandHandler:
 
     async def on_REMOTE_SERVER_UP(self, cmd: RemoteServerUpCommand):
         """"Called when get a new REMOTE_SERVER_UP command."""
+        self._replication_data_handler.on_remote_server_up(cmd.data)
 
     def get_currently_syncing_users(self):
         """Get the list of currently syncing users (if any). This is called