Skip to content

Commit 8bcddd3

Browse files
rename some env variable
1 parent 96db0e3 commit 8bcddd3

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

cads_broker/config.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -30,12 +30,12 @@ class BrokerConfig(pydantic_settings.BaseSettings):
3030
high_priority_user_uid: str = "8d8ee054-6a09-4da8-a5be-d5dff52bbc5f"
3131
broker_priority_algorithm: str = "legacy"
3232
broker_priority_interval_hours: int = 24
33-
get_number_of_workers_cache_time: int = 10
34-
qos_rules_cache_time: int = 10
35-
get_tasks_from_scheduler_cache_time: int = 1
36-
rules_path: str = "/src/rules.qos"
37-
wait_time: float = 2.
38-
sync_database_cache_time: int = 10
33+
broker_get_number_of_workers_cache_time: int = 10
34+
broker_qos_rules_cache_time: int = 10
35+
broker_get_tasks_from_scheduler_cache_time: int = 1
36+
broker_rules_path: str = "/src/rules.qos"
37+
broker_wait_time: float = 2.
38+
broker_sync_database_cache_time: int = 10
3939
broker_requeue_on_killed_worker_requests: bool = False
4040
broker_requeue_on_lost_requests: bool = True
4141
broker_requeue_limit: int = 3

cads_broker/dispatcher.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343

4444
@cachetools.cached( # type: ignore
4545
cache=cachetools.TTLCache(
46-
maxsize=1024, ttl=CONFIG.get_number_of_workers_cache_time
46+
maxsize=1024, ttl=CONFIG.broker_get_number_of_workers_cache_time
4747
),
4848
info=True,
4949
)
@@ -56,7 +56,7 @@ def get_number_of_workers(client: distributed.Client) -> int:
5656

5757

5858
@cachetools.cached( # type: ignore
59-
cache=cachetools.TTLCache(maxsize=1024, ttl=CONFIG.qos_rules_cache_time),
59+
cache=cachetools.TTLCache(maxsize=1024, ttl=CONFIG.broker_qos_rules_cache_time),
6060
info=True,
6161
)
6262
def get_rules_hash(rules_path: str):
@@ -70,7 +70,7 @@ def get_rules_hash(rules_path: str):
7070

7171
@cachetools.cached( # type: ignore
7272
cache=cachetools.TTLCache(
73-
maxsize=1024, ttl=CONFIG.get_tasks_from_scheduler_cache_time
73+
maxsize=1024, ttl=CONFIG.broker_get_tasks_from_scheduler_cache_time
7474
),
7575
info=True,
7676
)
@@ -188,7 +188,7 @@ def reset(self) -> None:
188188
class QoSRules:
189189
def __init__(self, number_of_workers) -> None:
190190
self.environment = Environment.Environment(number_of_workers=number_of_workers)
191-
self.rules_path = CONFIG.rules_path
191+
self.rules_path = CONFIG.broker_rules_path
192192
if os.path.exists(self.rules_path):
193193
self.rules = self.rules_path
194194
else:
@@ -206,9 +206,9 @@ class Broker:
206206
address: str
207207
session_maker_read: sa.orm.sessionmaker
208208
session_maker_write: sa.orm.sessionmaker
209-
wait_time: float = CONFIG.wait_time
209+
wait_time: float = CONFIG.broker_wait_time
210210
ttl_cache = cachetools.TTLCache(
211-
maxsize=1024, ttl=CONFIG.sync_database_cache_time
211+
maxsize=1024, ttl=CONFIG.broker_sync_database_cache_time
212212
)
213213

214214
futures: dict[str, distributed.Future] = attrs.field(

0 commit comments

Comments
 (0)