lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-10 09:11:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

core: add exceededMaximumTradeStoreSize check

Browse Source
This commit is contained in:
c9s 2023-12-12 18:24:33 +08:00
parent 685f332495
commit 21c8593c45
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/core/tradestore.go
View File

@ -10,7 +10,7 @@ import (
)
const TradeExpiryTime = 3 * time.Hour
const PruneTriggerNumOfTrades = 10_000
const MaximumTradeStoreSize = 1_000
type TradeStore struct {
// any created trades for tracking trades
@ -142,6 +142,10 @@ func (s *TradeStore) isCoolTrade(trade types.Trade) bool {
return !s.lastTradeTime.IsZero() && time.Time(trade.Time).Sub(s.lastTradeTime) > time.Hour
}
func (s *TradeStore) exceededMaximumTradeStoreSize() bool {
return len(s.trades) > MaximumTradeStoreSize
}
func (s *TradeStore) BindStream(stream types.Stream) {
stream.OnTradeUpdate(func(trade types.Trade) {
s.Add(trade)
@ -149,7 +153,7 @@ func (s *TradeStore) BindStream(stream types.Stream) {
if s.EnablePrune {
stream.OnTradeUpdate(func(trade types.Trade) {
if s.isCoolTrade(trade) {
if s.isCoolTrade(trade) || s.exceededMaximumTradeStoreSize() {
s.Prune(time.Time(trade.Time))
}
})