uriel/app/lib/statsFactory.js

A NodeJS statsd agent to push memory, cpu, network, swap, diskio, and disk usage over UDP

// app/lib/statsFactory.js
const snakeCase = require('lodash.snakecase');
const isNil = require('lodash.isnil');

class StatsFactory {
  constructor(hostname, statsd, tags, log) {
    this.hostname = hostname || 'localhost';
    this.statsd = statsd;
    this.log = log;
    if (!tags) {
      this.tags = [];
    } else if (Array.isArray(tags)) {
      // Make a copy of the tags array (the original array should remain immutable)
      this.tags = Object.assign([], tags);
    } else {
      this.tags = [tags];
    }
    this.tags.push('serverName:' + this.hostname);
  }

  create(name, value, tags) {
    tags = tags || [];
    name = name.split('.')
      .map((str) => {
        return snakeCase(str);
      })
      .join('.');

    const retVal = {
      name: name,
      value: value,
      tags: tags
    };

    return retVal;
  }

  send(collection) {
    if (!Array.isArray(collection)) {
      if (typeof collection !== 'object') {
        throw new Error('Can not send a value');
      } else {
        collection = [collection];
      }
    }

    // Only send if the statsd connection has not been shut down
    if (!isNil(this.statsd)) {
      for (let itr = 0, jtr = collection.length; itr < jtr; itr++) {
        let tempTags = Object.assign([], this.tags);
        const stat = collection[itr];
        this.log.silly('Sending statistic %s = %d', stat.name, stat.value);

        if (typeof stat !== 'object' ||
          !Object.prototype.hasOwnProperty.call(stat, 'name') ||
          !Object.prototype.hasOwnProperty.call(stat, 'value')) {
          continue;
        }

        if (Array.isArray(stat.tags) && stat.tags.length !== 0) {
          tempTags = tempTags.concat(stat.tags);
        }

        this.statsd.gauge(stat.name, stat.value, tempTags, (error) => {
          if (error) {
            this.log.error(error.stack || error);
          } else {
            this.log.silly('Success');
          }
        });
      }
    }
  }
}

module.exports = StatsFactory;