@sex-pomelo/sex-pomelo/lib/connectors/mqttsocket.js

[![NPM version][npm-image-pomelo]][npm-url-pomelo] [![NPM version][npm-image-down]][npm-url-pomelo]

"use strict";

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

const ST_INITED = 1;
const ST_CLOSED = 2;

/**
 * Socket class that wraps socket and websocket to provide unified interface for up level.
 * 
 * @class
 * @constructor
 * @memberof MqttConnector
 */
class Socket extends EventEmitter{

  constructor(id, socket, adaptor) {
    super();

    this.id = id;
    this.socket = socket;
    this.remoteAddress = {
      ip: socket.stream.remoteAddress,
      port: socket.stream.remotePort
    };
    this.adaptor = adaptor;
  
    let self = this;
  
    socket.on('close', this.emit.bind(this, 'disconnect'));
    socket.on('error', this.emit.bind(this, 'disconnect'));
    socket.on('disconnect', this.emit.bind(this, 'disconnect'));
  
    socket.on('pingreq', function(packet) {
      socket.pingresp();
    });
  
    socket.on('subscribe', this.adaptor.onSubscribe.bind(this.adaptor, this));
  
    socket.on('publish', this.adaptor.onPublish.bind(this.adaptor, this));
  
    this.state = ST_INITED;
  
    // TODO: any other events?
  }

  send (msg) {
    if(this.state !== ST_INITED) {
      return;
    }
    if(msg instanceof Buffer) {
      // if encoded, send directly
      this.socket.stream.write(msg);
    } else {
      this.adaptor.publish(this, msg);
    }
  }
  
  sendBatch(msgs) {
    for(let i = 0, l = msgs.length; i<l; i++) {
      this.send(msgs[i]);
    }
  }
  
  disconnect () {
    if(this.state === ST_CLOSED) {
      return;
    }
  
    this.state = ST_CLOSED;
    this.socket.stream.destroy();
  }

}


module.exports = Socket;