|
3 | 3 | """ |
4 | 4 |
|
5 | 5 | import os |
6 | | -import warnings |
7 | 6 | from enum import Enum |
8 | 7 | from pathlib import Path |
9 | 8 |
|
|
79 | 78 |
|
80 | 79 | SERVER_GCP_LOGGING_PROJECT = os.getenv("DSTACK_SERVER_GCP_LOGGING_PROJECT") |
81 | 80 |
|
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 |
100 | 83 | ) |
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 |
104 | 89 | ) |
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 |
107 | 92 | ) |
108 | 93 |
|
109 | 94 | SERVER_EVENTS_TTL_SECONDS = int( |
@@ -159,34 +144,9 @@ class JobNetworkMode(Enum): |
159 | 144 |
|
160 | 145 |
|
161 | 146 | 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