Merge pull request #1859 from c9s/c9s/account-margin-level-metrics

FEATURE: add account margin level metrics

pkg/bbgo/session.go

@@ -17,6 +17,7 @@ import (
 	"github.com/c9s/bbgo/pkg/cache"
 	"github.com/c9s/bbgo/pkg/envvar"
 	"github.com/c9s/bbgo/pkg/exchange/retry"
+	"github.com/c9s/bbgo/pkg/metrics"
 	"github.com/c9s/bbgo/pkg/util/templateutil"
 
 	exchange2 "github.com/c9s/bbgo/pkg/exchange"
@@ -232,13 +233,32 @@ func (session *ExchangeSession) GetAccount() (a *types.Account) {
 	return a
 }
 
+func (session *ExchangeSession) GetIsolatedSymbol() string {
+	isolatedSymbol := ""
+	if session.IsolatedMarginSymbol != "" {
+		isolatedSymbol = session.IsolatedMarginSymbol
+	} else if session.IsolatedFuturesSymbol != "" {
+		isolatedSymbol = session.IsolatedFuturesSymbol
+	}
+
+	return isolatedSymbol
+}
+
 // UpdateAccount locks the account mutex and update the account object
 func (session *ExchangeSession) UpdateAccount(ctx context.Context) (*types.Account, error) {
 	account, err := session.Exchange.QueryAccount(ctx)
 	if err != nil {
 		return nil, err
 	}
 
+	isolatedSymbol := session.GetIsolatedSymbol()
+
+	metrics.AccountMarginLevelMetrics.With(prometheus.Labels{
+		"exchange":        session.ExchangeName.String(),
+		"account_type":    string(account.AccountType),
+		"isolated_symbol": isolatedSymbol,
+	}).Set(account.MarginLevel.Float64())
+
 	session.setAccount(account)
 	return account, nil
 }

pkg/exchange/max/stream.go

@@ -242,6 +242,8 @@ func (s *Stream) handleTradeEvent(e max.TradeUpdateEvent) {
 	}
 }
 
+// handleBookEvent returns a callback that will be registered to the websocket stream
+// this callback will be called when the websocket stream receives a book event
 func (s *Stream) handleBookEvent(ex *Exchange) func(e max.BookEvent) {
 	return func(e max.BookEvent) {
 		symbol := toGlobalSymbol(e.Market)

pkg/metrics/account.go

@@ -0,0 +1,13 @@
+package metrics
+
+import "github.com/prometheus/client_golang/prometheus"
+
+var AccountMarginLevelMetrics = prometheus.NewGaugeVec(
+	prometheus.GaugeOpts{
+		Name: "bbgo_account_margin_level",
+		Help: "account margin level metrics",
+	}, []string{"exchange", "account_type", "isolated_symbol"})
+
+func init() {
+	prometheus.MustRegister(AccountMarginLevelMetrics)
+}