Skip to content

Commit db1a842

Browse files
authored
Drop legacy server environment variables (#3349)
* DSTACK_SERVER_METRICS_TTL_SECONDS * DSTACK_FORCE_BRIDGE_NETWORK
1 parent b780ad3 commit db1a842

1 file changed

Lines changed: 15 additions & 55 deletions

File tree

src/dstack/_internal/server/settings.py

Lines changed: 15 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
"""
44

55
import os
6-
import warnings
76
from enum import Enum
87
from pathlib import Path
98

@@ -79,31 +78,17 @@
7978

8079
SERVER_GCP_LOGGING_PROJECT = os.getenv("DSTACK_SERVER_GCP_LOGGING_PROJECT")
8180

82-
SERVER_METRICS_RUNNING_TTL_SECONDS: int
83-
_SERVER_METRICS_RUNNING_TTL_SECONDS = os.getenv("DSTACK_SERVER_METRICS_RUNNING_TTL_SECONDS")
84-
if _SERVER_METRICS_RUNNING_TTL_SECONDS is None:
85-
_SERVER_METRICS_RUNNING_TTL_SECONDS = os.getenv("DSTACK_SERVER_METRICS_TTL_SECONDS")
86-
if _SERVER_METRICS_RUNNING_TTL_SECONDS is not None:
87-
warnings.warn(
88-
(
89-
"DSTACK_SERVER_METRICS_TTL_SECONDS is deprecated,"
90-
" use DSTACK_SERVER_METRICS_RUNNING_TTL_SECONDS instead"
91-
),
92-
DeprecationWarning,
93-
)
94-
else:
95-
_SERVER_METRICS_RUNNING_TTL_SECONDS = 3600
96-
SERVER_METRICS_RUNNING_TTL_SECONDS = int(_SERVER_METRICS_RUNNING_TTL_SECONDS)
97-
del _SERVER_METRICS_RUNNING_TTL_SECONDS
98-
SERVER_METRICS_FINISHED_TTL_SECONDS = int(
99-
os.getenv("DSTACK_SERVER_METRICS_FINISHED_TTL_SECONDS", 7 * 24 * 3600)
81+
SERVER_METRICS_RUNNING_TTL_SECONDS = environ.get_int(
82+
"DSTACK_SERVER_METRICS_RUNNING_TTL_SECONDS", default=3600
10083
)
101-
102-
SERVER_INSTANCE_HEALTH_TTL_SECONDS = int(
103-
os.getenv("DSTACK_SERVER_INSTANCE_HEALTH_TTL_SECONDS", 7 * 24 * 3600)
84+
SERVER_METRICS_FINISHED_TTL_SECONDS = environ.get_int(
85+
"DSTACK_SERVER_METRICS_FINISHED_TTL_SECONDS", default=7 * 24 * 3600
86+
)
87+
SERVER_INSTANCE_HEALTH_TTL_SECONDS = environ.get_int(
88+
"DSTACK_SERVER_INSTANCE_HEALTH_TTL_SECONDS", default=7 * 24 * 3600
10489
)
105-
SERVER_INSTANCE_HEALTH_MIN_COLLECT_INTERVAL_SECONDS = int(
106-
os.getenv("DSTACK_SERVER_INSTANCE_HEALTH_MIN_COLLECT_INTERVAL_SECONDS", 60)
90+
SERVER_INSTANCE_HEALTH_MIN_COLLECT_INTERVAL_SECONDS = environ.get_int(
91+
"DSTACK_SERVER_INSTANCE_HEALTH_MIN_COLLECT_INTERVAL_SECONDS", default=60
10792
)
10893

10994
SERVER_EVENTS_TTL_SECONDS = int(
@@ -159,34 +144,9 @@ class JobNetworkMode(Enum):
159144

160145

161146
DEFAULT_JOB_NETWORK_MODE = JobNetworkMode.HOST_WHEN_POSSIBLE
162-
163-
164-
def _get_job_network_mode() -> JobNetworkMode:
165-
# Current default
166-
mode = DEFAULT_JOB_NETWORK_MODE
167-
bridge_var = "DSTACK_FORCE_BRIDGE_NETWORK"
168-
force_bridge = environ.get_bool(bridge_var)
169-
mode_var = "DSTACK_SERVER_JOB_NETWORK_MODE"
170-
mode_from_env = environ.get_enum(mode_var, JobNetworkMode, value_type=int)
171-
if mode_from_env is not None:
172-
if force_bridge is not None:
173-
logger.warning(
174-
f"{bridge_var} is deprecated since 0.19.27 and ignored when {mode_var} is set"
175-
)
176-
return mode_from_env
177-
if force_bridge is not None:
178-
if force_bridge:
179-
mode = JobNetworkMode.FORCED_BRIDGE
180-
logger.warning(
181-
(
182-
f"{bridge_var} is deprecated since 0.19.27."
183-
f" Set {mode_var} to {mode.value} and remove {bridge_var}"
184-
)
185-
)
186-
else:
187-
logger.warning(f"{bridge_var} is deprecated since 0.19.27. Remove {bridge_var}")
188-
return mode
189-
190-
191-
JOB_NETWORK_MODE = _get_job_network_mode()
192-
del _get_job_network_mode
147+
JOB_NETWORK_MODE = environ.get_enum(
148+
"DSTACK_SERVER_JOB_NETWORK_MODE",
149+
JobNetworkMode,
150+
value_type=int,
151+
default=DEFAULT_JOB_NETWORK_MODE,
152+
)

0 commit comments

Comments
 (0)