ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-01-31 02:46:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix crash when mongodb host is overridden with cluster settings

Browse Source
This commit is contained in:
clearml 2024-12-31 22:04:10 +02:00
parent 893ba48eda
commit 478f6b531b
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apiserver/database/__init__.py
View File

@ -37,6 +37,8 @@ OVERRIDE_QUERY_ENV_KEY = "CLEARML_MONGODB_SERVICE_QUERY"
class DatabaseEntry(models.Base): class DatabaseEntry(models.Base):
host = StringField(required=True) host = StringField(required=True)
alias = StringField() alias = StringField()
name = StringField()
db = StringField()
class DatabaseFactory: class DatabaseFactory:
@ -78,10 +80,13 @@ class DatabaseFactory:
missing.append(key) missing.append(key)
continue continue
entry = cls._create_db_entry(alias=alias, settings=db_entries.get(key)) settings = {**db_entries.get(key)}
if not any(field in settings for field in ("name", "db")):
settings["name"] = key
entry = cls._create_db_entry(alias=alias, settings=settings)
if override_connection_string: if override_connection_string:
con_str = furl(override_connection_string).add(path=key).url con_str = override_connection_string
log.info(f"Using override mongodb connection string for {alias}: {con_str}") log.info(f"Using override mongodb connection string for {alias}: {con_str}")
entry.host = con_str entry.host = con_str
else: else: