async.lock around update cache

osdu_az/partition/partition_info.py

@@ -2,7 +2,6 @@ import logging
 from typing import Optional
 from datetime import datetime
 
-from osdu_az import conf
 from osdu_az.identity.az_identity import AzureIdentity
 
 

osdu_az/partition/partition_service.py

 import logging
 from os import environ
-
+import asyncio
 from osdu_az.opendes.fake_partition_client import FakePartitionClient
 from osdu_az.opendes.partition_client import PartitionClient
 from osdu_az.partition.partition_info import PartitionInfo
@@ -26,11 +26,15 @@ PARTITION_ID = 'id'
 class PartitionService:
     partition_client = None
     scope = ''
+    partition_cache_lock = asyncio.Lock()
 
     @staticmethod
     async def get_partition(data_partition_id: str) -> PartitionInfo:
         partition_info = PartitionsCache.get(data_partition_id)
-        if not partition_info:
+        if partition_info is None:
+            async with PartitionService.partition_cache_lock:
+                partition_info = PartitionsCache.get(data_partition_id)
+                if partition_info is None:
                     partition_info = await PartitionService._partition_client().get_partition(data_partition_id)
                     if partition_info:
                         _LOGGER.info(f'PartitionService fetching partition info for "{data_partition_id}"')