Function main

crates/hyperqueue/src/bin/hq.rs:432–564

()

#[tokio::main(flavor = "current_thread")]
async fn main() -> hyperqueue::Result<()> {
    // Augment panics - first print the error and backtrace like normally,
    // and then print our own custom error message.
    let std_panic = std::panic::take_hook();
    std::panic::set_hook(Box::new(move |info: &PanicHookInfo| {
        std_panic(info);
        hq_panic_hook(info);
    }));

    // Also enable backtraces by default.
    // This enables backtraces when panicking, but also for normal anyhow errors.
    // SAFETY: we are at the beginning of the program, no other threads that could set
    // environment variables should be executing.
    unsafe {
        std::env::set_var("RUST_BACKTRACE", "full");
    }

    // This further disables backtraces for normal anyhow errors.
    // They should not be printed to users in release mode.
    #[cfg(not(debug_assertions))]
    {
        unsafe {
            std::env::set_var("RUST_LIB_BACKTRACE", "0");
        }
    }

    let matches = RootOptions::command().get_matches();
    let top_opts = match RootOptions::from_arg_matches(&matches) {
        Ok(opts) => opts,
        Err(error) => error.exit(),
    };

    setup_logging(top_opts.common.debug);

    let gsettings = make_global_settings(top_opts.common);

    let is_cli_like = match &top_opts.subcmd {
        SubCommand::Server(ServerOpts {
            subcmd: ServerCommand::Start(_),
        }) => false,
        SubCommand::Worker(WorkerOpts {
            subcmd: WorkerCommand::Start(_),
        }) => false,
        #[cfg(feature = "dashboard")]
        SubCommand::Dashboard(_) => false,
        _ => true,
    };

    if is_cli_like {
        // When our stdout is attached to a pipe and the pipe is closed,
        // it manifests as an I/O error, because the Rust runtime ignores
        // SIGPIPE by default.
        // This in turn causes `println!` to panic, which is not ideal,
        // because it crashes HQ when used with Unix CLI utilities (such as `head`).
        // Therefore, we reset SIGPIPE to its default behavior (terminate the process)
        // to avoid the panics.
        // See https://github.com/It4innovations/hyperqueue/issues/851.
        // However, we only do this for client commands, which are short running and