Save a tast reference
Important: Save a reference of the created task, to avoid a task disappearing mid-execution. The event loop only keeps weak references to tasks. A task that isn’t referenced elsewhere may get garbage collected at any time, even before it’s done. For reliable “fire-and-forget” background tasks, gather them in a collection
This commit is contained in:
@@ -327,6 +327,7 @@ class SolarmanV5(SolarmanBase):
|
|||||||
self.sensor_list = 0
|
self.sensor_list = 0
|
||||||
self.mb_regs = [{'addr': 0x3000, 'len': 48},
|
self.mb_regs = [{'addr': 0x3000, 'len': 48},
|
||||||
{'addr': 0x2000, 'len': 96}]
|
{'addr': 0x2000, 'len': 96}]
|
||||||
|
self.background_tasks = set()
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Our puplic methods
|
Our puplic methods
|
||||||
@@ -339,6 +340,7 @@ class SolarmanV5(SolarmanBase):
|
|||||||
self.inverter = None
|
self.inverter = None
|
||||||
self.switch.clear()
|
self.switch.clear()
|
||||||
self.log_lvl.clear()
|
self.log_lvl.clear()
|
||||||
|
self.background_tasks.clear()
|
||||||
super().close()
|
super().close()
|
||||||
|
|
||||||
def send_start_cmd(self, snr: int, host: str,
|
def send_start_cmd(self, snr: int, host: str,
|
||||||
@@ -690,8 +692,10 @@ class SolarmanV5(SolarmanBase):
|
|||||||
self.__forward_msg()
|
self.__forward_msg()
|
||||||
|
|
||||||
def publish_mqtt(self, key, data): # pragma: no cover
|
def publish_mqtt(self, key, data): # pragma: no cover
|
||||||
asyncio.ensure_future(
|
task = asyncio.ensure_future(
|
||||||
Proxy.mqtt.publish(key, data))
|
Proxy.mqtt.publish(key, data))
|
||||||
|
self.background_tasks.add(task)
|
||||||
|
task.add_done_callback(self.background_tasks.discard)
|
||||||
|
|
||||||
def get_cmd_rsp_log_lvl(self) -> int:
|
def get_cmd_rsp_log_lvl(self) -> int:
|
||||||
ftype = self.ifc.rx_peek()[self.header_len]
|
ftype = self.ifc.rx_peek()[self.header_len]
|
||||||
|
|||||||
@@ -39,6 +39,7 @@ class InverterBase(InverterIfc, Proxy):
|
|||||||
self.use_emulation = False
|
self.use_emulation = False
|
||||||
self.__ha_restarts = -1
|
self.__ha_restarts = -1
|
||||||
self.remote = StreamPtr(None)
|
self.remote = StreamPtr(None)
|
||||||
|
self.background_tasks = set()
|
||||||
ifc = AsyncStreamServer(reader, writer,
|
ifc = AsyncStreamServer(reader, writer,
|
||||||
self.async_publ_mqtt,
|
self.async_publ_mqtt,
|
||||||
self.create_remote,
|
self.create_remote,
|
||||||
@@ -73,6 +74,7 @@ class InverterBase(InverterIfc, Proxy):
|
|||||||
if self.remote.ifc:
|
if self.remote.ifc:
|
||||||
self.remote.ifc.close()
|
self.remote.ifc.close()
|
||||||
self.remote.ifc = None
|
self.remote.ifc = None
|
||||||
|
self.background_tasks.clear()
|
||||||
|
|
||||||
async def disc(self, shutdown_started=False) -> None:
|
async def disc(self, shutdown_started=False) -> None:
|
||||||
if self.remote.stream:
|
if self.remote.stream:
|
||||||
@@ -137,7 +139,10 @@ class InverterBase(InverterIfc, Proxy):
|
|||||||
logging.info(f'[{self.remote.stream.node_id}:'
|
logging.info(f'[{self.remote.stream.node_id}:'
|
||||||
f'{self.remote.stream.conn_no}] '
|
f'{self.remote.stream.conn_no}] '
|
||||||
f'Connected to {addr}')
|
f'Connected to {addr}')
|
||||||
asyncio.create_task(self.remote.ifc.client_loop(addr))
|
task = asyncio.create_task(
|
||||||
|
self.remote.ifc.client_loop(addr))
|
||||||
|
self.background_tasks.add(task)
|
||||||
|
task.add_done_callback(self.background_tasks.discard)
|
||||||
|
|
||||||
except (ConnectionRefusedError,
|
except (ConnectionRefusedError,
|
||||||
TimeoutError,
|
TimeoutError,
|
||||||
|
|||||||
Reference in New Issue
Block a user