Line data    Source code

       1              : use clap::Parser;
       2              : use utils::logging;
       3              : 
       4              : /// Re-usable pieces of code that aren't CLI-specific.
       5              : mod util {
       6              :     pub(crate) mod request_stats;
       7              :     #[macro_use]
       8              :     pub(crate) mod tokio_thread_local_stats;
       9              :     /// Re-usable pieces of CLI-specific code.
      10              :     pub(crate) mod cli {
      11              :         pub(crate) mod targets;
      12              :     }
      13              : }
      14              : 
      15              : /// The pagebench CLI sub-commands, dispatched in [`main`] below.
      16              : mod cmd {
      17              :     pub(super) mod basebackup;
      18              :     pub(super) mod getpage_latest_lsn;
      19              :     pub(super) mod trigger_initial_size_calculation;
      20              : }
      21              : 
      22              : /// Component-level performance test for pageserver.
      23            0 : #[derive(clap::Parser)]
      24              : enum Args {
      25              :     Basebackup(cmd::basebackup::Args),
      26              :     GetPageLatestLsn(cmd::getpage_latest_lsn::Args),
      27              :     TriggerInitialSizeCalculation(cmd::trigger_initial_size_calculation::Args),
      28              : }
      29              : 
      30            0 : fn main() {
      31            0 :     logging::init(
      32            0 :         logging::LogFormat::Plain,
      33            0 :         logging::TracingErrorLayerEnablement::Disabled,
      34            0 :         logging::Output::Stderr,
      35            0 :     )
      36            0 :     .unwrap();
      37            0 :     logging::replace_panic_hook_with_tracing_panic_hook().forget();
      38            0 : 
      39            0 :     let args = Args::parse();
      40            0 :     match args {
      41            0 :         Args::Basebackup(args) => cmd::basebackup::main(args),
      42            0 :         Args::GetPageLatestLsn(args) => cmd::getpage_latest_lsn::main(args),
      43            0 :         Args::TriggerInitialSizeCalculation(args) => {
      44            0 :             cmd::trigger_initial_size_calculation::main(args)
      45              :         }
      46              :     }
      47            0 :     .unwrap()
      48            0 : }