Add check to end simulation and add drain function properly (DeviaVir…

…#1977)
olexiyb · Feb 2, 2020 · 0d954a7 · 0d954a7
1 parent 9b3caf0
commit 0d954a7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/commands/sim.js b/commands/sim.js
@@ -295,6 +295,10 @@ module.exports = function (program, conf) {
           setImmediate(async () => await getNext())
         }
 
+        if(totalTrades === 0) {
+          onCollectionCursorEnd()
+        }
+
         collectionCursorStream.on('data', function(trade) {
           lastTrade = trade
           numTrades++

diff --git a/lib/engine.js b/lib/engine.js
@@ -990,12 +990,12 @@ module.exports = function (s, conf) {
     update: onTrades,
     exit: function (cb) {
       if(tradeProcessingQueue.length()){
-        tradeProcessingQueue.drain = () => {
+        tradeProcessingQueue.drain(() => {
           if(s.strategy.onExit) {
             s.strategy.onExit.call( s.ctx, s )
           }
           cb()
-        }
+        })
       } else {
         if(s.strategy.onExit) {
           s.strategy.onExit.call( s.ctx, s )