statebus 6.1.21 → 6.1.22

package/client.js

@@ -62,7 +62,7 @@
         var preprefix = prefix.slice(0,-1)
         var has_prefix = new RegExp('^' + preprefix)
         var is_absolute = /^https?:\/\//
-        var client_fetched_keys = new bus.Set()
+        var subscriptions = {}
         //if (url[url.length-1]=='/') url = url.substr(0,url.length-1)
         function add_prefix (key) {
             return is_absolute.test(key) ? key : preprefix + key }
@@ -91,27 +91,85 @@
         }
 
         bus(prefix).to_fetch  = function (key, t) {
-            try {
-                braid_fetch(url + rem_prefix(key),
-                            {
-                                method: 'get',
-                                subscribe: true,
-                                headers: {accept: 'application/json'}
-                            }
-                           ).andThen( function (x) {
-                               t.return({
-                                   key: key,
-                                   val: add_prefixes(JSON.parse(x.body))
-                               })
-                           })
-            } catch (e) {
-                console.error('Braid-HTTP network error:', e)
+            // Subscription can be in states:
+            // - connecting
+            // - connected
+            // - reconnect
+            // - reconnecting
+            // - aborted
+
+            // If we have an outstanding fetch running, then let's tell it to
+            // re-activate!
+            if (subscriptions[key]) {
+                // We should only be here if an existing subscription was
+                // aborted, but hasn't cleared yet.
+                console.assert(subscriptions[key].status === 'aborted',
+                               'Refetching a subscription of status '
+                               + subscriptions[key].status)
+
+                console.trace('foo')
+
+                // Let's tell it to reconnect when it tries to clear!
+                subscriptions[key].status = 'reconnect'
+            }
+
+            // Otherwise, create a new subscription
+            else
+                subscribe (key, t)
+
+            function subscribe (key, t) {
+                var aborter = new AbortController(),
+                    reconnect_attempts = 0
+
+                // Start the subscription!
+                braid_fetch(
+                    // URL
+                    url + rem_prefix(key),
+
+                    // Options
+                    {
+                        method: 'get',
+                        subscribe: true,
+                        headers: {accept: 'application/json'},
+                        signal: aborter.signal
+                    }
+                ).andThen( function (x) {
+                    // New update received!
+                    if (subscriptions[key].status === 'connecting') {
+                        console.log('%c[*] opened ' + key,
+                                    'color: blue')
+                        reconnect_attempts = 0
+                        subscriptions[key].status = 'connected'
+                    }
+
+                    // Return the update
+                    t.return({
+                        key: key,
+                        val: add_prefixes(JSON.parse(x.body))
+                    })
+                }).catch( function (e) {
+                    if (subscriptions[key].status === 'aborted') {
+                        // Then this fetch is over and done with!
+                        delete subscriptions[key]
+                        return
+                    }
+
+                    // Reconnect!
+                    setTimeout(function () { subscribe(key, t) },
+                               reconnect_attempts > 0 ? 5000 : 1500)
+                    subscriptions[key].status = 'reconnecting'
+                })
+
+                // Remember this subscription
+                subscriptions[key] = {
+                    aborter: aborter,
+                    status: 'connecting'
+                }
             }
-            client_fetched_keys.add(key)
         }
         bus(prefix).to_forget = function (key) {
-            // send({forget: key}),
-            client_fetched_keys.delete(key)
+            subscriptions[key].status = 'aborted'
+            subscriptions[key].aborter.abort()
         }
     }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "statebus",
-  "version": "6.1.21",
+  "version": "6.1.22",
   "description": "Synchronize state.",
   "main": "statebus.js",
   "scripts": {

package/server.js

@@ -219,24 +219,25 @@ function import_server (bus, options)
             cbus.fetch(key, cb = (o) => {
                 var body = bus.to_http_body(o)
 
-                // // Return 404 if the value is undefined
-                // if (!body) {
-                //     console.log('no body! time to 404')
-                //     res.statusCode = 404
-                //     res.end()
-                // }
-
-                // Or if we're braid, send via subscription
-                // else
+                // If we're braiding, send via subscription
                 if (braidify)
+                    // Note: if body === undefined, we need to send the
+                    // equivalent of a 404.  This is missing in braid spec:
+                    // https://github.com/braid-org/braid-spec/issues/110
                     res.sendVersion({body: body || 'null'})
 
                 // Or just return the current version
-                else
-                    res.send(body)
+                else {
+                    if (body !== undefined)
+                        res.send(body)
+                    else {
+                        res.statusCode = 404
+                        res.end()
+                    }
+                }
 
-                // And shut down the connection if nothing's left to do
-                if (!braidify || !req.subscribe || !body)
+                // And shut down the connection if there's no subscription
+                if (!braidify || !req.subscribe)
                     end_it_all()
             })
         }
@@ -587,6 +588,7 @@ function import_server (bus, options)
             }
         })
 
+        // console.log('websocket listening on', '/' + bus.options.websocket_path)
         s.installHandlers(httpserver, {prefix:'/' + bus.options.websocket_path})
     },
 
@@ -595,7 +597,7 @@ function import_server (bus, options)
         url = url.replace(/^istate:\/\//, 'ws://')
         url = url.replace(/^statei:\/\//, 'ws://')
         WebSocket = require('websocket').w3cwebsocket
-        return new WebSocket(url+'/'+bus.options.websocket_path+'/websocket')
+        return new WebSocket(url + '/' + bus.options.websocket_path + '/websocket')
     },
     client_creds: function client_creds (server_url) {
         // Right now the server just creates a different random id each time