hub-proxy: close when receive error event

Author: williamstein

src/packages/hub/proxy/handle-upgrade.ts

@@ -109,6 +109,7 @@ export default function init(
     proxy.on("error", (err) => {
       logger.debug(`websocket proxy error, so clearing cache -- ${err}`);
       cache.delete(target);
+      proxy.close();
     });
     proxy.on("close", () => {
       dbg("websocket proxy closed, so removing from cache");

src/packages/hub/proxy/target.ts

@@ -28,7 +28,14 @@ const logger = getLogger("proxy:target");
 // Also, if the project stops and starts, the host=ip address could
 // change, so we need to timeout so we see that thange.
 
-const cache = new LRU({ max: 20000, ttl: 1000 * 30 });
+const cache = new LRU<
+  string,
+  {
+    host: string;
+    port: number;
+    internal_url: string | undefined;
+  }
+>({ max: 20000, ttl: 1000 * 30 });
 
 // This gets explicitly called from outside when certain errors occur.
 export function invalidateTargetCache(remember_me: string, url: string): void {

src/packages/hub/servers/nats.ts

@@ -38,6 +38,10 @@ export async function proxyNatsWebsocket(req, socket, head) {
     timeout: 5000,
   });
   proxy.ws(req, socket, head);
+  proxy.on("error", (err) => {
+    logger.debug(`nats websocket proxy error, so closing -- ${err}`);
+    proxy.close();
+  });
 
   while (socket.readyState !== socket.CLOSED) {
     if (versionCheckFails(req)) {