mobility-toolbox-js 1.7.8-beta.3 → 1.7.8-beta.6

package/api/tralis/TralisAPI.js

@@ -77,7 +77,7 @@ class TralisAPI {
     }
 
     const { apiKey } = opt;
-    let { url, projection, bbox } = opt;
+    let { url, projection, bbox, buffer = [100, 100] } = opt;
     const conn = new WebSocketConnector();
 
     if (apiKey) {
@@ -114,6 +114,17 @@ class TralisAPI {
           }
         },
       },
+      buffer: {
+        get: () => buffer,
+        set: (newBuffer) => {
+          if (JSON.stringify(newBuffer) !== JSON.stringify(buffer)) {
+            buffer = newBuffer;
+            if (this.conn) {
+              this.conn.send(`BUFFER ${buffer.join(' ')}`);
+            }
+          }
+        },
+      },
       /**
        * The websocket helper class to connect the websocket.
        *
@@ -178,10 +189,15 @@ class TralisAPI {
     if (this.projection) {
       this.conn.send(`PROJECTION ${this.projection}`);
     }
+
     if (this.bbox) {
       this.conn.send(`BBOX ${this.bbox.join(' ')}`);
     }
 
+    if (this.buffer) {
+      this.conn.send(`BUFFER ${this.buffer.join(' ')}`);
+    }
+
     /**
      * Keep websocket alive
      */

package/api/tralis/WebSocketConnector.js

@@ -172,16 +172,27 @@ class WebSocketConnector {
     this.unlisten(params, cb);
 
     const onMessage = (evt) => {
-      const data = JSON.parse(evt.data);
+      let data = {};
+      try {
+        data = JSON.parse(evt.data);
+      } catch (err) {
+        // eslint-disable-next-line no-console
+        console.error('WebSocket: unable to parse JSON data', err, evt.data);
+      }
       let source = params.channel;
       source += params.args ? ` ${params.args}` : '';
 
-      if (
-        data.source === source &&
-        (!params.id || params.id === data.client_reference)
-      ) {
-        cb(data);
-      }
+      // Buffer channel message return a list of other channels to propagate to proper callbacks.
+      const contents = data.source === 'buffer' ? data.content : [data];
+      contents.forEach((content) => {
+        // Because of backend optimization, the last content is null.
+        if (
+          content?.source === source &&
+          (!params.id || params.id === data.client_reference)
+        ) {
+          cb(content);
+        }
+      });
     };
 
     if (this.websocket) {