feat: improve stop logic to close all open http/2 connections

FireBlinkLTD · Dec 11, 2023 · 80fb4de · 80fb4de
1 parent 153744e
commit 80fb4de
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/Prxi.ts b/src/Prxi.ts
@@ -570,6 +570,10 @@ export class Prxi {
             s.destroy();
             this.sockets.delete(s);
           });
+
+          this.proxies.forEach(p => {
+            p.http2?.closeAllConnections();
+          });
         }
 
         server.close((err) => {
@@ -578,10 +582,14 @@ export class Prxi {
             return rej(err);
           }
 
+          this.proxies.forEach(p => {
+            p.http2?.closeAllConnections();
+          });
+
+          this.logInfo('Prxi stopped');
           res();
         });
       });
-      this.logInfo('Prxi stopped');
     } else {
       this.logInfo('Prxi stopping skipped, not running');
     }

diff --git a/src/handlers/Http2ProxyHandler.ts b/src/handlers/Http2ProxyHandler.ts
@@ -44,7 +44,6 @@ export class Http2ProxyHandler {
     return connection;
   }
 
-
   /**
    * Close connection
    * @param session
@@ -57,6 +56,16 @@ export class Http2ProxyHandler {
     }
   }
 
+  /**
+   * Close all connections
+   */
+  public closeAllConnections(): void {
+    this.connections.forEach((connection) => {
+      connection.close();
+    });
+    this.connections = new Map();
+  }
+
   /**
    * Proxy request
    * @param id