Skip to content

Commit

Permalink
TEMP: fetchCargoVendor: decrease concurrency
Browse files Browse the repository at this point in the history
  • Loading branch information
TomaSajt committed Nov 17, 2024
1 parent f4e9a21 commit ff707e5
Showing 1 changed file with 9 additions and 10 deletions.
19 changes: 9 additions & 10 deletions pkgs/build-support/rust/fetch-cargo-vendor-util.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,18 +118,17 @@ def create_vendor_staging(lockfile_path: Path, out_dir: Path) -> None:
out_dir.mkdir(exist_ok=True)
shutil.copy(lockfile_path, out_dir / "Cargo.lock")

# create a pool with at most 10 concurrent jobs
with mp.Pool(min(10, mp.cpu_count())) as pool:

if len(git_packages) != 0:
(out_dir / "git").mkdir()
# run download jobs in parallel
git_args_gen = ((url, git_sha_rev, out_dir) for git_sha_rev, url in git_sha_rev_to_url.items())
pool.starmap(download_git_tree, git_args_gen)

# git tree fetching is prone to fail when running concurrently
if len(git_packages) != 0:
(out_dir / "git").mkdir()
for git_sha_rev, url in git_sha_rev_to_url.items():
download_git_tree(url, git_sha_rev, out_dir)

# create a pool with at most 5 concurrent jobs
with mp.Pool(min(5, mp.cpu_count())) as pool:
if len(registry_packages) != 0:
(out_dir / "tarballs").mkdir()
# run download jobs in parallel
# run tarball download jobs in parallel
tarball_args_gen = ((pkg, out_dir) for pkg in registry_packages)
pool.starmap(download_tarball, tarball_args_gen)

Expand Down

0 comments on commit ff707e5

Please sign in to comment.