Handle socket closed events.

tediousjs · Jan 5, 2012 · 886cfd5 · 886cfd5
1 parent f926f77
commit 886cfd5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/connection2.coffee b/lib/connection2.coffee
@@ -164,6 +164,8 @@ class Connection extends EventEmitter
     @socket.connect(@config.options.port, @config.server)
     @socket.on('error', @socketError)
     @socket.on('connect', @socketConnect)
+    @socket.on('close', @socketClose)
+    @socket.on('end', @socketClose)
 
     @messageIo = new MessageIO(@socket, @config.options.packetSize, @debug)
     @messageIo.on('packet', (packet) =>
@@ -218,6 +220,10 @@ class Connection extends EventEmitter
     @debug.log("connected to #{@config.server}:#{@config.options.port}")
     @dispatchEvent('socketConnect')
 
+  socketClose: =>
+    @debug.log("connection to #{@config.server}:#{@config.options.port} closed")
+    @transitionTo(@STATE.FINAL)
+
   sendPreLogin: ->
     payload = new PreloginPayload()
     @messageIo.sendMessage(TYPE.PRELOGIN, payload.data)