David/staging2

cli/lib/nf3.mjs

@@ -666,9 +666,17 @@ class Nf3 {
     const newGasBlockEmitter = new EventEmitter();
     const connection = new WebSocket(this.optimistWsUrl);
     this.websockets.push(connection); // save so we can close it properly later
+    // Ping function to keep WS open. Send beat every 15 seconds
+    const ping = async () => {
+      if (!connection) return;
+      if (connection.readyState !== WebSocket.OPEN) return;
+      connection.ping();
+      setTimeout(ping, 15000);
+    };
     connection.onopen = () => {
       logger.debug('websocket connection opened');
       connection.send('blocks');
+      ping();
     };
     connection.onmessage = async message => {
       const msg = JSON.parse(message.data);

nightfall-optimist/src/event-handlers/subscribe.mjs

@@ -72,28 +72,55 @@ export async function startEventQueue(callback, ...arg) {
 }
 
 export async function subscribeToChallengeWebSocketConnection(callback, ...args) {
-  wss.on('connection', ws =>
+  wss.on('connection', ws => {
     ws.on('message', message => {
       if (message === 'challenge') callback(ws, args);
-    }),
-  );
+    });
+    ws.on('error', () => {
+      logger.debug('ERROR challenge WS');
+    });
+    ws.on('open', () => {
+      logger.debug('OPEN challenge WS');
+    });
+    ws.on('close', err => {
+      logger.debug(`CLOSE challenge WS: ${err}`);
+    });
+  });
   logger.debug('Subscribed to Challenge WebSocket connection');
 }
 
 export async function subscribeToBlockAssembledWebSocketConnection(callback, ...args) {
-  wss.on('connection', ws =>
+  wss.on('connection', ws => {
     ws.on('message', message => {
       if (message === 'blocks') callback(ws, args);
-    }),
-  );
+    });
+    ws.on('error', () => {
+      logger.debug('ERROR block-assembly  WS');
+    });
+    ws.on('open', () => {
+      logger.debug('OPEN block-assembly WS');
+    });
+    ws.on('close', msg => {
+      logger.debug(`CLOSE block-assembly ${msg}`);
+    });
+  });
   logger.debug('Subscribed to BlockAssembled WebSocket connection');
 }
 
 export async function subscribeToInstantWithDrawalWebSocketConnection(callback, ...args) {
-  wss.on('connection', ws =>
+  wss.on('connection', ws => {
     ws.on('message', message => {
       if (message === 'instant') callback(ws, args);
-    }),
-  );
+    });
+    ws.on('error', () => {
+      logger.debug('ERROR instant-withdraw');
+    });
+    ws.on('open', () => {
+      logger.debug('OPEN instant-withdraw');
+    });
+    ws.on('close', err => {
+      logger.debug(`CLOSE instant-withdraw ${err}`);
+    });
+  });
   logger.debug('Subscribed to InstantWithDrawal WebSocket connection');
 }

nightfall-optimist/src/services/block-assembler.mjs

@@ -58,7 +58,9 @@ export async function conditionalMakeBlock(proposer) {
           transactions.map(t => Transaction.buildSolidityStruct(t)),
         )
         .encodeABI();
-      if (ws)
+      // TODO - check ws readyState is OPEN => CLOSED .WebSocket.OPEN(1), CONNECTING(0), CLOSING(2), CLOSED(3)
+      //  before sending Poposed block. If not Open, try to open it
+      if (ws) {
         await ws.send(
           JSON.stringify({
             type: 'block',
@@ -67,7 +69,8 @@ export async function conditionalMakeBlock(proposer) {
             transactions,
           }),
         );
-      logger.debug('Send unsigned block-assembler transaction to ws client');
+        logger.debug('Send unsigned block-assembler transaction to ws client');
+      }
       // remove the transactiosn from the mempool so we don't keep making new
       // blocks with them
       await removeTransactionsFromMemPool(block.transactionHashes);