better-sqlite3-multiple-ciphers/benchmark/index.js

'use strict';
var path = require('path');
var fs = require('fs-extra');
var clc = require('cli-color');
var spawn = require('child_process').spawn;
var factory = require('./factory');
process.chdir(path.dirname(__dirname));
process.on('SIGINT', exit);
process.on('SIGHUP', exit);
process.on('SIGTERM', exit);
var trials;

(function init() {
	fs.removeSync('temp/');
	fs.ensureDirSync('temp/');
	
	trials = getTrials();
	if (!trials.length) {
		console.log(clc.yellow('No matching benchmarks found!'));
		return exit();
	}
	
	console.log('Generating tables...');
	factory.buildTables().then(function () {
		console.log(clc.magenta('--- Benchmarks ---'));
		nextTrial();
	}, function (err) {
		console.log(clc.red(err && err.stack || err));
		exit(1);
	});
}());

function getTrials() {
	if (process.argv.length === 2) {return require('./trials').default.map(addSearchTerms);}
	return process.argv.slice(2).reduce(filterByArgs, require('./trials').searchable.map(addSearchTerms));
	
	function addSearchTerms(trial) {
		var size = trial.table.toLowerCase().indexOf('large') === -1 ? 'small' : 'large';
		var columns = trial.columns.join(',').toLowerCase();
		if (trial.columns.length > 1) {columns = '(' + columns + ')';}
		trial.terms = [trial.type.toLowerCase(), size, columns];
		trial.looseTerms = (trial.pragma || []).filter(customPragma).join('; ').toLowerCase();
		return trial;
	}
	function filterByArgs(trials, arg) {
		arg = arg.toLowerCase();
		return trials.filter(function (obj) {
			return obj.terms.indexOf(arg) !== -1 || obj.looseTerms.indexOf(arg) !== -1;
		});
	}
	function customPragma(str) {
		return str.indexOf('cache_size') === -1;
	}
}

function exit(code) {
	fs.removeSync('temp/');
	process.exit(typeof code === 'number' ? code : 0);
}

function nextTrial() {
	if (!trials.length) {
		console.log(clc.green('All benchmarks complete!'));
		return exit();
	}
	
	var trial = trials.shift();
	var extraName = trial.looseTerms ? clc.yellow(' | ' + trial.looseTerms) : '';
	console.log(clc.cyan(trial.terms.join(' ')) + extraName);
	
	var child = spawn('node', [path.join(__dirname, 'types', trial.type), JSON.stringify(trial)], {stdio: 'inherit'});
	child.on('exit', function (code) {
		if (code !== 0) {
			console.log(clc.red('ERROR (probably out of memory)'));
		}
		setTimeout(nextTrial, 0);
	});
}