fix(telegram): display realtime balance

rodrigo-brito · Jan 23, 2022 · 4952fe6 · 4952fe6
1 parent 3cec963
commit 4952fe6
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 1 deletion.
diff --git a/exchange/binance.go b/exchange/binance.go
@@ -101,6 +101,14 @@ func NewBinance(ctx context.Context, options ...BinanceOption) (*Binance, error)
 	return exchange, nil
 }
 
+func (b *Binance) LastQuote(ctx context.Context, pair string) (float64, error) {
+	candles, err := b.CandlesByLimit(ctx, pair, "1m", 1)
+	if err != nil || len(candles) < 1 {
+		return 0, err
+	}
+	return candles[0].Close, nil
+}
+
 func (b *Binance) AssetsInfo(pair string) model.AssetInfo {
 	return b.assetsInfo[pair]
 }

diff --git a/exchange/csvfeed.go b/exchange/csvfeed.go
@@ -117,6 +117,10 @@ func (c CSVFeed) feedTimeframeKey(pair, timeframe string) string {
 	return fmt.Sprintf("%s--%s", pair, timeframe)
 }
 
+func (c CSVFeed) LastQuote(_ context.Context, _ string) (float64, error) {
+	return 0, errors.New("invalid operation")
+}
+
 func isFistCandlePeriod(t time.Time, fromTimeframe, targetTimeframe string) (bool, error) {
 	fromDuration, err := str2duration.ParseDuration(fromTimeframe)
 	if err != nil {

diff --git a/exchange/paperwallet.go b/exchange/paperwallet.go
@@ -121,6 +121,10 @@ func (p *PaperWallet) Pairs() []string {
 	return pairs
 }
 
+func (p *PaperWallet) LastQuote(ctx context.Context, pair string) (float64, error) {
+	return p.feeder.LastQuote(ctx, pair)
+}
+
 func (p *PaperWallet) AssetValues(pair string) []AssetValue {
 	return p.assetValues[pair]
 }

diff --git a/notification/telegram.go b/notification/telegram.go
@@ -144,13 +144,14 @@ func (t telegram) BalanceHandle(m *tb.Message) {
 			return
 		}
 
-		assetValue, err := t.orderController.PositionValue(pair)
+		quote, err := t.orderController.LastQuote(pair)
 		if err != nil {
 			log.Error(err)
 			t.OnError(err)
 			return
 		}
 
+		assetValue := assetSize * quote
 		quotesValue[quotePair] = quoteSize
 		total += assetValue
 		message += fmt.Sprintf("%s: `%.4f` ≅ `%.2f` %s \n", assetPair, assetSize, assetValue, quotePair)

diff --git a/order/controller.go b/order/controller.go
@@ -307,6 +307,10 @@ func (c *Controller) Position(pair string) (asset, quote float64, err error) {
 	return c.exchange.Position(pair)
 }
 
+func (c *Controller) LastQuote(pair string) (float64, error) {
+	return c.exchange.LastQuote(c.ctx, pair)
+}
+
 func (c *Controller) PositionValue(pair string) (float64, error) {
 	asset, _, err := c.exchange.Position(pair)
 	if err != nil {

diff --git a/service/service.go b/service/service.go
@@ -14,6 +14,7 @@ type Exchange interface {
 
 type Feeder interface {
 	AssetsInfo(pair string) model.AssetInfo
+	LastQuote(ctx context.Context, pair string) (float64, error)
 	CandlesByPeriod(ctx context.Context, pair, period string, start, end time.Time) ([]model.Candle, error)
 	CandlesByLimit(ctx context.Context, pair, period string, limit int) ([]model.Candle, error)
 	CandlesSubscription(ctx context.Context, pair, timeframe string) (chan model.Candle, chan error)