Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Performance: caching region collection (#7120)

Browse Source
This commit is contained in:
Bert Blommers 2023-12-13 10:05:40 -01:00 committed by GitHub
parent 44044ddf8f
commit e422e95e5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
moto/core/base_backend.py
View File

@ -190,6 +190,8 @@ class AccountSpecificBackend(Dict[str, SERVICE_BACKEND]):
account_specific_backend[region: str] = backend: BaseBackend account_specific_backend[region: str] = backend: BaseBackend
""" """
session = Session()
def __init__( def __init__(
self, self,
service_name: str, service_name: str,
@ -198,18 +200,24 @@ class AccountSpecificBackend(Dict[str, SERVICE_BACKEND]):
use_boto3_regions: bool, use_boto3_regions: bool,
additional_regions: Optional[List[str]], additional_regions: Optional[List[str]],
): ):
self._id = str(uuid4())
self.service_name = service_name self.service_name = service_name
self.account_id = account_id self.account_id = account_id
self.backend = backend self.backend = backend
self.regions = [] self.regions = []
if use_boto3_regions: if use_boto3_regions:
sess = Session() self.regions.extend(self._generate_regions(service_name))
for partition in sess.get_available_partitions():
self.regions.extend(
sess.get_available_regions(service_name, partition_name=partition)
)
self.regions.extend(additional_regions or []) self.regions.extend(additional_regions or [])
self._id = str(uuid4())
@lru_cache()
def _generate_regions(self, service_name: str) -> List[str]:
regions = []
for partition in AccountSpecificBackend.session.get_available_partitions():
partition_regions = AccountSpecificBackend.session.get_available_regions(
service_name, partition_name=partition
)
regions.extend(partition_regions)
return regions
def __hash__(self) -> int: # type: ignore[override] def __hash__(self) -> int: # type: ignore[override]
return hash(self._id) return hash(self._id)