ws-reconnect-js/lib/index.js

websocket client that reconnect

var WSRECONNECTCLIENT = function(url, options, handlers) {
	if(!url){
		throw new Error("Please define dest url");
	}
	this.url 					= url && url.indexOf("ws") == -1 ? "ws://"+url : url;
	this.options 				= options || {};
	this.socket 				= null;	
	this.isConnected 			= false;
	this.reconnectTimeoutId 	= 0;
	this.retryCount 			= this.options.retryCount || 2;
	this._retryCount            = this.retryCount;
	this.reconnectInterval 		= this.options.reconnectInterval !== undefined ? this.options.reconnectInterval : 5;
	this.shouldAttemptReconnect = !!this.reconnectInterval;
	// handlers
	this.onMessageHandler 		= handlers.onMessageHandler || function () {};
		
	
	
    // METHODS
    this.start = function(){
        this.shouldAttemptReconnect = !!this.reconnectInterval;
        this.isConnected 		 	= false;
        this.socket 			 	= new WebSocket(this.url);
        this.socket.onmessage 	 	= this.onMessage.bind(this);
        this.socket.onopen 		 	= this.onOpen.bind(this);
        this.socket.onerror		 	= this.onError.bind(this);
        this.socket.onclose 	 	= this.onClose.bind(this);
    }
    
    this.destroy = function(){
        clearTimeout(this.reconnectTimeoutId);
	    this.shouldAttemptReconnect = false;
        this.socket.close();
    }
    
    this.onError = function(reason){
        // hook before close 
    }
    
    this.onOpen = function(){
        this.isConnected 	= true;
	    // set again the retry count
        this.retryCount = this._retryCount;
    }
    
    this.onClose = function(reason) {
        if (this.shouldAttemptReconnect && this.retryCount > 0) {
            this.retryCount--;
            clearTimeout(this.reconnectTimeoutId);
            this.reconnectTimeoutId = setTimeout(function(){
                this.start();
            }.bind(this), this.reconnectInterval*1000);
        }
    }
    
    this.onMessage = function(message) {
	   this.onMessageHandler("message",message.data);
    };

    
};

module.exports = WSRECONNECTCLIENT;