home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-03 04:35:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Minor cleanup of MQTT ACK handling (#39217)

Browse Source
This commit is contained in:
Erik Montnemery
2020-08-24 17:28:07 +02:00
committed by GitHub
parent bee6d87e7a
commit b1c0d8fb6c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/mqtt/__init__.py
View File

@@ -938,7 +938,10 @@ class MQTT:
"""Publish / Subscribe / Unsubscribe callback."""
self.hass.add_job(self._mqtt_handle_mid, mid)
async def _mqtt_handle_mid(self, mid) -> None:
@callback
def _mqtt_handle_mid(self, mid) -> None:
# Create the mid event if not created, either _mqtt_handle_mid or _wait_for_mid
# may be executed first.
if mid not in self._pending_operations:
self._pending_operations[mid] = asyncio.Event()
self._pending_operations[mid].set()
@@ -956,6 +959,8 @@ class MQTT:
async def _wait_for_mid(self, mid):
"""Wait for ACK from broker."""
# Create the mid event if not created, either _mqtt_handle_mid or _wait_for_mid
# may be executed first.
if mid not in self._pending_operations:
self._pending_operations[mid] = asyncio.Event()
try: