Add scheduling_policy column to safekeepers table

libs/pageserver_api/src/controller_api.rs

@@ -387,6 +387,7 @@ pub struct SafekeeperDescribeResponse {
     pub port: i32,
     pub http_port: i32,
     pub availability_zone_id: String,
+    pub scheduling_policy: Option<String>,
 }
 
 #[cfg(test)]

storage_controller/migrations/2024-12-12-212515_safekeepers_scheduling_policy/down.sql

@@ -0,0 +1 @@
+ALTER TABLE safekeepers DROP scheduling_policy;

storage_controller/migrations/2024-12-12-212515_safekeepers_scheduling_policy/up.sql

@@ -0,0 +1 @@
+ALTER TABLE safekeepers ADD scheduling_policy VARCHAR;

storage_controller/src/persistence.rs

@@ -1227,6 +1227,7 @@ pub(crate) struct SafekeeperPersistence {
     pub(crate) active: bool,
     pub(crate) http_port: i32,
     pub(crate) availability_zone_id: String,
+    pub(crate) scheduling_policy: Option<String>,
 }
 
 impl SafekeeperPersistence {
@@ -1252,6 +1253,7 @@ impl SafekeeperPersistence {
             port: self.port,
             http_port: self.http_port,
             availability_zone_id: self.availability_zone_id.clone(),
+            scheduling_policy: self.scheduling_policy.clone(),
         }
     }
 }

storage_controller/src/schema.rs

@@ -39,6 +39,7 @@ diesel::table! {
         active -> Bool,
         http_port -> Int4,
         availability_zone_id -> Text,
+        scheduling_policy -> Nullable<Varchar>,
     }
 }
 

test_runner/regress/test_storage_controller.py

@@ -3009,7 +3009,7 @@ def test_safekeeper_deployment_time_update(neon_env_builder: NeonEnvBuilder):
 def eq_safekeeper_records(a: dict[str, Any], b: dict[str, Any]) -> bool:
     compared = [dict(a), dict(b)]
 
-    masked_keys = ["created_at", "updated_at", "active"]
+    masked_keys = ["created_at", "updated_at", "active", "scheduling_policy"]
 
     for d in compared:
         # keep deleting these in case we are comparing the body as it will be uploaded by real scripts