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

move balance printing to debug-balance env var

Browse Source
This commit is contained in:
c9s 2021-05-17 20:03:26 +08:00
parent f80c98b97c
commit b8fe100b5e
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pkg/types/account.go
View File

@ -6,10 +6,17 @@ import (
"sync" "sync"
"github.com/sirupsen/logrus" "github.com/sirupsen/logrus"
"github.com/spf13/viper"
"github.com/c9s/bbgo/pkg/fixedpoint" "github.com/c9s/bbgo/pkg/fixedpoint"
) )
var debugBalance = false
func init() {
debugBalance = viper.GetBool("debug-balance")
}
type Balance struct { type Balance struct {
Currency string `json:"currency"` Currency string `json:"currency"`
Available fixedpoint.Value `json:"available"` Available fixedpoint.Value `json:"available"`
@ -216,13 +223,16 @@ func (a *Account) UpdateBalances(balances BalanceMap) {
} }
} }
func printBalanceUpdate(balances BalanceMap) {
logrus.Infof("balance update: %+v", balances)
}
func (a *Account) BindStream(stream Stream) { func (a *Account) BindStream(stream Stream) {
stream.OnBalanceUpdate(a.UpdateBalances) stream.OnBalanceUpdate(a.UpdateBalances)
stream.OnBalanceSnapshot(a.UpdateBalances) stream.OnBalanceSnapshot(a.UpdateBalances)
stream.OnBalanceUpdate(func(balances BalanceMap) { if debugBalance {
logrus.Infof("balance update: %+v", balances) stream.OnBalanceUpdate(printBalanceUpdate)
}) }
} }
func (a *Account) Print() { func (a *Account) Print() {