HH-235762 add skip_empty_upstream_check_for_upstreams option

hhru · Nov 6, 2024 · 026ba7f · 026ba7f
1 parent b1fb812
commit 026ba7f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/frontik/options.py b/frontik/options.py
@@ -113,6 +113,7 @@ class Options:
     # upstream options
     upstreams: list = field(default_factory=lambda: [])
     fail_start_on_empty_upstream: bool = True
+    skip_empty_upstream_check_for_upstreams: list = field(default_factory=lambda: [])
 
     # opentelemetry options
     opentelemetry_collector_url: str = 'http://127.0.0.1:2360'

diff --git a/frontik/service_discovery.py b/frontik/service_discovery.py
@@ -209,7 +209,8 @@ def deregister_service_and_close(self) -> None:
             log.info('Failed to deregister service %s normally', self.service_id)
 
     def __check_empty_upstreams_on_startup(self) -> None:
-        empty_upstreams = [k for k, v in self._upstreams.items() if not v.servers]
+        skip_check_upstreams = options.skip_empty_upstream_check_for_upstreams
+        empty_upstreams = [k for k, v in self._upstreams.items() if not v.servers and k not in skip_check_upstreams]
         if empty_upstreams:
             msg = f'failed startup application, because for next upstreams got empty servers: {empty_upstreams}'
             raise RuntimeError(msg)