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

hack: skip default keep-alive

Browse source
This commit is contained in:
Cole Mickens 2024-08-08 16:23:24 -07:00
parent 57eb3e75c0
commit f82811cc9c
1 changed files with 22 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -37,6 +37,7 @@ use clap::Parser;
use http_body_util::BodyExt; use http_body_util::BodyExt;
use hyper_util::rt::{TokioExecutor, TokioIo}; use hyper_util::rt::{TokioExecutor, TokioIo};
use futures::StreamExt; use futures::StreamExt;
use serde::{Deserialize,Serialize};
use tempfile::NamedTempFile; use tempfile::NamedTempFile;
use tokio::fs::File; use tokio::fs::File;
use tokio::io::AsyncWriteExt; use tokio::io::AsyncWriteExt;
@ -52,6 +53,15 @@ use gha_cache::Credentials;
const DETERMINATE_STATE_DIR: &str = "/nix/var/determinate"; const DETERMINATE_STATE_DIR: &str = "/nix/var/determinate";
const DETERMINATE_NIXD_SOCKET_NAME: &str = "determinate-nixd.socket"; const DETERMINATE_NIXD_SOCKET_NAME: &str = "determinate-nixd.socket";
// TODO(colemickens): refactor, move with other UDS stuff (or all PBH stuff) to new file
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(tag = "c", rename_all = "kebab-case")]
pub struct BuiltPathResponseEventV1 {
pub drv: PathBuf,
pub outputs: Vec<PathBuf>,
}
type State = Arc<StateInner>; type State = Arc<StateInner>;
/// GitHub Actions-powered Nix binary cache /// GitHub Actions-powered Nix binary cache
@ -352,15 +362,21 @@ async fn main_cli() -> Result<()> {
let response = sender.send_request(request).await?; let response = sender.send_request(request).await?;
let mut data = response.into_data_stream(); let mut data = response.into_data_stream();
while let Some(foo) = data.next().await { while let Some(event_str) = data.next().await {
tracing::info!("got {:?}", foo); tracing::info!("got {:?}", event_str);
// TOOD: skip our keep-alive, maybe we should set it, we rely on the axum default "\n\n" right now
// but we need to skip those lines, anyway, and not bother trying to parse them
// TODO(colemickens): error handle
let sp = String::from_utf8_lossy(&foo.unwrap()).to_string(); let event_str = event_str.unwrap();
let store_paths = vec![sp]; if event_str == "\n\n" {
// TODO: hacky, could be better
continue
}
let event: BuiltPathResponseEventV1 = serde_json::from_slice(&event_str)?;
// TODO(colemickens): error handling::: // TODO(colemickens): error handling:::
let store_paths = store_paths let store_paths = event.outputs
.iter() .iter()
.map(|path| state.store.follow_store_path(path).map_err(|_| anyhow!("ahhhhh"))) .map(|path| state.store.follow_store_path(path).map_err(|_| anyhow!("ahhhhh")))
.collect::<Result<Vec<_>>>()?; .collect::<Result<Vec<_>>>()?;