From 80bfbd5dcbb1263f96caeabf09b5fd638d15b461 Mon Sep 17 00:00:00 2001
From: Stephen Diehl <stephen.m.diehl@gmail.com>
Date: Fri, 6 Apr 2012 13:31:39 -0400
Subject: [PATCH] Performance reports take either socket address or socket.

---
 zipline/finance/performance.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/zipline/finance/performance.py b/zipline/finance/performance.py
index e3564c54..3b784413 100644
--- a/zipline/finance/performance.py
+++ b/zipline/finance/performance.py
@@ -185,11 +185,14 @@ class PerformanceTracker():
         Publish the performance results asynchronously to a
         socket.
         """
-        ctx = context or zmq.Context.instance()
-        sock = ctx.socket(zmq.PUSH)
-        sock.connect(zmq_socket)
+        if isinstance(zmq_socket, zmq.Socket):
+            self.result_stream = zmq_socket
+        else:
+            ctx = context or zmq.Context.instance()
+            sock = ctx.socket(zmq.PUSH)
+            sock.connect(zmq_socket)
 
-        self.result_stream = sock
+            self.result_stream = sock
 
     def to_dict(self):
         """