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 balance map add

Browse Source
This commit is contained in:
c9s 2022-05-04 17:39:35 +08:00
parent 75adb8f3c3
commit 4404098bf9
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 30 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/types/account_test.go
View File

@ -19,6 +19,23 @@ func TestBalanceMap_Add(t *testing.T) {
},
})
assert.Len(t, bm2, 1)
var bm3 = bm2.Add(BalanceMap{
"BTC": Balance{
Currency: "BTC",
Available: fixedpoint.MustNewFromString("1.0"),
Locked: fixedpoint.MustNewFromString("0"),
NetAsset: fixedpoint.MustNewFromString("1.0"),
},
"LTC": Balance{
Currency: "LTC",
Available: fixedpoint.MustNewFromString("20.0"),
Locked: fixedpoint.MustNewFromString("0"),
NetAsset: fixedpoint.MustNewFromString("20.0"),
},
})
assert.Len(t, bm3, 2)
assert.Equal(t, fixedpoint.MustNewFromString("11.0"), bm3["BTC"].Available)
}
func TestAccountLockAndUnlock(t *testing.T) {

10
pkg/types/balance.go
View File

@ -41,7 +41,6 @@ func (b Balance) String() (o string) {
return o
}
type BalanceMap map[string]Balance
func (m BalanceMap) Currencies() (currencies []string) {
@ -52,14 +51,18 @@ func (m BalanceMap) Currencies() (currencies []string) {
}
func (m BalanceMap) Add(bm BalanceMap) BalanceMap {
var total = BalanceMap{}
var total = m.Copy()
for _, b := range bm {
tb := total[b.Currency]
tb, ok := total[b.Currency]
if ok {
tb.Available = tb.Available.Add(b.Available)
tb.Locked = tb.Locked.Add(b.Locked)
tb.Borrowed = tb.Borrowed.Add(b.Borrowed)
tb.NetAsset = tb.NetAsset.Add(b.NetAsset)
tb.Interest = tb.Interest.Add(b.Interest)
} else {
tb = b
}
total[b.Currency] = tb
}
return total
@ -154,7 +157,6 @@ func (m BalanceMap) Print() {
}
}
func findUSDMarketPrice(currency string, prices map[string]fixedpoint.Value) (string, fixedpoint.Value, bool) {
usdMarkets := []string{currency + "USDT", currency + "USDC", currency + "USD", "USDT" + currency}
for _, market := range usdMarkets {