fix: refetch fiat balance after purchase

Primexz · Jul 31, 2024 · f21245c · f21245c
1 parent 0f05e83
commit f21245c
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/bot/bot.go b/bot/bot.go
@@ -43,10 +43,7 @@ func (b *Bot) StartBot() {
 }
 
 func (b *Bot) run() {
-	if fiatAmnt, err := b.krakenApi.GetFiatBalance(); err == nil {
-		b.fiatAmount = fiatAmnt
-	} else {
-		b.log.Error("Error getting fiat balance: ", err)
+	if err := b.updateFiatBalance(); err != nil {
 		return
 	}
 
@@ -87,6 +84,11 @@ func (b *Bot) run() {
 		b.log.Info("Placing bitcoin purchase order. ₿")
 
 		b.krakenApi.BuyBtc()
+
+		if err := b.updateFiatBalance(); err != nil {
+			return
+		}
+
 		b.calculateTimeOfNextOrder()
 	}
 
@@ -98,6 +100,16 @@ func (b *Bot) run() {
 	}).Info("Next order in")
 }
 
+func (b *Bot) updateFiatBalance() error {
+	if fiatAmnt, err := b.krakenApi.GetFiatBalance(); err == nil {
+		b.fiatAmount = fiatAmnt
+		return nil
+	} else {
+		b.log.Error("Error getting fiat balance: ", err)
+		return err
+	}
+}
+
 func (b *Bot) updateMetrics() {
 	metrics.Metrics.NextOrder = b.timeOfNextOrder.Unix()
 }