[Add]DrEngine新增收盘时间功能，解决部分交易所没有收盘Tick推送导致最后一根K线缺失的问题 #957 #888

examples/DataRecording/DR_setting.json

@@ -1,5 +1,6 @@
 {
     "working": true,
+	"marketCloseTime": "15:05:00",
 
     "tick":
     [

vnpy/trader/app/ctaStrategy/ctaTemplate.py

@@ -449,6 +449,13 @@ class BarGenerator(object):
             # 清空老K线缓存对象
             self.xminBar = None
 
+    #----------------------------------------------------------------------
+    def generate(self):
+        """手动强制立即完成K线合成"""
+        self.onBar(self.bar)
+        self.bar = None
+
+
 
 ########################################################################
 class ArrayManager(object):

vnpy/trader/app/dataRecorder/drEngine.py

@@ -12,7 +12,7 @@ import os
 import copy
 import traceback
 from collections import OrderedDict
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, time
 from Queue import Queue, Empty
 from threading import Thread
 from pymongo.errors import DuplicateKeyError
@@ -60,6 +60,11 @@ class DrEngine(object):
         self.queue = Queue()                    # 队列
         self.thread = Thread(target=self.run)   # 线程
         
+        # 收盘相关
+        self.marketCloseTime = None             # 收盘时间
+        self.timerCount = 0                     # 定时器计数
+        self.lastTimerTime = None               # 上一次记录时间
+        
         # 载入设置，订阅行情
         self.loadSetting()
         
@@ -80,6 +85,11 @@ class DrEngine(object):
             if not working:
                 return
             
+            # 加载收盘时间
+            if 'marketCloseTime' in drSetting:
+                timestamp = drSetting['marketCloseTime']
+                self.marketCloseTime = datetime.strptime(timestamp, '%H:%M:%S').time()
+
             # Tick记录配置
             if 'tick' in drSetting:
                 l = drSetting['tick']
@@ -186,6 +196,36 @@ class DrEngine(object):
         if bm:
             bm.updateTick(tick)
     
+    #----------------------------------------------------------------------
+    def processTimerEvent(self, event):
+        """处理定时事件"""
+        # 如果没有设置收盘时间，则无需处理
+        if not self.marketCloseTime:
+            return
+        
+        # 10秒检查一次
+        self.timerCount += 1
+        if self.timerCount < 10:
+            return
+        self.timerCount = 0
+        
+        # 获取当前时间
+        currentTime = datetime.now().time()
+        
+        if not self.lastTimerTime:
+            self.lastTimerTime = currentTime
+            return
+        
+        # 上一个时间戳尚未到收盘时间，且当前时间戳已经到收盘时间
+        if (self.lastTimerTime < self.marketCloseTime and
+            currentTime >= self.marketCloseTime):
+            # 强制所有的K线生成器立即完成K线
+            for bg in self.bgDict.values():
+                bg.generate()
+        
+        # 记录新的时间
+        self.lastTimerTime = currentTime
+    
     #----------------------------------------------------------------------
     def onTick(self, tick):
         """Tick更新"""
@@ -227,6 +267,7 @@ class DrEngine(object):
     def registerEvent(self):
         """注册事件监听"""
         self.eventEngine.register(EVENT_TICK, self.procecssTickEvent)
+        self.eventEngine.register(EVENT_TIMER, self.processTimerEvent)
  
     #----------------------------------------------------------------------
     def insertData(self, dbName, collectionName, data):