From e8b514f4321677a653a78e233890a46de26d7a1f Mon Sep 17 00:00:00 2001 From: Andrey Vertiprahov Date: Tue, 19 Jan 2021 20:00:31 +0500 Subject: [PATCH] Move get_leader_channel to try block. For catch connection exception. --- core/liftbridge/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/liftbridge/base.py b/core/liftbridge/base.py index 76446fb303..1f096f96a9 100644 --- a/core/liftbridge/base.py +++ b/core/liftbridge/base.py @@ -315,6 +315,7 @@ class LiftBridgeClient(object): await self.close_channel(broker) async def __refresh_leaders(self): + logger.info("Refresh leaders") await self.fetch_metadata(wait_for_stream=True) def __reset_leaders(self): @@ -666,8 +667,8 @@ class LiftBridgeClient(object): async def set_cursor(self, stream: str, partition: int, cursor_id: str, offset: int) -> None: with rpc_error(): while True: - channel = await self.get_leader_channel(CURSOR_STREAM, 0) try: + channel = await self.get_leader_channel(CURSOR_STREAM, 0) await channel.SetCursor( SetCursorRequest( stream=stream, -- GitLab