aico-pack/lib/shared/processConcurrency.js

A tool to pack repository contents to single file for AI consumption

import os from 'node:os';
import { Piscina } from 'piscina';
import { logger } from './logger.js';
export const getProcessConcurrency = () => {
    return typeof os.availableParallelism === 'function' ? os.availableParallelism() : os.cpus().length;
};
export const getWorkerThreadCount = (numOfTasks) => {
    const processConcurrency = getProcessConcurrency();
    const minThreads = 1;
    // Limit max threads based on number of tasks
    const maxThreads = Math.max(minThreads, Math.min(processConcurrency, Math.ceil(numOfTasks / 100)));
    return {
        minThreads,
        maxThreads,
    };
};
export const initPiscina = (numOfTasks, workerPath) => {
    const { minThreads, maxThreads } = getWorkerThreadCount(numOfTasks);
    logger.trace(`Initializing worker pool with min=${minThreads}, max=${maxThreads} threads. Worker path: ${workerPath}`);
    return new Piscina({
        filename: workerPath,
        minThreads,
        maxThreads,
        idleTimeout: 5000,
        env: Object.assign(Object.assign({}, process.env), { REPOMIX_LOGLEVEL: logger.getLogLevel().toString() }),
    });
};
//# sourceMappingURL=processConcurrency.js.map