From 2a6db53dc8ab148476846d3b401861d36d7b7151 Mon Sep 17 00:00:00 2001
From: linuswck <linuswck@m-labs.hk>
Date: Tue, 21 Jan 2025 11:27:48 +0800
Subject: [PATCH] driver: wait for the read_response task to complete before
 connection retry

---
 pykirdy/driver/kirdy.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pykirdy/driver/kirdy.py b/pykirdy/driver/kirdy.py
index c0ed743..1feaee0 100644
--- 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