Fixed race condition with on_open and session.close()

mrjoes · mrjoes · commit 9b9d4a4a5be1 · 2012-02-03T16:27:08.000+02:00
diff --git a/tornadio2/session.py b/tornadio2/session.py
@@ -107,10 +107,6 @@ def __init__(self, conn, server, request, expiry=None):
         # If everything is fine - continue
         self.send_message(proto.connect())
 
-        result = self.conn.on_open(info)
-        if result is not None and not result:
-            raise HTTPError(401)
-
         # Heartbeat related stuff
         self._heartbeat_timer = None
         self._heartbeat_interval = self.server.settings['heartbeat_interval'] * 1000
@@ -119,6 +115,10 @@ def __init__(self, conn, server, request, expiry=None):
         # Endpoints
         self.endpoints = dict()
 
+        result = self.conn.on_open(info)
+        if result is not None and not result:
+            raise HTTPError(401)
+
     # Session callbacks
     def on_delete(self, forced):
         """Session expiration callback
