TLA  Line data    Source code

       1                 : use std::sync::Arc;
       2                 : 
       3                 : use pageserver_client::mgmt_api;
       4                 : use tracing::info;
       5                 : use utils::id::TenantTimelineId;
       6                 : 
       7                 : pub(crate) struct Spec {
       8                 :     pub(crate) limit_to_first_n_targets: Option<usize>,
       9                 :     pub(crate) targets: Option<Vec<TenantTimelineId>>,
      10                 : }
      11                 : 
      12 UBC           0 : pub(crate) async fn discover(
      13               0 :     api_client: &Arc<mgmt_api::Client>,
      14               0 :     spec: Spec,
      15               0 : ) -> anyhow::Result<Vec<TenantTimelineId>> {
      16               0 :     let mut timelines = if let Some(targets) = spec.targets {
      17               0 :         targets
      18                 :     } else {
      19               0 :         mgmt_api::util::get_pageserver_tenant_timelines_unsharded(api_client).await?
      20                 :     };
      21                 : 
      22               0 :     if let Some(limit) = spec.limit_to_first_n_targets {
      23               0 :         timelines.sort(); // for determinism
      24               0 :         timelines.truncate(limit);
      25               0 :         if timelines.len() < limit {
      26               0 :             anyhow::bail!("pageserver has less than limit_to_first_n_targets={limit} tenants");
      27               0 :         }
      28               0 :     }
      29                 : 
      30               0 :     info!("timelines:\n{:?}", timelines);
      31               0 :     info!("number of timelines:\n{:?}", timelines.len());
      32                 : 
      33               0 :     Ok(timelines)
      34               0 : }