cy/nixcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

make recursive optional and tune down concurrency

Browse source
This commit is contained in:
cy 2025-04-01 15:26:01 -04:00
parent dbb1e9109e
commit 73950bd2a3
Signed by: cy
SSH key fingerprint: SHA256:o/geVWV4om1QhUSkKvDQeW/eAihwnjyXkqMwrVdbuts
1 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/main.rs
View file

@ -30,12 +30,13 @@ struct PathInfo {
impl PathInfo { impl PathInfo {
// find derivations related to package // find derivations related to package
async fn from_package(package: &str) -> Vec<Self> { async fn from_package(package: &str, recursive: bool) -> Vec<Self> {
let mut args = vec!["path-info", "--derivation", "--json"];
if recursive {
args.push("--recursive");
}
let path_infos = Command::new("nix") let path_infos = Command::new("nix")
.arg("path-info") .args(args)
.arg("--derivation")
.arg("--recursive")
.arg("--json")
.arg(package) .arg(package)
.output() .output()
.await .await
@ -81,8 +82,12 @@ struct Cli {
#[arg(long, short)] #[arg(long, short)]
upstream_cache: Vec<String>, upstream_cache: Vec<String>,
/// Whether to pass --recursive to nix path-info. Can queue a huge number of paths to upload
#[arg(long, short, default_value_t = false)]
recursive: bool,
/// Concurrent upstream cache checkers /// Concurrent upstream cache checkers
#[arg(long, default_value_t = 64)] #[arg(long, default_value_t = 32)]
upstream_checker_concurrency: u8, upstream_checker_concurrency: u8,
/// Concurrent uploaders /// Concurrent uploaders
@ -90,7 +95,7 @@ struct Cli {
uploader_concurrency: u8, uploader_concurrency: u8,
/// Concurrent nix-store commands to run /// Concurrent nix-store commands to run
#[arg(long, default_value_t = 128)] #[arg(long, default_value_t = 32)]
nix_store_concurrency: u8, nix_store_concurrency: u8,
} }
@ -109,7 +114,7 @@ async fn main() {
debug!("upstream caches: {:#?}", upstream_caches); debug!("upstream caches: {:#?}", upstream_caches);
println!("querying nix path-info"); println!("querying nix path-info");
let derivations = PathInfo::from_package(package).await; let derivations = PathInfo::from_package(package, cli.recursive).await;
println!("got {} derivations", derivations.len()); println!("got {} derivations", derivations.len());
println!("querying nix-store"); println!("querying nix-store");