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

fix distribution chart

Browse Source
This commit is contained in:
c9s 2020-08-11 10:24:56 +08:00
parent 54a5e41c4f
commit 261da73a9a
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
bbgo/stock.go
View File

@ -4,6 +4,7 @@ import (
"fmt"
"github.com/c9s/bbgo/pkg/bbgo/types"
"math"
"sort"
"strings"
"sync"
)
@ -58,6 +59,7 @@ type StockManager struct {
}
type Distribution struct {
PriceLevels []int `json:"priceLevels"`
Quantities map[int]float64 `json:"quantities"`
Stocks map[int]StockSlice `json:"stocks"`
}
@ -78,6 +80,12 @@ func (m *StockManager) Distribution(level int) *Distribution {
d.Quantities[priceLevel] += stock.Quantity
}
for level := range d.Stocks {
d.PriceLevels = append(d.PriceLevels, level)
}
sort.Ints(d.PriceLevels)
return &d
}