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

Support acting as a pull-through cache

Browse source
This commit is contained in:
Zhaofeng Li 2023-05-08 12:59:57 -06:00
parent 7b5da1fb69
commit b1b798ef40
2 changed files with 23 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -9,7 +9,7 @@ This project depends on internal APIs used by the GitHub Actions Cache.
See `gha-cache/README.md` for more details on how to obtain the required tokens. See `gha-cache/README.md` for more details on how to obtain the required tokens.
``` ```
cargo run -- -c creds.json cargo run -- -c creds.json --upstream https://cache.nixos.org
cargo build --release --target x86_64-unknown-linux-musl cargo build --release --target x86_64-unknown-linux-musl
cargo build --release --target aarch64-unknown-linux-musl cargo build --release --target aarch64-unknown-linux-musl
nix copy --to 'http://127.0.0.1:3000' $(which bash) nix copy --to 'http://127.0.0.1:3000' $(which bash)

21
nix-actions-cache/src/main.rs
View file

@ -58,12 +58,20 @@ struct Args {
/// Using another version string allows you to "bust" the cache. /// Using another version string allows you to "bust" the cache.
#[arg(long)] #[arg(long)]
cache_version: Option<String>, cache_version: Option<String>,
/// The upstream cache.
///
/// Requests for unknown NARs are redirected to this cache
/// instead.
#[arg(long)]
upstream: Option<String>,
} }
/// The global server state. /// The global server state.
#[derive(Debug)] #[derive(Debug)]
struct StateInner { struct StateInner {
api: Api, api: Api,
upstream: Option<String>,
} }
#[tokio::main] #[tokio::main]
@ -91,7 +99,10 @@ async fn main() {
api.mutate_version(cache_version.as_bytes()); api.mutate_version(cache_version.as_bytes());
} }
let state = Arc::new(StateInner { api }); let state = Arc::new(StateInner {
api,
upstream: args.upstream,
});
let app = Router::new() let app = Router::new()
.route("/", get(root)) .route("/", get(root))
@ -160,8 +171,12 @@ async fn get_narinfo(
return Ok(Redirect::temporary(&url)); return Ok(Redirect::temporary(&url));
} }
if let Some(upstream) = &state.upstream {
Ok(Redirect::temporary(&format!("{}/{}", upstream, path)))
} else {
Err(Error::NotFound) Err(Error::NotFound)
} }
}
async fn put_narinfo( async fn put_narinfo(
Extension(state): Extension<State>, Extension(state): Extension<State>,
Path(path): Path<String>, Path(path): Path<String>,
@ -193,8 +208,12 @@ async fn get_nar(Extension(state): Extension<State>, Path(path): Path<String>) -
return Ok(Redirect::temporary(&url)); return Ok(Redirect::temporary(&url));
} }
if let Some(upstream) = &state.upstream {
Ok(Redirect::temporary(&format!("{}/nar/{}", upstream, path)))
} else {
Err(Error::NotFound) Err(Error::NotFound)
} }
}
async fn put_nar( async fn put_nar(
Extension(state): Extension<State>, Extension(state): Extension<State>,
Path(path): Path<String>, Path(path): Path<String>,