mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-29 04:25:37 +08:00
Jenkins T. Quantopian, III
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
import copy
|
|
import multiprocessing
|
|
import zmq
|
|
import logging
|
|
import json
|
|
|
|
from backtest.util import *
|
|
|
|
class TestClient(object):
|
|
|
|
def __init__(self,feed, address, bind=False):
|
|
self.logger = logging.getLogger()
|
|
self.feed = feed
|
|
self.address = address
|
|
self.sync = FeedSync(feed, "testclient")
|
|
self.bind = bind
|
|
self.received_count = 0
|
|
|
|
def run(self):
|
|
|
|
self.logger.info("running the client")
|
|
self.context = zmq.Context()
|
|
|
|
self.data_feed = self.context.socket(zmq.PULL)
|
|
|
|
if(self.bind):
|
|
self.logger.info("binding to {address}".format(address=self.address))
|
|
self.data_feed.bind(self.address)
|
|
else:
|
|
self.logger.info("connecting to {address}".format(address=self.address))
|
|
self.data_feed.connect(self.address)
|
|
|
|
self.sync.confirm()
|
|
|
|
self.logger.info("Starting the client loop")
|
|
|
|
prev_dt = None
|
|
while True:
|
|
msg = self.data_feed.recv()
|
|
if(msg == "DONE"):
|
|
self.logger.info("DONE!")
|
|
break
|
|
self.received_count += 1
|
|
event = json.loads(msg)
|
|
if(prev_dt != None):
|
|
if(not event['dt'] >= prev_dt):
|
|
raise Exception("message arrived out of order: {date} after {prev}".format(date=event['dt'], prev=prev_dt))
|
|
|
|
prev_dt = event['dt']
|
|
if(self.received_count % 100 == 0):
|
|
self.logger.info("received {n} messages".format(n=self.received_count))
|
|
|
|
self.logger.info("received {n} messages".format(n=self.received_count))
|
|
self.data_feed.close()
|
|
self.context.term() |