pnpm 12.9で社内レジストリだけ並列リクエスト数を下げる

networkConcurrency: 4
registries:
  "http://127.0.0.1:45287/":
    scopes: ["@corp"]
    networkConcurrency: 1
  "http://127.0.0.1:33967/":
    scopes: ["@public"]

pnpm install --ignore-scripts --store-dir ./store --reporter append-only

node verify.mjs /work/bin/12.12.0/node_modules/.bin/pnpm limited /work/runs/12.12.0-limited-1

// 2つのローカルレジストリ（corp / public）を立て、pnpm install 中の同時リクエスト数の最大値を数える。
// 使い方: node verify.mjs <pnpmのパス> <limited|control> <出力先ディレクトリ>
import { createServer } from 'node:http';
import { createHash } from 'node:crypto';
import { execFileSync, spawn } from 'node:child_process';
import { mkdirSync, writeFileSync, existsSync } from 'node:fs';
import { join } from 'node:path';

const [pnpm, mode, outDir] = process.argv.slice(2);
const limited = mode === 'limited';
const PACKAGES = 8;
const DELAY_MS = 200;

// tar.gz のダミーパッケージを作る（tar コマンドを使う）
function makeTarball(dir, name, i) {
  const pkgDir = join(dir, 'src', name, 'package');
  mkdirSync(pkgDir, { recursive: true });
  writeFileSync(join(pkgDir, 'package.json'), JSON.stringify({ name, version: '1.0.0', main: 'index.js' }));
  writeFileSync(join(pkgDir, 'index.js'), `module.exports = ${i};`);
  return execFileSync('tar', ['-czf', '-', '-C', join(dir, 'src', name), 'package']);
}

function startRegistry(scope, dir) {
  const reg = { scope, active: 0, maxInFlight: 0, metadata: 0, tarball: 0, blobs: {} };
  for (let i = 0; i < PACKAGES; i++) {
    const name = `@${scope}/pkg${i}`;
    reg.blobs[name] = makeTarball(dir, name, i);
  }
  reg.server = createServer((req, res) => {
    const path = decodeURIComponent(req.url.split('?')[0]);
    const kind = path.endsWith('.tgz') ? 'tarball' : 'metadata';
    const name = path.slice(1).split('/-/')[0];
    if (!reg.blobs[name]) { res.writeHead(404).end(); return; }
    reg.active++;
    reg.maxInFlight = Math.max(reg.maxInFlight, reg.active);
    reg[kind]++;
    setTimeout(() => {
      let body = reg.blobs[name];
      if (kind === 'metadata') {
        const integrity = 'sha512-' + createHash('sha512').update(body).digest('base64');
        const tarball = `http://127.0.0.1:${reg.server.address().port}/${name}/-/pkg-1.0.0.tgz`;
        body = Buffer.from(JSON.stringify({
          name, 'dist-tags': { latest: '1.0.0' },
          versions: { '1.0.0': { name, version: '1.0.0', dist: { tarball, integrity } } },
          time: { '1.0.0': '2020-01-01T00:00:00Z' },
        }));
      }
      res.writeHead(200, { 'Content-Type': kind === 'metadata' ? 'application/json' : 'application/octet-stream', 'Content-Length': body.length });
      res.end(body);
      reg.active--;
    }, DELAY_MS);
  });
  return new Promise((r) => reg.server.listen(0, '127.0.0.1', () => r(reg)));
}

mkdirSync(outDir, { recursive: true });
const corp = await startRegistry('corp', outDir);
const pub = await startRegistry('public', outDir);
const registries = [corp, pub];

const project = join(outDir, 'project');
mkdirSync(project, { recursive: true });
const deps = {};
for (const r of registries) for (let i = 0; i < PACKAGES; i++) deps[`@${r.scope}/pkg${i}`] = '1.0.0';
writeFileSync(join(project, 'package.json'), JSON.stringify({ name: 'registry-concurrency-test', private: true, dependencies: deps }, null, 2));
let config = `networkConcurrency: 4\nenableGlobalVirtualStore: false\ncacheDir: ${join(outDir, 'cache')}\nregistries:\n`;
for (const r of registries) {
  config += `  "http://127.0.0.1:${r.server.address().port}/":\n    scopes: ["@${r.scope}"]\n`;
  if (limited && r.scope === 'corp') config += '    networkConcurrency: 1\n';
}
writeFileSync(join(project, 'pnpm-workspace.yaml'), config);

// ユーザー設定・キャッシュを出力先の中に閉じ込める
const env = { ...process.env, HOME: join(outDir, 'home'), XDG_CONFIG_HOME: join(outDir, 'config'), XDG_CACHE_HOME: join(outDir, 'xdg-cache'), XDG_DATA_HOME: join(outDir, 'data'), XDG_STATE_HOME: join(outDir, 'state'), CI: 'true' };
mkdirSync(env.HOME, { recursive: true });

const version = execFileSync(pnpm, ['--version'], { env, encoding: 'utf8' }).trim();
// 子プロセスの完了を待つ間もサーバーが応答できるよう、非同期で起動する
const child = spawn(pnpm, ['install', '--ignore-scripts', '--store-dir', join(outDir, 'store'), '--reporter', 'append-only'], { cwd: project, env });
let log = '';
child.stdout.on('data', (d) => (log += d));
child.stderr.on('data', (d) => (log += d));
const exit = await new Promise((r) => child.on('close', r));
writeFileSync(join(outDir, 'install.log'), log);

let installed = 0;
for (const r of registries) for (let i = 0; i < PACKAGES; i++) if (existsSync(join(project, 'node_modules', `@${r.scope}`, `pkg${i}`, 'package.json'))) installed++;
const result = {
  pnpm: version, mode, exit, installedPackages: installed,
  corp: { maxInFlight: corp.maxInFlight, metadata: corp.metadata, tarball: corp.tarball },
  public: { maxInFlight: pub.maxInFlight, metadata: pub.metadata, tarball: pub.tarball },
};
writeFileSync(join(outDir, 'result.json'), JSON.stringify(result, null, 2));
console.log(JSON.stringify(result));
for (const r of registries) r.server.close();