gistfile1.txt

/**
 *     Copyright 2012 Couchbase, Inc.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 *   (for jsdoc)
 *   @copyright 2013 Couchbase, Inc.
 */

var Memcached = require("memcached");
var Couchbase = require("../lib/couchbase.js");
var Util = require("util");

// Minimal performance benchmarking with nanoseconds
function trackPerf(subject, completed, callback) {
  if (completed) {
    var diff = process.hrtime(completed);
    var nanoseconds = diff[0] * 1e9 + diff[1];
    var message = subject + ': benchmark took ' + nanoseconds / 1000000 + ' milliseconds (' + nanoseconds + ' nanoseconds)';
    console.log(message);
    return message;
  } else {
    var started = process.hrtime();
    callback(subject, started);
  }
}

// Globals
var maxClients = 20;
var maxConcurrency = 100;
var maxOperations=10000000;
var currentOperations = 0;

var key = "keybase";
var value = "valbase";
var counter = 0;

function genKey() {
    var ret = key + process.pid;
    ret += counter;
    counter++;
    return ret;
}

for (var i = 0; i < 10; i++) {
    value += "***";
}

/* How many operations are remaining */
var remaining=maxOperations;
var BEGIN_TIME = Date.now();

setInterval(function(arg) {
    // Get time elapsed
    var elapsedTime = (Date.now() - BEGIN_TIME) / 1000;
    var opsPerSecond = currentOperations / elapsedTime;
    var s = Util.format("Elapsed: %d; Ops=%d [%d]/S",
        elapsedTime,
        currentOperations,
        Math.round(opsPerSecond));
    for (var i = 0; i < 20; i++) {
        s+= ' ';
    }
    process.stdout.write(s + "\r");
}, 100, null);


function markOperation(err, subject, start) {
    if (err) {
        trackPerf(subject, start);
        console.error("Got error: " + err);
    }
    currentOperations++;
    remaining--;
    if (!remaining) {
        console.log("");
        console.log("Done!")
        process.exit(1);
    }
}


function scheduleMemcached(cb) {
    var curKey = genKey();
    cb.set(curKey, value, 0, function(err) {
        markOperation(err);

        cb.get(curKey, function(err, meta) {
            markOperation(err);
            scheduleMemcached(cb);
        });
    });
}

function scheduleCouchbase(cb) {
    var curKey = genKey();
    trackPerf('set', null, function(subject, start) {
        cb.set(curKey, value, {}, function(err, meta) {
            markOperation(err, subject, start);
            trackPerf('get', null, function(subject, start) {
                cb.get(curKey, function(err, meta) {
                    markOperation(err, subject, start);
                    scheduleCouchbase(cb);
                });
            });
        });
    });
}


function launchMemcachedClient() {
    var mc = new Memcached("localhost:11311");
    scheduleMemcached(mc);
}

function launchCouchbaseClient() {
    var cbOptions = {
        bucket: "default",
    };

    var cb = new Couchbase.Connection(cbOptions, function(err) {
        if (err) {
            console.log("Got error on connect: " + err);
            process.exit(1);
        } else {
            console.log("Connected!");
        }
    });
    for (var i = 0; i < maxConcurrency; i++) {
        scheduleCouchbase(cb);
    }
}



for (var i = 0; i < maxClients; i++) {
    if (process.argv[2] == "memcached") {
        //console.log("Starting Memcached");
        launchMemcachedClient();
    } else {
        //console.log("Starting Couchbase");
        launchCouchbaseClient();
    }
}

/** OUTPUT:
 mnunberg@csure:~/src/couchnode$ time node memdbench.js memcached
Elapsed: 12.706; Ops=99283 [7814]/S
Done!

real  0m12.850s
user	0m10.549s
sys	0m2.368s
mnunberg@csure:~/src/couchnode$ time node memdbench.js
Elapsed: 4.842; Ops=99850 [20622]/S
Done!

real	0m4.915s
user	0m3.356s
sys	0m1.576s
**/