Add small speed improvement when unloading RainMachine (#28759)

2019-11-13 14:27:42 -07:00 · 2019-11-13 14:27:42 -07:00 · 3f2c344e4f
parent fae8cd48c4
commit 3f2c344e4f
1 changed files with 6 additions and 2 deletions
--- a/homeassistant/components/rainmachine/init.py
+++ b/homeassistant/components/rainmachine/init.py
@ -351,8 +351,12 @@ async def async_unload_entry(hass, config_entry):
    remove_listener = hass.data[DOMAIN][DATA_LISTENER].pop(config_entry.entry_id)
    remove_listener()

-    for component in ("binary_sensor", "sensor", "switch"):
-        await hass.config_entries.async_forward_entry_unload(config_entry, component)
+    tasks = [
+        hass.config_entries.async_forward_entry_unload(config_entry, component)
+        for component in ("binary_sensor", "sensor", "switch")
+    ]
+
+    await asyncio.gather(*tasks)

    return True