mirror of https://github.com/seald/nedb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
2.9 KiB
124 lines
2.9 KiB
/**
|
|
* Functions that are used in several benchmark tests
|
|
*/
|
|
|
|
var customUtils = require('../lib/customUtils')
|
|
, fs = require('fs')
|
|
, path = require('path')
|
|
;
|
|
|
|
|
|
/**
|
|
* Ensure the workspace exists and the db is empty
|
|
*/
|
|
module.exports.prepareDb = function (filename, cb) {
|
|
customUtils.ensureDirectoryExists(path.dirname(filename), function () {
|
|
fs.exists(filename, function (exists) {
|
|
if (exists) {
|
|
fs.unlink(filename, cb);
|
|
} else { return cb(); }
|
|
});
|
|
});
|
|
};
|
|
|
|
|
|
/**
|
|
* Return an array with the numbers from 0 to n-1, in a random order
|
|
* Useful to get fair tests
|
|
*/
|
|
function getRandomArray (n) {
|
|
var res, next;
|
|
|
|
if (n === 0) { return []; }
|
|
if (n === 1) { return [0]; }
|
|
|
|
res = getRandomArray(n - 1);
|
|
next = Math.floor(Math.random() * n);
|
|
res.splice(next, 0, n - 1); // Add n-1 at a random position in the array
|
|
|
|
return res;
|
|
};
|
|
module.exports.getRandomArray = getRandomArray;
|
|
|
|
|
|
/**
|
|
* Insert a certain number of documents for testing
|
|
*/
|
|
module.exports.insertDocs = function (d, n, profiler, cb) {
|
|
var beg = new Date()
|
|
;
|
|
|
|
profiler.step('Begin inserting ' + n + ' docs');
|
|
|
|
function runFrom(i) {
|
|
if (i === n) { // Finished
|
|
console.log("Average time for one insert: " + (profiler.elapsedSinceLastStep() / n) + "ms");
|
|
profiler.step('Finished inserting ' + n + ' docs');
|
|
return cb();
|
|
}
|
|
|
|
d.insert({ docNumber: i }, function (err) {
|
|
process.nextTick(function () {
|
|
runFrom(i + 1);
|
|
});
|
|
});
|
|
}
|
|
runFrom(0);
|
|
};
|
|
|
|
|
|
/**
|
|
* Find documents with find
|
|
*/
|
|
module.exports.findDocs = function (d, n, profiler, cb) {
|
|
var beg = new Date()
|
|
, order = getRandomArray(n)
|
|
;
|
|
|
|
profiler.step("Finding " + n + " documents");
|
|
|
|
function runFrom(i) {
|
|
if (i === n) { // Finished
|
|
console.log("Average time for one find in a collection of " + n + " docs: " + (profiler.elapsedSinceLastStep() / n) + "ms");
|
|
profiler.step('Finished finding ' + n + ' docs');
|
|
return cb();
|
|
}
|
|
|
|
d.find({ docNumber: order[i] }, function (err, docs) {
|
|
if (docs.length !== 1 || docs[0].docNumber !== order[i]) { return cb('One find didnt work'); }
|
|
process.nextTick(function () {
|
|
runFrom(i + 1);
|
|
});
|
|
});
|
|
}
|
|
runFrom(0);
|
|
};
|
|
|
|
|
|
/**
|
|
* Find documents with findOne
|
|
*/
|
|
module.exports.findOneDocs = function (d, n, profiler, cb) {
|
|
var beg = new Date()
|
|
, order = getRandomArray(n)
|
|
;
|
|
|
|
profiler.step("FindingOne " + n + " documents");
|
|
|
|
function runFrom(i) {
|
|
if (i === n) { // Finished
|
|
console.log("Average time for one findOne in a collection of " + n + " docs: " + (profiler.elapsedSinceLastStep() / n) + "ms");
|
|
profiler.step('Finished finding ' + n + ' docs');
|
|
return cb();
|
|
}
|
|
|
|
d.findOne({ docNumber: order[i] }, function (err, doc) {
|
|
if (!doc || doc.docNumber !== order[i]) { return cb('One find didnt work'); }
|
|
process.nextTick(function () {
|
|
runFrom(i + 1);
|
|
});
|
|
});
|
|
}
|
|
runFrom(0);
|
|
};
|
|
|
|
|