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

improve error messages

Browse Source
This commit is contained in:
c9s 2020-08-04 20:04:15 +08:00
parent ef2fd84dce
commit f276520186
1 changed files with 25 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
bbgo/stock.go
View File

@ -5,6 +5,7 @@ import (
"github.com/c9s/bbgo/pkg/bbgo/types"
"math"
"strings"
"sync"
)
func zero(a float64) bool {
@ -48,18 +49,25 @@ func (slice StockSlice) Quantity() (total float64) {
}
type StockManager struct {
mu sync.Mutex
Symbol string
TradingFeeCurrency string
Stocks StockSlice
PendingSells StockSlice
}
func (m *StockManager) Stock(buy Stock) error {
m.Stocks = append(m.Stocks, buy)
func (m *StockManager) stock(stock Stock) error {
m.mu.Lock()
m.Stocks = append(m.Stocks, stock)
m.mu.Unlock()
return m.flushPendingSells()
}
func (m *StockManager) squash() {
m.mu.Lock()
defer m.mu.Unlock()
var squashed StockSlice
for _, stock := range m.Stocks {
if !zero(stock.Quantity) {
@ -70,22 +78,27 @@ func (m *StockManager) squash() {
}
func (m *StockManager) flushPendingSells() error {
if len(m.Stocks) > 0 && len(m.PendingSells) > 0 {
if len(m.Stocks) == 0 || len(m.PendingSells) == 0 {
return nil
}
pendingSells := m.PendingSells
m.PendingSells = nil
pendingSells := m.PendingSells
m.PendingSells = nil
for _, sell := range pendingSells {
if err := m.Consume(sell); err != nil {
return err
}
for _, sell := range pendingSells {
if err := m.consume(sell); err != nil {
return err
}
}
return nil
}
func (m *StockManager) Consume(sell Stock) error {
func (m *StockManager) consume(sell Stock) error {
m.mu.Lock()
defer m.mu.Unlock()
if len(m.Stocks) == 0 {
m.PendingSells = append(m.PendingSells, sell)
return nil
@ -158,12 +171,12 @@ func (m *StockManager) AddTrades(trades []types.Trade) (checkpoints []int, err e
}
stock := toStock(trade)
if err := m.Stock(stock); err != nil {
if err := m.stock(stock); err != nil {
return checkpoints, err
}
} else {
stock := toStock(trade)
if err := m.Consume(stock); err != nil {
if err := m.consume(stock); err != nil {
return checkpoints, err
}
}