kafka-node/lib/wrapper/BrokerWrapper.js

Client for Apache Kafka v0.9.x, v0.10.x and v0.11.x

'use strict';

var BrokerReadable = require('./BrokerReadable');
var BrokerTransform = require('./BrokerTransform');

const util = require('util');
const EventEmitter = require('events');

var BrokerWrapper = function (socket, noAckBatchOptions, idleConnectionMs, needAuthentication) {
  EventEmitter.call(this);
  this.socket = socket;
  this.idleConnectionMs = idleConnectionMs;
  this.needAuthentication = !!needAuthentication;
  this.authenticated = false;

  var self = this;
  var readable = new BrokerReadable();
  var transform = new BrokerTransform(noAckBatchOptions);

  readable.pipe(transform);

  transform.on('readable', function () {
    var bulkMessage = null;
    self._lastWrite = Date.now();
    while ((bulkMessage = transform.read())) {
      // eslint-disable-line no-cond-assign
      self.socket.write(bulkMessage);
    }
  });

  this.readableSocket = readable;
};

util.inherits(BrokerWrapper, EventEmitter);

BrokerWrapper.prototype.getReadyEventName = function () {
  const lp = this.socket.longpolling ? '-longpolling' : '';
  return `${this.socket.addr}${lp}-ready`;
};

BrokerWrapper.prototype.isConnected = function () {
  return !this.socket.destroyed && !this.socket.closing && !this.socket.error;
};

BrokerWrapper.prototype.isReady = function () {
  return this.apiSupport != null && (!this.needAuthentication || this.authenticated);
};

BrokerWrapper.prototype.isIdle = function () {
  return Date.now() - this._lastWrite >= this.idleConnectionMs;
};

BrokerWrapper.prototype.write = function (buffer) {
  this._lastWrite = Date.now();
  this.socket.write(buffer);
};

BrokerWrapper.prototype.writeAsync = function (buffer) {
  this.readableSocket.push(buffer);
};

BrokerWrapper.prototype.toString = function () {
  return `[${this.constructor.name} ${
    this.socket.addr
  } (connected: ${this.isConnected()}) (ready: ${
    this.isReady()
  }) (idle: ${
    this.isIdle()
  }) (needAuthentication: ${
    this.needAuthentication
  }) (authenticated: ${
    this.authenticated
  })]`;
};

module.exports = BrokerWrapper;