@@ -6,7 +6,7 @@ sessionHandler = require('./session').handler
6
6
STATISTICS_INTERVAL = 10000 # 10 seconds
7
7
CLIENT_TIMEOUT = 30000 # 30 seconds
8
8
9
- wrapSession = (conn ) ->
9
+ wrapSession = (conn , req ) ->
10
10
wrapper = new EventEmitter
11
11
wrapper .abort = -> conn .close ()
12
12
wrapper .stop = -> conn .close ()
@@ -28,17 +28,17 @@ wrapSession = (conn) ->
28
28
else
29
29
wrapper .emit ' message' , msg
30
30
31
- wrapper .headers = conn . upgradeReq .headers
31
+ wrapper .headers = req .headers
32
32
# TODO - I don't think this is the right way to get the address
33
33
wrapper .address = conn ._socket .server ._connectionKey ?
34
34
wrapper
35
35
36
36
exports .attach = (server , createAgent , options ) ->
37
37
options .prefix or= ' /websocket'
38
38
wss = new WebSocketServer {server : server, path : options .prefix , headers : options .headers }
39
- wss .on ' connection' , (conn ) ->
39
+ wss .on ' connection' , (conn , req ) ->
40
40
conn .isAlive = true
41
- sessionHandler wrapSession (conn), createAgent
41
+ sessionHandler wrapSession (conn, req ), createAgent
42
42
43
43
if !! options .trackStats
44
44
setInterval ->
0 commit comments