riemann-dash 0.2.1 → 0.2.3

data/lib/riemann/dash/public/subs.js

@@ -1,4 +1,5 @@
 var subs = (function() {
+
   // What server shall we connect to by default?
   var server;
 
@@ -25,18 +26,13 @@ var subs = (function() {
 
   // Close a subscription's websocket channel.
   var close = function(sub) {
-    if (sub.ws == null) {
-      return sub;
-    }
-    sub.ws.close();
-    sub.ws == null;
-    return sub;
+    return sub.close();
   }
 
   // Closes a subscription and deletes it from the subscription manager.
   var unsubscribe = function(sub) {
     delete subs[sub.id];
-    close(sub);
+    return sub.close();
   }
 
   // Unsubscribe from all subscriptions.
@@ -46,61 +42,80 @@ var subs = (function() {
 
   // Open a subscription's websocket channel.
   var open = function(sub) {
-    if (sub.ws != null && sub.ws.readyState != WebSocket.CLOSED) {
-      return sub;
-    }
+    return sub.open();
+  }
 
-    var f = sub.f;
-    var queryString = "query=" + encodeURI(sub.query);
-    var uri = "ws://" + server + "/index?subscribe=true&" + queryString;
-    sub.ws = new WebSocket(uri);
-    var $ws = $(sub.ws);
-    
-    $ws.bind('open', function() {
-      console.log("Socket opened", sub.query);
-    });
+  var Subscription = Backbone.Model.extend({
 
-    $ws.bind('close', function(e) {
-      console.log("Socket closed", sub.query);
-      sub.ws = null;
-    });
+    initialize: function(id, query, f) {
+      this.id = id;
+      this.query = query;
+      this.f = f;
+    },
 
-    $ws.bind('error', function(e) {
-      console.log("Socket error", sub.query);
-      errorQueue.push(e);
-      ws.close();
-    });
+    isOpen: function() {
+      return this.ws && (this.ws.readyState != WebSocket.CLOSED)
+    },
+    isClosed: function() { return !this.isOpen() },
 
-    $ws.bind('message', function(e) {
-      t1 = Date.now();
-      if (active) {
-        f(JSON.parse(e.originalEvent.data));
-      }
-      load1(t1, Date.now());
-      load5(t1, Date.now());
-    });
+    url: function() {
+      var queryString = "query=" + encodeURI(this.query);
+      return "ws://" + server + "/index?subscribe=true&" + queryString;
+    },
 
-    return sub;
-  }
+    open: function() {
+      if (this.isOpen()) return this;
+      var ws = this.ws = new WebSocket(this.url());
+
+      ws.onopen = _.bind(function() {
+        console.log("Socket opened", this.query);
+      }, this);
+
+      ws.onclose = _.bind(function(e) {
+        console.log("Socket closed", this.query);
+        this.ws = null;
+      }, this);
+
+      ws.onerror = _.bind(function(e) {
+        console.log("Socket error", this.query);
+        errorQueue.push(e);
+        this.close();
+      }, this);
+
+      ws.onmessage = _.bind(function(e) {
+        t1 = Date.now();
+        if (active) {
+          this.f(JSON.parse(e.data));
+        }
+        load1(t1, Date.now());
+        load5(t1, Date.now());
+      }, this);
+
+      return this;
+
+    },
+
+    close: function() {
+      if (this.ws) {
+        this.ws.close();
+        this.ws = void 0;
+      }
+      return this;
+    }
+  });
 
   // Add a subscription. Returns a subscription object. Subscriptions are
   // opened immediately.
   var subscribe = function(query, f) {
-    var sub = {
-      id: newId(),
-      query: query,
-      f: f,
-      ws: null
-    }
+    var sub = new Subscription(newId(), query, f).open();
     subs[sub.id] = sub;
-    open(sub);
     return sub;
   }
 
   // Reconnect all inactive subs.
   var converge = function() {
     var closed = _.filter(subs, function(sub) {
-      return (sub.ws == null || sub.ws.readyState == WebSocket.CLOSED);
+      return sub.isClosed();
     });
     if (_.isEmpty(closed)) {
       // Done here.
@@ -108,7 +123,7 @@ var subs = (function() {
     }
   
     // Display reconnection notice
-    toastr.warning(_.size(closed) + " lost connections");
+    toastr.warning(_.size(closed) + " lost connections—check the server field above.");
 
     // Reopen
     _.each(closed, function(sub) {
@@ -120,7 +135,7 @@ var subs = (function() {
     if (errorQueue.length == 0) {
       return;
     }
-    _.warning(errorQueue.length + " socket errors");
+    toastr.warning(errorQueue.length + " socket errors");
     errorQueue.length = 0;
     converge();
   }

data/lib/riemann/dash/public/util.js

@@ -1,53 +1,46 @@
 var util = (function() {
-  // Unique IDs
-  var uniqueId = function(bytes) {
-    bytes = bytes || 20;
-    var s = '';
-    for (var i = 0; i < bytes * 2; i++) {
-      s += Math.floor(Math.random() * 16).toString(16);
-    }
-    return s;
-  }
-
-  // Merge two maps nondestructively.
-  var merge = function(m1, m2) {
-    return _.extend(_.clone(m1), m2);
-  }
+  return {
+    merge: function(m1, m2) {
+      // Merge two maps nondestructively.
+      return _.extend({}, m1, m2)
+    },
+    slur: function(period, f) {
+      // Wraps a function in another, which calls f at most once every period
+      // milliseconds. Tries to minimize latency.
 
-  // Wraps a function in another, which calls f at most once every period
-  // milliseconds. Tries to minimize latency.
-  var slur = function(period, f) {
-    var lastRun = new Date();
-    lastRun.setYear(0);
-    var queued = false;
-    var execute = function(context, args) {
-      f.apply(context, args);
-      lastRun = new Date();
-      queued = false;
-    };
+      var lastRun = new Date();
+      lastRun.setYear(0);
+      var queued = false;
+      var execute = function(context, args) {
+        f.apply(context, args);
+        lastRun = new Date();
+        queued = false;
+      };
 
-    return function() {
-      // If queued, do nothing
-      if (queued) {
-        return;
-      }
+      return function() {
+        // If queued, do nothing
+        if (queued) {
+          return;
+        }
 
-      var dt = (new Date()) - lastRun;
-      if (period <= dt) {
-        // We're free to go
-        execute(this, arguments);
-      } else {
-        // Too soon, enqueue a new job.
-        window.setTimeout(execute, period - dt, this, arguments);
+        var dt = (new Date()) - lastRun;
+        if (period <= dt) {
+          // We're free to go
+          execute(this, arguments);
+        }
+        else {
+          // Too soon, enqueue a new job.
+          window.setTimeout(execute, period - dt, this, arguments);
+        }
       }
+    },
+    uniqueId: function(length) {
+      // Unique-ish IDs as a length sized string of hex
+      var id = '', hex = '0123456789abcdef';
+      _(length || 40).times(function() { id += hex[_.random(15)]; });
+      return id;
     }
-  }
-
-  return {
-    merge: merge,
-    slur: slur,
-    uniqueId: uniqueId
-  }
+  };
 })();
 
 $(function() {