storage: safely change connection peer id temporarily

95fecbfb · Julien Muchembled · 192b2ba8 · 95fecbfb
Commit 95fecbfb authored Aug 14, 2012 by Julien Muchembled
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

neo/storage/app.py neo/storage/app.py +5 -3

No files found.
--- a/neo/storage/app.py
+++ b/neo/storage/app.py
@@ -363,8 +363,10 @@ class Application(object):
            if conn.isClosed():
                continue
            orig_msg_id = conn.getPeerId()
+            try:
                conn.setPeerId(msg_id)
                some_callable(conn, *args)
+            finally:
                conn.setPeerId(orig_msg_id)

    def logQueuedEvents(self):