rocket.chat.mqtt/node_modules/aedes/benchmarks/pingpong.js

It's a MQTT Server, using redis to scale horizontally.

#! /usr/bin/env node

var mqtt = require('mqtt')
var convertHrtime = require('convert-hrtime')
var mode = require('compute-mode')
var client = mqtt.connect({ port: 1883, host: 'localhost', clean: true, keepalive: 0 })

var sent = 0
var interval = 5000
var latencies = []

function count () {
  console.log('sent/s', sent / interval * 1000)
  sent = 0
}

setInterval(count, interval)

function publish () {
  sent++
  client.publish('test', JSON.stringify(process.hrtime()), { qos: 1 })
}

function subscribe () {
  client.subscribe('test', { qos: 1 }, publish)
}

client.on('connect', subscribe)
client.on('message', publish)
client.on('message', function (topic, payload) {
  var sentAt = JSON.parse(payload)
  var diff = process.hrtime(sentAt)
  latencies.push(convertHrtime(diff).ms)
})

client.on('offline', function () {
  console.log('offline')
})

client.on('error', function () {
  console.log('reconnect!')
  client.stream.end()
})

process.on('SIGINT', function () {
  var total = latencies.reduce(function (acc, num) {
    return acc + num
  })
  console.log('total', total)
  console.log('average', total / latencies.length)
  console.log('mode', mode(latencies))
  process.exit(0)
})