feedc0de/homeassistant-core
1
0
Fork 0
forked from home-assistant/core
Code Pull Requests Activity

Fix race when waiting for MQTT ACK (#39193)

Browse Source
This commit is contained in:
Erik Montnemery
2020-08-24 11:13:12 +02:00
committed by GitHub
parent 71acb2c665
commit 3df67ff9e1
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -939,10 +939,9 @@ class MQTT:
self.hass.add_job(self._mqtt_handle_mid, mid)
async def _mqtt_handle_mid(self, mid) -> None:
if mid in self._pending_operations:
if mid not in self._pending_operations:
self._pending_operations[mid] = asyncio.Event()
self._pending_operations[mid].set()
else:
_LOGGER.warning("Unknown mid %d", mid)
def _mqtt_on_disconnect(self, _mqttc, _userdata, result_code: int) -> None:
"""Disconnected callback."""
@@ -957,6 +956,7 @@ class MQTT:
async def _wait_for_mid(self, mid):
"""Wait for ACK from broker."""
if mid not in self._pending_operations:
self._pending_operations[mid] = asyncio.Event()
try:
await asyncio.wait_for(self._pending_operations[mid].wait(), TIMEOUT_ACK)