Skip to content

Commit 5126415

Browse files
author
Andrei Neagu
committed
fixed issue with namespace not being capitalised
1 parent 27ff515 commit 5126415

File tree

3 files changed

+17
-10
lines changed

3 files changed

+17
-10
lines changed

packages/service-library/src/servicelib/long_running_tasks/_redis_store.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,17 +23,21 @@ def _load_from_redis_hash(data: dict[str, str]) -> dict[str, Any]:
2323
return {k: json_loads(v) for k, v in data.items()}
2424

2525

26+
def to_redis_namespace(namespace: LRTNamespace) -> str:
27+
return namespace.upper()
28+
29+
2630
class RedisStore:
2731
def __init__(self, redis_settings: RedisSettings, namespace: LRTNamespace):
2832
self.redis_settings = redis_settings
29-
self.namespace: LRTNamespace = namespace.upper()
33+
self.redis_namespace: str = to_redis_namespace(namespace)
3034

3135
self._client: RedisClientSDK | None = None
3236

3337
async def setup(self) -> None:
3438
self._client = RedisClientSDK(
3539
self.redis_settings.build_redis_dsn(RedisDatabase.LONG_RUNNING_TASKS),
36-
client_name=f"long_running_tasks_store_{self.namespace}",
40+
client_name=f"long_running_tasks_store_{self.redis_namespace}",
3741
)
3842
await self._client.setup()
3943

@@ -47,10 +51,10 @@ def _redis(self) -> aioredis.Redis:
4751
return self._client.redis
4852

4953
def _get_redis_key_task_data_match(self) -> str:
50-
return f"{self.namespace}:{_STORE_TYPE_TASK_DATA}*"
54+
return f"{self.redis_namespace}:{_STORE_TYPE_TASK_DATA}*"
5155

5256
def _get_redis_task_data_key(self, task_id: TaskId) -> str:
53-
return f"{self.namespace}:{_STORE_TYPE_TASK_DATA}:{task_id}"
57+
return f"{self.redis_namespace}:{_STORE_TYPE_TASK_DATA}:{task_id}"
5458

5559
async def get_task_data(self, task_id: TaskId) -> TaskData | None:
5660
result: dict[str, Any] = await handle_redis_returns_union_types(

packages/service-library/src/servicelib/long_running_tasks/long_running_client_helper.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
from ..logging_utils import log_context
77
from ..redis._client import RedisClientSDK
8+
from ._redis_store import to_redis_namespace
89
from .models import LRTNamespace
910

1011
_logger = logging.getLogger(__name__)
@@ -32,15 +33,16 @@ def _redis(self) -> aioredis.Redis:
3233
assert self._client # nosec
3334
return self._client.redis
3435

35-
async def cleanup(self, lrt_namespace: LRTNamespace) -> None:
36+
async def cleanup(self, namespace: LRTNamespace) -> None:
3637
"""removes Redis keys associated to the LRTNamespace if they exist"""
38+
redis_namespace = to_redis_namespace(namespace)
3739
keys_to_remove: list[str] = [
38-
x async for x in self._redis.scan_iter(f"{lrt_namespace}*")
40+
x async for x in self._redis.scan_iter(f"{redis_namespace}*")
3941
]
4042
with log_context(
4143
_logger,
42-
logging.DEBUG,
43-
msg=f"Removing {keys_to_remove=} from Redis for {lrt_namespace=}",
44+
logging.INFO,
45+
msg=f"Removing {keys_to_remove=} from Redis for {redis_namespace=}",
4446
):
4547
if len(keys_to_remove) > 0:
4648
await self._redis.delete(*keys_to_remove)

services/director-v2/src/simcore_service_director_v2/modules/dynamic_sidecar/scheduler/_core/_events_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -286,12 +286,13 @@ async def service_remove_sidecar_proxy_docker_networks_and_volumes(
286286
await app.state.dynamic_sidecar_scheduler.scheduler.remove_service_from_observation(
287287
scheduler_data.node_uuid
288288
)
289+
290+
await _cleanup_long_running_tasks(app, scheduler_data.node_uuid)
291+
289292
await task_progress.update(
290293
message="finished removing resources", percent=ProgressPercent(1)
291294
)
292295

293-
await _cleanup_long_running_tasks(app, scheduler_data.node_uuid)
294-
295296

296297
async def _cleanup_long_running_tasks(app: FastAPI, node_id: NodeID) -> None:
297298
long_running_client_helper = get_long_running_client_helper(app)

0 commit comments

Comments
 (0)