New quota runner configuration options

Author: tisnik

src/models/config.py

@@ -1323,6 +1323,22 @@ class QuotaSchedulerConfiguration(ConfigurationBase):
         description="Quota scheduler period specified in seconds",
     )
 
+    database_reconnection_count: PositiveInt = Field(
+        10,
+        title="Database reconnection count on startup",
+        description="Database reconnection count on startup. When database for "
+        "quota is not available on startup, the service tries to reconnect N "
+        "times with specified delay.",
+    )
+
+    database_reconnection_delay: PositiveInt = Field(
+        1,
+        title="Database reconnection delay",
+        description="Database reconnection delay specified in seconds. When database for "
+        "quota is not available on startup, the service tries to reconnect N "
+        "times with specified delay.",
+    )
+
 
 class QuotaHandlersConfiguration(ConfigurationBase):
     """Quota limiter configuration.

tests/unit/models/config/test_dump_configuration.py

@@ -183,7 +183,11 @@ def test_dump_configuration(tmp_path: Path) -> None:
                 "sqlite": None,
                 "postgres": None,
                 "limiters": [],
-                "scheduler": {"period": 1},
+                "scheduler": {
+                    "period": 1,
+                    "database_reconnection_count": 10,
+                    "database_reconnection_delay": 1,
+                },
                 "enable_token_history": False,
             },
         }
@@ -498,7 +502,11 @@ def test_dump_configuration_with_quota_limiters(tmp_path: Path) -> None:
                         "type": "cluster_limiter",
                     },
                 ],
-                "scheduler": {"period": 10},
+                "scheduler": {
+                    "period": 10,
+                    "database_reconnection_count": 10,
+                    "database_reconnection_delay": 1,
+                },
                 "enable_token_history": True,
             },
         }
@@ -678,7 +686,11 @@ def test_dump_configuration_byok(tmp_path: Path) -> None:
                 "sqlite": None,
                 "postgres": None,
                 "limiters": [],
-                "scheduler": {"period": 1},
+                "scheduler": {
+                    "period": 1,
+                    "database_reconnection_count": 10,
+                    "database_reconnection_delay": 1,
+                },
                 "enable_token_history": False,
             },
         }
@@ -844,7 +856,11 @@ def test_dump_configuration_pg_namespace(tmp_path: Path) -> None:
                 "sqlite": None,
                 "postgres": None,
                 "limiters": [],
-                "scheduler": {"period": 1},
+                "scheduler": {
+                    "period": 1,
+                    "database_reconnection_count": 10,
+                    "database_reconnection_delay": 1,
+                },
                 "enable_token_history": False,
             },
         }

tests/unit/models/config/test_quota_scheduler_config.py

@@ -13,13 +13,21 @@ def test_quota_scheduler_default_configuration() -> None:
     assert cfg is not None
     # default value
     assert cfg.period == 1
+    assert cfg.database_reconnection_count == 10
+    assert cfg.database_reconnection_delay == 1
 
 
 def test_quota_scheduler_custom_configuration() -> None:
     """Test the custom configuration."""
-    cfg = QuotaSchedulerConfiguration(period=10)
+    cfg = QuotaSchedulerConfiguration(
+        period=10,
+        database_reconnection_count=2,
+        database_reconnection_delay=3,
+    )
     assert cfg is not None
     assert cfg.period == 10
+    assert cfg.database_reconnection_count == 2
+    assert cfg.database_reconnection_delay == 3
 
 
 def test_quota_scheduler_custom_configuration_zero_period() -> None:
@@ -32,3 +40,15 @@ def test_quota_scheduler_custom_configuration_negative_period() -> None:
     """Test that negative period value raises ValidationError."""
     with pytest.raises(ValidationError, match="Input should be greater than 0"):
         QuotaSchedulerConfiguration(period=-10)
+
+
+def test_quota_scheduler_custom_configuration_negative_reconnection_count() -> None:
+    """Test that negative period value raises ValidationError."""
+    with pytest.raises(ValidationError, match="Input should be greater than 0"):
+        QuotaSchedulerConfiguration(database_reconnection_count=-10)
+
+
+def test_quota_scheduler_custom_configuration_negative_reconnection_delay() -> None:
+    """Test that negative period value raises ValidationError."""
+    with pytest.raises(ValidationError, match="Input should be greater than 0"):
+        QuotaSchedulerConfiguration(database_reconnection_delay=-10)