From d1c49bdf7b0b8d8956d47e3206d02c4594fb3661 Mon Sep 17 00:00:00 2001
From: Feng Wang <wang6922@126.com>
Date: Mon, 23 Dec 2024 16:59:45 +0800
Subject: [PATCH] fix thread naming

---
 .../xiaomi_home/miot/miot_mips.py             | 20 ++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/custom_components/xiaomi_home/miot/miot_mips.py b/custom_components/xiaomi_home/miot/miot_mips.py
index 98d19dc..626b020 100644
--- a/custom_components/xiaomi_home/miot/miot_mips.py
+++ b/custom_components/xiaomi_home/miot/miot_mips.py
@@ -227,7 +227,7 @@ class _MipsClient(ABC):
     _event_connect: asyncio.Event
     _event_disconnect: asyncio.Event
     _internal_loop: asyncio.AbstractEventLoop
-    _mips_thread: threading.Thread
+    _mips_thread: threading.Thread | None = None
     _mips_reconnect_tag: bool
     _mips_reconnect_interval: float
     _mips_reconnect_timer: Optional[asyncio.TimerHandle]
@@ -302,20 +302,23 @@ class _MipsClient(ABC):
         """
         return self._mqtt and self._mqtt.is_connected()
 
-    @final
-    def connect(self) -> None:
+    def connect(self, thread_name: Optional[str] = None) -> None:
         """mips connect."""
         # Start mips thread
+        if self._mips_thread:
+            return
         self._internal_loop = asyncio.new_event_loop()
         self._mips_thread = threading.Thread(target=self.__mips_loop_thread)
         self._mips_thread.daemon = True
-        self._mips_thread.name = self._client_id
+        self._mips_thread.name = \
+            self._client_id if thread_name is None else thread_name
         self._mips_thread.start()
 
     @final
     def close(self) -> None:
         self._internal_loop.call_soon_threadsafe(self.__mips_close)
-        self._mips_thread.join()
+        if self._mips_thread:
+            self._mips_thread.join()
         self._internal_loop.close()
 
         self._logger = None
@@ -1066,8 +1069,6 @@ class MipsLocalClient(_MipsClient):
         super().__init__(
             client_id=did, host=host, port=port,
             ca_file=ca_file, cert_file=cert_file, key_file=key_file, loop=loop)
-        # MIPS local thread name use group_id
-        self._mips_thread.name = self._group_id
 
     @property
     def group_id(self) -> str:
@@ -1085,6 +1086,11 @@ class MipsLocalClient(_MipsClient):
         if self._logger:
             self._logger.error(f'{self._home_name}, '+msg, *args, **kwargs)
 
+    @final
+    def connect(self, thread_name: Optional[str] = None) -> None:
+        # MIPS local thread name use group_id
+        super().connect(self._group_id)
+
     @final
     def disconnect(self) -> None:
         super().disconnect()