jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-17 18:27:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Speed up bond setup with gather (#48454)

Browse Source
This commit is contained in:
J. Nick Koston 2021-03-28 20:58:48 -10:00 committed by GitHub
parent 14ef0531f0
commit 2796a58ead
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
homeassistant/components/bond/utils.py
View File

@ -1,15 +1,18 @@
"""Reusable utilities for the Bond component.""" """Reusable utilities for the Bond component."""
from __future__ import annotations from __future__ import annotations
import asyncio
import logging import logging
from typing import Any, cast from typing import Any, cast
from aiohttp import ClientResponseError from aiohttp import ClientResponseError
from bond_api import Action, Bond from bond_api import Action, Bond
from homeassistant.util.async_ import gather_with_concurrency
from .const import BRIDGE_MAKE from .const import BRIDGE_MAKE
MAX_REQUESTS = 6
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -114,15 +117,25 @@ class BondHub:
# Fetch all available devices using Bond API. # Fetch all available devices using Bond API.
device_ids = await self.bond.devices() device_ids = await self.bond.devices()
self._devices = [] self._devices = []
setup_device_ids = []
tasks = []
for idx, device_id in enumerate(device_ids): for idx, device_id in enumerate(device_ids):
if max_devices is not None and idx >= max_devices: if max_devices is not None and idx >= max_devices:
break break
setup_device_ids.append(device_id)
device, props = await asyncio.gather( tasks.extend(
self.bond.device(device_id), self.bond.device_properties(device_id) [self.bond.device(device_id), self.bond.device_properties(device_id)]
) )
self._devices.append(BondDevice(device_id, device, props)) responses = await gather_with_concurrency(MAX_REQUESTS, *tasks)
response_idx = 0
for device_id in setup_device_ids:
self._devices.append(
BondDevice(
device_id, responses[response_idx], responses[response_idx + 1]
)
)
response_idx += 2
_LOGGER.debug("Discovered Bond devices: %s", self._devices) _LOGGER.debug("Discovered Bond devices: %s", self._devices)
try: try: