ws-client.js

/**
 * Module dependencies.
 */
var crypto = require('crypto');
var WebSocket = require('ws');

/**
 * ARGV Set
 */
var Size = parseInt(process.argv[2].toString().trim()); // the first argument
var MaxIter = process.argv[3] ? parseInt(process.argv[3].toString().trim()) : 20; // the second argument
var Protocol = process.argv[4] ? 'wss': 'ws' // the third argument

/**
 * WevSocketClient connect
 */
//var ServerName = 'server.info';
var ServerName = 'localhost';
var PortN = 8082
var ws = new WebSocket(Protocol + '://' + ServerName + ':' + PortN);

ws.on('open', function() {
  var start = undefined;
  var totalTime = 0;
  var i = MaxIter;

  /**
   * Download Event
   */
  ws.on('message', function(message) {
    var time = Date.now() - start;
    //*console.log(message);
    console.log('Data size: ' + Size + ', roudtrip time: ' + time + ' ms');
    totalTime += time;
    if (--i) {
      uploadStart(Size);
    } else {
      var ave = totalTime / MaxIter;
      console.log('Average roundtrip time: ' + ave + ' ms');

      /**
       * Transfer ratio: Size(Bytes) / (ave(ms) / 2(roundtrip)) * 1000 = (Bytes per Second)
       * (Bytes per Second) * 8 / 1024 / 1024= (Mbps)
       */
      console.log('Transfer ratio: ' + (Size / ave * 2000).toFixed(1) + '[Bytes per Second] = ' + (Size / ave * 0.0152587890625).toFixed(1) + '[Mbps]');
      process.exit(0);
    }
  });

  /**
   * Upload File
   */
  function uploadStart(size) {
    crypto.randomBytes(size, function(ex, buf) {
      if (ex) throw ex;
      //*console.log(buf);
      start = Date.now();
      ws.send(buf, {binary: true});
    });
  }

  uploadStart(Size);
});