#67 - Introducing hacky bus class.

sockjs · May 22, 2012 · 9b4fb19 · 9b4fb19
1 parent 4121344
commit 9b4fb19
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/sockjs.js b/lib/sockjs.js
@@ -15,6 +15,7 @@ var SockJS = function(url, dep_protocols_whitelist, options) {
     }
     that._base_url = utils.amendUrl(url);
     that._server = that._options.server || utils.random_number_string(1000);
+    that._bus = new EventEmitter(['send', 'close', 'heartbeat']);
     if (that._options.protocols_whitelist &&
         that._options.protocols_whitelist.length) {
         protocols_whitelist = that._options.protocols_whitelist;
@@ -78,6 +79,7 @@ SockJS.prototype._dispatchOpen = function() {
             that._transport_tref = null;
         }
         that.readyState = SockJS.OPEN;
+        that._startHeartbeats();
         that.dispatchEvent(new SimpleEvent("open"));
     } else {
         // The server might have been restarted, and lost track of our
@@ -97,6 +99,7 @@ SockJS.prototype._dispatchHeartbeat = function(data) {
     var that = this;
     if (that.readyState !== SockJS.OPEN)
         return;
+    that._bus.emit('heartbeat');
     that.dispatchEvent(new SimpleEvent('heartbeat', {}));
 };
 
@@ -133,6 +136,7 @@ SockJS.prototype._didClose = function(code, reason, force) {
     }
     that.readyState = SockJS.CLOSED;
 
+    that._bus.emit('close');
     utils.delay(function() {
                    that.dispatchEvent(close_event);
                 });