[Add]增加SocketIO数据广播推送

This commit is contained in:
vn.py 2018-01-21 17:07:13 +08:00
parent d8542ada05
commit 20bb34c045

View File

@ -7,7 +7,9 @@ sys.setdefaultencoding('utf8')
# 创建主引擎代理对象
from vnpy.event import EventEngine2
from vnpy.trader.vtEvent import EVENT_LOG
from vnpy.trader.vtEvent import (EVENT_TICK, EVENT_ORDER, EVENT_TRADE,
EVENT_ACCOUNT, EVENT_POSITION, EVENT_LOG,
EVENT_ERROR, EVENT_CONTRACT)
from vnpy.trader.vtObject import VtSubscribeReq, VtOrderReq, VtCancelOrderReq
from vnpy.trader.app.rpcService.rsClient import MainEngineProxy
@ -30,9 +32,12 @@ ee.register(EVENT_LOG, printLog)
# 创建Flask对象
from flask import Flask
from flask.ext.restful import Api, Resource, reqparse
from flask.ext.socketio import SocketIO
app = Flask(__name__)
api = Api(app)
socketio = SocketIO(app)
# 创建资源
########################################################################
@ -44,7 +49,7 @@ class Gateway(Resource):
"""初始化"""
self.parser = reqparse.RequestParser()
self.parser.add_argument('gatewayName')
super(Gateways, self).__init__()
super(Gateway, self).__init__()
#----------------------------------------------------------------------
def get(self):
@ -78,7 +83,7 @@ class Order(Resource):
self.deleteParser = reqparse.RequestParser()
self.deletaParser.add_argument('vtOrderID')
super(Gateways, self).__init__()
super(Order, self).__init__()
#----------------------------------------------------------------------
def get(self):
@ -210,7 +215,7 @@ class Tick(Resource):
"""初始化"""
self.parser = reqparse.RequestParser()
self.parser.add_argument('vtSymbol')
super(Gateways, self).__init__()
super(Tick, self).__init__()
#----------------------------------------------------------------------
def post(self):
@ -242,6 +247,26 @@ api.add_resource(Error, '/error')
api.add_resource(Tick, '/tick')
# SocketIO
#----------------------------------------------------------------------
def handleEvent(event):
"""处理事件"""
eventType = event.type_
eventData = event.dict_['data'].__dict__
socketio.emit(eventType, eventData, broadcast=True)
ee.register(EVENT_TICK, handleEvent)
ee.register(EVENT_ORDER, handleEvent)
ee.register(EVENT_TRADE, handleEvent)
ee.register(EVENT_ACCOUNT, handleEvent)
ee.register(EVENT_POSITION, handleEvent)
ee.register(EVENT_CONTRACT, handleEvent)
ee.register(EVENT_LOG, handleEvent)
ee.register(EVENT_ERROR, handleEvent)
if __name__ == '__main__':
app.run(debug=True)