type-compiler
Version:
A TypeScript compiler plugin for enhanced runtime type checking and analysis with Zod validation
174 lines (173 loc) • 6.24 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.WorkerPoolSchema = exports.WorkerPool = void 0;
exports.getWorkerPool = getWorkerPool;
const zod_1 = require("zod");
const os = __importStar(require("os"));
const worker_threads_1 = require("worker_threads");
const logger_1 = require("./logger");
// Worker thread code
if (!worker_threads_1.isMainThread && worker_threads_1.parentPort) {
// This code runs in worker threads
logger_1.logger.debug('Worker thread initialized');
worker_threads_1.parentPort.on('message', (data) => {
try {
const taskId = data.taskId;
const typeData = data.typeData;
logger_1.logger.trace(`Worker processing task: ${taskId}`, {
typeName: typeData.name
});
// Process the type data
// This is a simplified simulation - actual implementation would do real processing
let result = '';
if (typeData.name.includes('string')) {
result = 'z.string()';
}
else if (typeData.name.includes('number')) {
result = 'z.number()';
}
else if (typeData.name.includes('boolean')) {
result = 'z.boolean()';
}
else if (typeData.name.includes('Date')) {
result = 'z.date()';
}
else if (typeData.name.includes('Array') || typeData.name.includes('[]')) {
result = 'z.array(z.any())';
}
else if (typeData.name.includes('Record') || typeData.name.includes('object')) {
result = 'z.record(z.string(), z.any())';
}
else {
result = 'z.object({/* properties would be processed here */})';
}
logger_1.logger.trace(`Worker completed task: ${taskId}`);
// Send the result back to the main thread
worker_threads_1.parentPort.postMessage({
success: true,
result,
taskId
});
}
catch (error) {
logger_1.logger.error(`Worker error processing task: ${data.taskId}`, {
error: error instanceof Error ? error.message : String(error)
});
// Send error back to main thread
worker_threads_1.parentPort.postMessage({
success: false,
error: error instanceof Error ? error.message : String(error),
taskId: data.taskId
});
}
});
}
/**
* A pool of worker threads for parallel processing
*/
class WorkerPool {
constructor(options) {
this.options = options;
this.workers = [];
this.availableWorkers = [];
this.taskQueue = [];
this.isProcessing = false;
// Initialize workers
const workerCount = options.workerCount || os.cpus().length - 1;
logger_1.logger.info(`Initializing worker pool with ${workerCount} workers`);
for (let i = 0; i < workerCount; i++) {
this.createWorker();
}
}
/**
* Process a type using a worker thread
*/
processType(typeData) {
// Simple placeholder implementation
logger_1.logger.debug(`Worker pool received task: ${typeData.name || 'unnamed'}`);
return Promise.resolve('z.object({})');
}
/**
* Create a new worker
*/
createWorker() {
logger_1.logger.trace('Creating new worker thread');
// In a real implementation, this would create a worker_threads Worker
// with the path to the worker script
return {};
}
/**
* Terminate all workers
*/
terminate() {
// In a real implementation, this would terminate all workers
logger_1.logger.info('Terminating workers');
}
/**
* Shut down the worker pool
*/
async shutdown() {
// In a real implementation, this would wait for pending tasks and terminate workers
this.terminate();
logger_1.logger.info('Worker pool shut down');
}
}
exports.WorkerPool = WorkerPool;
/**
* Get a worker pool for parallel processing
*/
function getWorkerPool(options) {
if (!options.parallelProcessing) {
logger_1.logger.debug('Parallel processing is disabled, not creating worker pool');
return null;
}
logger_1.logger.info('Creating worker pool for parallel processing', {
workerCount: options.workerCount,
batchSize: options.workerBatchSize
});
return new WorkerPool(options);
}
exports.WorkerPoolSchema = z.object({
workers: z.any(),
availableWorkers: z.any(),
taskQueue: z.any(),
isProcessing: z.any(),
options: z.any(),
processType: z.any(),
createWorker: z.any(),
terminate: z.any(),
shutdown: z.any()
});