zccache 1.12.9

Local-first compiler cache for C/C++/Rust/Emscripten
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

#[cfg(unix)]
#[global_allocator]
static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;

#[cfg(windows)]
#[global_allocator]
static GLOBAL_WIN: mimalloc::MiMalloc = mimalloc::MiMalloc;

use clap::Parser;
use zccache::download_protocol::daemon_mgmt;
use zccache::download_protocol::{Request, Response};

#[derive(Debug, Parser)]
#[command(name = "zccache-download-daemon", version, about)]
struct Args {
    #[arg(long, default_value = "info")]
    log_level: String,

    #[arg(long)]
    foreground: bool,

    #[arg(long)]
    endpoint: Option<String>,
}

fn main() {
    let args = Args::parse();
    if args.foreground {
        init_tracing(&args.log_level);
        run_server(args);
    } else {
        print_status(&args);
    }
}

fn print_status(args: &Args) {
    let endpoint = args
        .endpoint
        .clone()
        .unwrap_or_else(daemon_mgmt::default_endpoint);
    println!("zccache-download-daemon v{}", env!("CARGO_PKG_VERSION"));
    println!();
    println!("  endpoint:   {endpoint}");
    println!("  lock file:  {}", daemon_mgmt::lock_file_path().display());
    match query_daemon_status(&endpoint) {
        Ok(status) => {
            println!("  status:     running");
            println!("  uptime:     {}s", status.uptime_secs);
            println!("  downloads:  {}", status.active_downloads);
            println!("  clients:    {}", status.connected_clients);
        }
        Err(_) => {
            println!("  status:     not running");
        }
    }
}

/// Connect to a running daemon over IPC and query its status.
fn query_daemon_status(endpoint: &str) -> Result<zccache::download::DownloadDaemonStatus, String> {
    let rt = tokio::runtime::Builder::new_current_thread()
        .enable_all()
        .build()
        .map_err(|e| format!("failed to create runtime: {e}"))?;
    rt.block_on(async {
        let mut conn = zccache::ipc::connect(endpoint)
            .await
            .map_err(|e| format!("download daemon not running at {endpoint}: {e}"))?;
        conn.send(&Request::Status)
            .await
            .map_err(|e| format!("failed to query download daemon: {e}"))?;
        match conn.recv::<Response>().await {
            Ok(Some(Response::Status(status))) => Ok(status),
            Ok(Some(Response::Error { message })) => Err(message),
            Ok(Some(other)) => Err(format!("unexpected response: {other:?}")),
            Ok(None) => Err("download daemon closed connection unexpectedly".to_string()),
            Err(e) => Err(format!("broken connection to download daemon: {e}")),
        }
    })
}

fn run_server(args: Args) {
    let endpoint = args.endpoint.unwrap_or_else(daemon_mgmt::default_endpoint);
    let pid = std::process::id();
    if let Err(err) = daemon_mgmt::write_lock_file(pid) {
        tracing::warn!("failed to write download daemon lock file: {err}");
    }

    let rt = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()
        .expect("failed to create runtime");
    rt.block_on(async move {
        let mut server = match zccache::download_daemon::DownloadDaemon::bind(&endpoint) {
            Ok(server) => server,
            Err(err) => {
                eprintln!("failed to bind download daemon at {endpoint}: {err}");
                daemon_mgmt::remove_lock_file();
                std::process::exit(1);
            }
        };
        let shutdown = server.shutdown_handle();
        tokio::spawn(async move {
            if let Ok(()) = tokio::signal::ctrl_c().await {
                shutdown.notify_one();
            }
        });
        if let Err(err) = server.run().await {
            eprintln!("download daemon error: {err}");
            daemon_mgmt::remove_lock_file();
            std::process::exit(1);
        }
        daemon_mgmt::remove_lock_file();
    });
}

fn init_tracing(level: &str) {
    use tracing_subscriber::EnvFilter;
    let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(level));
    tracing_subscriber::fmt()
        .with_env_filter(filter)
        .with_target(true)
        .with_thread_ids(true)
        .init();
}