From f17b710d368ab1868a544e6cfc1e060035c5e23b Mon Sep 17 00:00:00 2001 From: chenxy123 Date: Tue, 1 Nov 2016 22:41:08 +0800 Subject: [PATCH] =?UTF-8?q?1.=20=E4=BF=AE=E6=94=B9eventEngine.py=E7=9A=84?= =?UTF-8?q?=E9=83=A8=E5=88=86=E4=BB=A3=E7=A0=81=EF=BC=8C=E6=9B=B4=E5=8A=A0?= =?UTF-8?q?=E6=B8=85=E6=99=B0=202.=20=E5=9C=A8CTA=E5=BC=95=E6=93=8E?= =?UTF-8?q?=E4=B8=AD=E5=A2=9E=E5=8A=A0=E7=AD=96=E7=95=A5=E5=AF=B9=E8=B1=A1?= =?UTF-8?q?=E5=87=BD=E6=95=B0=E6=89=A7=E8=A1=8C=E6=97=B6=E7=9A=84=E5=BC=82?= =?UTF-8?q?=E5=B8=B8=E6=8D=95=E6=8D=89=E5=8A=9F=E8=83=BD=EF=BC=8C=E9=81=BF?= =?UTF-8?q?=E5=85=8D=E5=9B=A0=E4=B8=BA=E4=B8=AA=E5=88=AB=E7=AD=96=E7=95=A5?= =?UTF-8?q?=E4=BB=A3=E7=A0=81=E9=97=AE=E9=A2=98=E5=AF=BC=E8=87=B4=E7=A8=8B?= =?UTF-8?q?=E5=BA=8F=E5=81=9C=E6=AD=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- vn.event/eventEngine.py | 38 ++++++++++----------- vn.trader/ctaAlgo/ctaEngine.py | 30 ++++++++++++---- vn.trader/eventEngine.py | 62 +++++++++++++--------------------- 3 files changed, 65 insertions(+), 65 deletions(-) diff --git a/vn.event/eventEngine.py b/vn.event/eventEngine.py index 45e1e4da..e800c57c 100644 --- a/vn.event/eventEngine.py +++ b/vn.event/eventEngine.py @@ -17,7 +17,6 @@ from eventType import * class EventEngine(object): """ 事件驱动引擎 - 事件驱动引擎中所有的变量都设置为了私有，这是为了防止不小心 从外部修改了这些变量的值或状态，导致bug。 @@ -129,9 +128,12 @@ class EventEngine(object): #---------------------------------------------------------------------- def register(self, type_, handler): """注册事件处理函数监听""" + # 尝试获取该事件类型对应的处理函数列表，若无defaultDict会自动创建新的list + handlerList = self.__handlers[type_] + # 若要注册的处理器不在该事件的处理器列表中，则注册该事件 - if handler not in self.__handlers[type_]: - self.__handlers[type_].append(handler) + if handler not in handlerList: + handlerList.append(handler) #---------------------------------------------------------------------- def unregister(self, type_, handler): @@ -146,6 +148,7 @@ class EventEngine(object): # 如果函数列表为空，则从引擎中移除该事件类型 if not handlerList: del self.__handlers[type_] + #---------------------------------------------------------------------- def put(self, event): """向事件队列中存入事件""" @@ -177,7 +180,7 @@ class EventEngine2(object): # 这里的__handlers是一个字典，用来保存对应的事件调用关系 # 其中每个键对应的值是一个列表，列表中保存了对该事件进行监听的函数功能 - self.__handlers = {} + self.__handlers = defaultdict(list) #---------------------------------------------------------------------- def __run(self): @@ -243,12 +246,8 @@ class EventEngine2(object): #---------------------------------------------------------------------- def register(self, type_, handler): """注册事件处理函数监听""" - # 尝试获取该事件类型对应的处理函数列表，若无则创建 - try: - handlerList = self.__handlers[type_] - except KeyError: - handlerList = [] - self.__handlers[type_] = handlerList + # 尝试获取该事件类型对应的处理函数列表，若无defaultDict会自动创建新的list + handlerList = self.__handlers[type_] # 若要注册的处理器不在该事件的处理器列表中，则注册该事件 if handler not in handlerList: @@ -257,19 +256,16 @@ class EventEngine2(object): #---------------------------------------------------------------------- def unregister(self, type_, handler): """注销事件处理函数监听""" - # 尝试获取该事件类型对应的处理函数列表，若无则忽略该次注销请求 - try: - handlerList = self.__handlers[type_] + # 尝试获取该事件类型对应的处理函数列表，若无则忽略该次注销请求 + handlerList = self.__handlers[type_] - # 如果该函数存在于列表中，则移除 - if handler in handlerList: - handlerList.remove(handler) + # 如果该函数存在于列表中，则移除 + if handler in handlerList: + handlerList.remove(handler) - # 如果函数列表为空，则从引擎中移除该事件类型 - if not handlerList: - del self.__handlers[type_] - except KeyError: - pass + # 如果函数列表为空，则从引擎中移除该事件类型 + if not handlerList: + del self.__handlers[type_] #---------------------------------------------------------------------- def put(self, event): diff --git a/vn.trader/ctaAlgo/ctaEngine.py b/vn.trader/ctaAlgo/ctaEngine.py index 16015a7e..0b095b20 100644 --- a/vn.trader/ctaAlgo/ctaEngine.py +++ b/vn.trader/ctaAlgo/ctaEngine.py @@ -18,6 +18,7 @@ import json import os +import traceback from collections import OrderedDict from datetime import datetime, timedelta @@ -256,7 +257,7 @@ class CtaEngine(object): # 逐个推送到策略实例中 l = self.tickStrategyDict[tick.vtSymbol] for strategy in l: - strategy.onTick(ctaTick) + self.callStrategyFunc(strategy, strategy.onTick, ctaTick) #---------------------------------------------------------------------- def processOrderEvent(self, event): @@ -265,7 +266,7 @@ class CtaEngine(object): if order.vtOrderID in self.orderStrategyDict: strategy = self.orderStrategyDict[order.vtOrderID] - strategy.onOrder(order) + self.callStrategyFunc(strategy, strategy.onOrder, order) #---------------------------------------------------------------------- def processTradeEvent(self, event): @@ -281,7 +282,7 @@ class CtaEngine(object): else: strategy.pos -= trade.volume - strategy.onTrade(trade) + self.callStrategyFunc(strategy, strategy.onTrade, trade) # 更新持仓缓存数据 if trade.vtSymbol in self.tickStrategyDict: @@ -417,7 +418,7 @@ class CtaEngine(object): if not strategy.inited: strategy.inited = True - strategy.onInit() + self.callStrategyFunc(strategy, strategy.onInit) else: self.writeCtaLog(u'请勿重复初始化策略实例：%s' %name) else: @@ -431,7 +432,7 @@ class CtaEngine(object): if strategy.inited and not strategy.trading: strategy.trading = True - strategy.onStart() + self.callStrategyFunc(strategy, strategy.onStart) else: self.writeCtaLog(u'策略实例不存在：%s' %name) @@ -443,7 +444,7 @@ class CtaEngine(object): if strategy.trading: strategy.trading = False - strategy.onStop() + self.callStrategyFunc(strategy, strategy.onStop) # 对该策略发出的所有限价单进行撤单 for vtOrderID, s in self.orderStrategyDict.items(): @@ -517,6 +518,23 @@ class CtaEngine(object): event = Event(EVENT_CTA_STRATEGY+name) self.eventEngine.put(event) + #---------------------------------------------------------------------- + def callStrategyFunc(self, strategy, func, params=None): + """调用策略的函数，若触发异常则捕捉""" + try: + if params: + func(params) + else: + func() + except Exception: + # 停止策略，修改状态为未初始化 + strategy.trading = False + strategy.inited = False + + # 发出日志 + content = '\n'.join([u'策略%s触发异常已停止' %strategy.name, + traceback.format_exc()]) + self.writeCtaLog(content) ######################################################################## diff --git a/vn.trader/eventEngine.py b/vn.trader/eventEngine.py index 90c30f02..e800c57c 100644 --- a/vn.trader/eventEngine.py +++ b/vn.trader/eventEngine.py @@ -4,6 +4,7 @@ from Queue import Queue, Empty from threading import Thread from time import sleep +from collections import defaultdict # 第三方模块 from PyQt4.QtCore import QTimer @@ -16,7 +17,6 @@ from eventType import * class EventEngine(object): """ 事件驱动引擎 - 事件驱动引擎中所有的变量都设置为了私有，这是为了防止不小心 从外部修改了这些变量的值或状态，导致bug。 @@ -68,7 +68,7 @@ class EventEngine(object): # 这里的__handlers是一个字典，用来保存对应的事件调用关系 # 其中每个键对应的值是一个列表，列表中保存了对该事件进行监听的函数功能 - self.__handlers = {} + self.__handlers = defaultdict(list) #---------------------------------------------------------------------- def __run(self): @@ -128,12 +128,8 @@ class EventEngine(object): #---------------------------------------------------------------------- def register(self, type_, handler): """注册事件处理函数监听""" - # 尝试获取该事件类型对应的处理函数列表，若无则创建 - try: - handlerList = self.__handlers[type_] - except KeyError: - handlerList = [] - self.__handlers[type_] = handlerList + # 尝试获取该事件类型对应的处理函数列表，若无defaultDict会自动创建新的list + handlerList = self.__handlers[type_] # 若要注册的处理器不在该事件的处理器列表中，则注册该事件 if handler not in handlerList: @@ -142,20 +138,17 @@ class EventEngine(object): #---------------------------------------------------------------------- def unregister(self, type_, handler): """注销事件处理函数监听""" - # 尝试获取该事件类型对应的处理函数列表，若无则忽略该次注销请求 - try: - handlerList = self.__handlers[type_] + # 尝试获取该事件类型对应的处理函数列表，若无则忽略该次注销请求 + handlerList = self.__handlers[type_] - # 如果该函数存在于列表中，则移除 - if handler in handlerList: - handlerList.remove(handler) + # 如果该函数存在于列表中，则移除 + if handler in handlerList: + handlerList.remove(handler) - # 如果函数列表为空，则从引擎中移除该事件类型 - if not handlerList: - del self.__handlers[type_] - except KeyError: - pass - + # 如果函数列表为空，则从引擎中移除该事件类型 + if not handlerList: + del self.__handlers[type_] + #---------------------------------------------------------------------- def put(self, event): """向事件队列中存入事件""" @@ -187,7 +180,7 @@ class EventEngine2(object): # 这里的__handlers是一个字典，用来保存对应的事件调用关系 # 其中每个键对应的值是一个列表，列表中保存了对该事件进行监听的函数功能 - self.__handlers = {} + self.__handlers = defaultdict(list) #---------------------------------------------------------------------- def __run(self): @@ -253,12 +246,8 @@ class EventEngine2(object): #---------------------------------------------------------------------- def register(self, type_, handler): """注册事件处理函数监听""" - # 尝试获取该事件类型对应的处理函数列表，若无则创建 - try: - handlerList = self.__handlers[type_] - except KeyError: - handlerList = [] - self.__handlers[type_] = handlerList + # 尝试获取该事件类型对应的处理函数列表，若无defaultDict会自动创建新的list + handlerList = self.__handlers[type_] # 若要注册的处理器不在该事件的处理器列表中，则注册该事件 if handler not in handlerList: @@ -267,19 +256,16 @@ class EventEngine2(object): #---------------------------------------------------------------------- def unregister(self, type_, handler): """注销事件处理函数监听""" - # 尝试获取该事件类型对应的处理函数列表，若无则忽略该次注销请求 - try: - handlerList = self.__handlers[type_] + # 尝试获取该事件类型对应的处理函数列表，若无则忽略该次注销请求 + handlerList = self.__handlers[type_] - # 如果该函数存在于列表中，则移除 - if handler in handlerList: - handlerList.remove(handler) + # 如果该函数存在于列表中，则移除 + if handler in handlerList: + handlerList.remove(handler) - # 如果函数列表为空，则从引擎中移除该事件类型 - if not handlerList: - del self.__handlers[type_] - except KeyError: - pass + # 如果函数列表为空，则从引擎中移除该事件类型 + if not handlerList: + del self.__handlers[type_] #---------------------------------------------------------------------- def put(self, event):