@sex-pomelo/sex-pomelo/lib/connectors/siosocket.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 = 0;
const ST_CLOSED = 1;

/**
 * Socket class that wraps socket.io socket to provide unified interface for up level.
 * 
 * @class
 * @constructor
 * @memberof SioConnector
 */

class Socket extends EventEmitter{
  constructor(id, socket) {
    super();

    this.id = id;
    this.socket = socket;
    this.remoteAddress = {
      ip: socket.handshake.address.address,
      port: socket.handshake.address.port
    };
  
    let self = this;
  
    socket.on('disconnect', this.emit.bind(this, 'disconnect'));
  
    socket.on('error', this.emit.bind(this, 'error'));
  
    socket.on('message', function(msg) {
      self.emit('message', msg);
    });
  
    this.state = ST_INITED;
  
    // TODO: any other events?
  };

  send(msg) {
    if(this.state !== ST_INITED) {
      return;
    }
    if(typeof msg !== 'string') {
      msg = JSON.stringify(msg);
    }
    this.socket.send(msg);
  }
  
  disconnect () {
    if(this.state === ST_CLOSED) {
      return;
    }
  
    this.state = ST_CLOSED;
    this.socket.disconnect();
  }
  
  sendBatch(msgs) {
    this.send(encodeBatch(msgs));
  }


}






module.exports = Socket;



/**
 * Encode batch msg to client
 * @access private
 */
let encodeBatch = function(msgs){
  let res = '[', msg;
  for(let i=0, l=msgs.length; i<l; i++) {
    if(i > 0) {
      res += ',';
    }
    msg = msgs[i];
    if(typeof msg === 'string') {
      res += msg;
    } else {
      res += JSON.stringify(msg);
    }
  }
  res += ']';
  return res;
};