Fixed bug with WebSocket/FlashSocket transport initialization, in case of

mrjoes · mrjoes · commit 80070fa6ff67 · 2011-06-08T21:15:35.000+03:00
malformed/unsupported WebSocket handshake.
diff --git a/tornadio/persistent.py b/tornadio/persistent.py
@@ -22,6 +22,7 @@ def __init__(self, router, session_id):
         logging.debug('Initializing WebSocket handler...')
 
         self.router = router
+        self.connection = None
 
         super(TornadioWebSocketHandler, self).__init__(router.application,
                                                        router.request)
@@ -80,11 +81,12 @@ def on_message(self, message):
         self.async_callback(self.connection.raw_message)(message)
 
     def on_close(self):
-        try:
-            self.connection.on_close()
-        finally:
-            self.connection.is_closed = True
-            self.connection.stop_heartbeat()
+        if self.connection is not None:
+            try:
+                self.connection.on_close()
+            finally:
+                self.connection.is_closed = True
+                self.connection.stop_heartbeat()
 
     def send(self, message):
         self.write_message(proto.encode(message))
