本文介绍了VNPY数据记录器的源码，包括如何注册并处理EVENT_TICK和EVENT_CONTRACT事件，以及如何保存Tick和Bar数据。详细讲解了RecorderEngine类的功能，如添加、移除记录，以及数据库管理接口的使用。同时讨论了数据存储的相关方法save_tick_data和save_bar_data。
VNPY源码学习系列文章：
一、源码
"""
注册EVENT_TICK、EVENT_CONTRACT，当有EVENT_TICK的时候，调用process_contract_event函数(其实就是record_tick函数)，
将task put到queue
通过run函数，从self.queue获得task(Tick、Bar),调用database_manager的方法储存数据
"""
from threading import Thread
from queue import Queue, Empty
from copy import copy
from vnpy.event import Event, EventEngine
from vnpy.trader.engine import BaseEngine, MainEngine
from vnpy.trader.object import (
SubscribeRequest,
TickData,
BarData,
ContractData
)
from vnpy.trader.event import EVENT_TICK, EVENT_CONTRACT
from vnpy.trader.utility import load_json, save_json, BarGenerator
from vnpy.trader.database import database_manager
APP_NAME = "DataRecorder"
EVENT_RECORDER_LOG = "eRecorderLog"
EVENT_RECORDER_UPDATE = "eRecorderUpdate"
class RecorderEngine(BaseEngine):
""""""
setting_filename = "data_recorder_setting.json"
def __init__(self, main_engine: MainEngine, event_engine: EventEngine):
""""""
super().__init__(main_engine, event_engine, APP_NAME)
self.queue = Queue()
self.thread = Thread(target=self.run)
self.active = False
self.tick_recordings = {}
self.bar_recordings = {}
self.bar_generators = {}
self.load_setting()
self.register_event()
self.start()
self.put_event()
def load_setting(self):
""""""
setting = load_json(self.setting_filename
最低0.47元/天 解锁文章
1304
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
