Line data    Source code

       1              : //! An utilization metric which is used to decide on which pageserver to put next tenant.
       2              : //!
       3              : //! The metric is exposed via `GET /v1/utilization`. Refer and maintain it's openapi spec as the
       4              : //! truth.
       5              : 
       6              : use anyhow::Context;
       7              : use std::path::Path;
       8              : 
       9              : use pageserver_api::models::PageserverUtilization;
      10              : 
      11            0 : pub(crate) fn regenerate(tenants_path: &Path) -> anyhow::Result<PageserverUtilization> {
      12              :     // TODO: currently the http api ratelimits this to 1Hz at most, which is probably good enough
      13              : 
      14            0 :     let statvfs = nix::sys::statvfs::statvfs(tenants_path)
      15            0 :         .map_err(std::io::Error::from)
      16            0 :         .context("statvfs tenants directory")?;
      17              : 
      18            0 :     let blocksz = statvfs.block_size();
      19            0 : 
      20            0 :     #[cfg_attr(not(target_os = "macos"), allow(clippy::unnecessary_cast))]
      21            0 :     let free = statvfs.blocks_available() as u64 * blocksz;
      22            0 :     let used = crate::metrics::RESIDENT_PHYSICAL_SIZE_GLOBAL.get();
      23            0 :     let captured_at = std::time::SystemTime::now();
      24            0 : 
      25            0 :     let doc = PageserverUtilization {
      26            0 :         disk_usage_bytes: used,
      27            0 :         free_space_bytes: free,
      28            0 :         // lower is better; start with a constant
      29            0 :         //
      30            0 :         // note that u64::MAX will be output as i64::MAX as u64, but that should not matter
      31            0 :         utilization_score: u64::MAX,
      32            0 :         captured_at,
      33            0 :     };
      34            0 : 
      35            0 :     // TODO: make utilization_score into a metric
      36            0 : 
      37            0 :     Ok(doc)
      38            0 : }