From 0f203354c751cad94cc1cecb7460740c7f5ad24d Mon Sep 17 00:00:00 2001
From: Dima Buzenkov <dimon@buzenkov.local>
Date: Sat, 3 Jun 2017 20:56:27 +0300
Subject: [PATCH] Added supported ws@3.0.0 Remove the `upgradeReq` from socket
 https://github.com/websockets/ws/pull/1099

---
 src/index.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/index.js b/src/index.js
index 3207dbd..dc6059d 100644
--- a/src/index.js
+++ b/src/index.js
@@ -42,8 +42,10 @@ export default function expressWs(app, httpServer, options = {}) {
   wsOptions.server = server;
   const wsServer = new ws.Server(wsOptions);
 
-  wsServer.on('connection', (socket) => {
-    const request = socket.upgradeReq;
+  wsServer.on('connection', (socket, request) => {
+    if ('upgradeReq' in socket) {
+      request = socket.upgradeReq;
+    }
 
     request.ws = socket;
     request.wsHandled = false;