fix(client): connection close during handshaking should not throw an …

…error
litert · Mar 22, 2024 · 4163ee9 · 4163ee9
1 parent 7e4324b
commit 4163ee9
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -1,5 +1,9 @@
 # Changes
 
+## v0.2.1
+
+- fix(client): connection close during handshaking should not throw an error.
+
 ## v0.2.0
 
 - build(project): use term `message` instead of `frame`.

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@litert/websocket",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "The websocket protocol library for LiteRT.",
   "main": "lib/index.js",
   "scripts": {

diff --git a/src/lib/Client.ts b/src/lib/Client.ts
@@ -222,6 +222,10 @@ function connect(
 
             resolve(ws);
         })
+            .on('close', () => {
+
+                reject(new E.E_HANDSHAKE_FAILED('Connection closed'));
+            })
             .on('error', (e) => {
 
                 reject(e);