driver: wait for the read_response task to complete before connection retry

2025-01-21 11:27:48 +08:00 · 2025-01-21 11:27:48 +08:00 · 2a6db53dc8
commit 2a6db53dc8
parent 7dd551201e
1 changed files with 2 additions and 0 deletions
--- a/pykirdy/driver/kirdy.py
+++ b/pykirdy/driver/kirdy.py
@ -773,6 +773,7 @@ class Kirdy:
    async def _handler(self):
        try:
            self._state = State.disconnected
+            self.read_response_task = None
            first_con = True
            task = None
            while True:
@ -811,6 +812,7 @@ class Kirdy:
                        logging.warning("Connection to Kirdy is dropped.")
                        first_con = True
                        self.read_response_task.cancel()
+                        await self.read_response_task

                        # State Transition
                        self._state = State.disconnected