mirrors/magic-nix-cache
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Merge pull request #16 from DeterminateSystems/http-startup-notification

Browse source 
Send startup notification via HTTP
This commit is contained in:
Eelco Dolstra 2024-02-23 18:25:04 +01:00 committed by GitHub
parent 0537b74a1e 0607f5efa4
commit 75b1450fdf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
magic-nix-cache/src/main.rs
View file

@ -23,7 +23,6 @@ use std::collections::HashSet;
use std::fs::{self, create_dir_all, OpenOptions};
use std::io::Write;
use std::net::SocketAddr;
use std::os::fd::FromRawFd;
use std::os::unix::fs::PermissionsExt;
use std::path::{Path, PathBuf};
use std::sync::Arc;
@ -103,9 +102,9 @@ struct Args {
#[arg(long)]
use_flakehub: bool,
/// File descriptor on which to send startup notification.
/// URL to which to post startup notification.
#[arg(long)]
notify_fd: Option<i32>,
startup_notification_url: Option<reqwest::Url>,
}
/// The global server state.
@ -300,9 +299,27 @@ async fn main_cli() {
tracing::info!("Listening on {}", args.listen);
if let Some(notify_fd) = args.notify_fd {
let mut f = unsafe { std::fs::File::from_raw_fd(notify_fd) };
writeln!(&mut f, "INIT").unwrap();
if let Some(startup_notification_url) = args.startup_notification_url {
let response = reqwest::Client::new()
.post(startup_notification_url)
.header(reqwest::header::CONTENT_TYPE, "application/json")
.body("{}")
.send()
.await;
match response {
Ok(response) => {
if !response.status().is_success() {
panic!(
"Startup notification returned an error: {}\n{}",
response.status(),
response.text().await.unwrap_or_else(|_| "".to_owned())
);
}
}
Err(err) => {
panic!("Startup notification failed: {}", err);
}
}
}
let ret = axum::Server::bind(&args.listen)
@ -339,7 +356,7 @@ async fn post_build_hook(out_paths: &str) {
let response = reqwest::Client::new()
.post(format!("http://{}/api/enqueue-paths", &args.server))
.header("Content-Type", "application/json")
.header(reqwest::header::CONTENT_TYPE, "application/json")
.body(serde_json::to_string(&request).unwrap())
.send()
.await;