CloseEvent error handling

hivenapp · Nov 12, 2019 · 12f26ee · 12f26ee
1 parent f49bd1e
commit 12f26ee
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/build/swarm.js b/build/swarm.js
@@ -56,8 +56,16 @@ var SwarmClient = /** @class */ (function (_super) {
             }
         });
         socket.addEventListener('close', function (event) {
-            _this.state = "closed";
-            _this.emit("SOCKET_STATE_CHANGE", "closed");
+            switch (event.code) {
+                case 4003: {
+                    _this.state = event.reason;
+                    _this.emit("SOCKET_STATE_CHANGE", event.reason);
+                }
+                default: {
+                    _this.state = "closed";
+                    _this.emit("SOCKET_STATE_CHANGE", "closed");
+                }
+            }
         });
     };
     SwarmClient.prototype.sendRaw = function (data) {

diff --git a/swarm.ts b/swarm.ts
@@ -61,8 +61,16 @@ export class SwarmClient extends EventEmitter {
     });
 
     socket.addEventListener('close', (event) => {
-      this.state = "closed";
-      this.emit("SOCKET_STATE_CHANGE", "closed");
+      switch(event.code) {
+        case 4003: {
+          this.state = event.reason;
+          this.emit("SOCKET_STATE_CHANGE", event.reason);
+        }
+        default: {
+          this.state = "closed";
+          this.emit("SOCKET_STATE_CHANGE", "closed");
+        }
+      }
     });
   }