chore(purchase_watcher): only query latest purchases when we have > 5…

…0 in db
Primexz · Jun 25, 2024 · 762c6ed · 762c6ed
1 parent ceb3106
commit 762c6ed
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 5 deletions.
diff --git a/go.mod b/go.mod
@@ -3,7 +3,7 @@ module github.com/Primexz/Kraken-InvestMetrics
 go 1.22
 
 require (
-	github.com/Primexz/go_kraken v0.1.7
+	github.com/Primexz/go_kraken v0.1.8
 	github.com/btcsuite/btcd v0.24.0
 	github.com/btcsuite/btcd/btcutil v1.1.5
 	github.com/jackc/pgx/v5 v5.5.5

diff --git a/go.sum b/go.sum
@@ -2,6 +2,8 @@ github.com/Primexz/go_kraken v0.1.6 h1:MsiwfrAyDZemWh8AWEryPRrrDmp/4HoTwnBBbgJMV
 github.com/Primexz/go_kraken v0.1.6/go.mod h1:T3DC9V+nbaIf3q2KCvCrf6RA81N1HEptRZg5PW4/6Gg=
 github.com/Primexz/go_kraken v0.1.7 h1:bKe6fJ1hPGChuJOOGzNp7Qqhn1Fe9WKnb2ViQYumPUU=
 github.com/Primexz/go_kraken v0.1.7/go.mod h1:T3DC9V+nbaIf3q2KCvCrf6RA81N1HEptRZg5PW4/6Gg=
+github.com/Primexz/go_kraken v0.1.8 h1:If7YktK4HblVheFHYJspiYcOu2kaT6Jh9w2knlx8A7w=
+github.com/Primexz/go_kraken v0.1.8/go.mod h1:T3DC9V+nbaIf3q2KCvCrf6RA81N1HEptRZg5PW4/6Gg=
 github.com/aead/siphash v1.0.1/go.mod h1:Nywa3cDsYNNK3gaciGTWPwHt0wlpNV15vwmswBAUSII=
 github.com/btcsuite/btcd v0.20.1-beta/go.mod h1:wVuoA8VJLEcwgqHBwHmzLRazpKxTv13Px/pDuV7OomQ=
 github.com/btcsuite/btcd v0.22.0-beta.0.20220111032746-97732e52810c/go.mod h1:tjmYdS6MLJ5/s0Fj4DbLgSbDHbEqLJrtnHecBFkdz5M=

diff --git a/modules/kraken/kraken.go b/modules/kraken/kraken.go
@@ -47,7 +47,7 @@ func (k *KrakenApi) GetPendingEuroOnKraken() (decimal.Decimal, error) {
 }
 
 func (k *KrakenApi) GetCachePayedToKraken() (float64, error) {
-	ledgerInfo, err := k.api.GetLedgersInfo("", 0, 0, "ZEUR")
+	ledgerInfo, err := k.api.GetLedgersInfo("", 0, 0, false, "ZEUR")
 	if err != nil {
 		return 0, err
 	}
@@ -62,8 +62,8 @@ func (k *KrakenApi) GetCachePayedToKraken() (float64, error) {
 	return totalCachePayedToKraken, nil
 }
 
-func (k *KrakenApi) GetAllBtcOrders() ([]rest.Ledger, error) {
-	ledgers, err := k.api.GetLedgersInfo("", 0, 0, "XXBT")
+func (k *KrakenApi) GetAllBtcOrders(latestOnly bool) ([]rest.Ledger, error) {
+	ledgers, err := k.api.GetLedgersInfo("", 0, 0, latestOnly, "XXBT")
 	if err != nil {
 		return nil, err
 	}

diff --git a/updateWatchers/watchers/purchaseWatcher.go b/updateWatchers/watchers/purchaseWatcher.go
@@ -34,7 +34,8 @@ func (pw *PurchaseWatcher) StartRoutine() {
 func (pw *PurchaseWatcher) UpdateData() {
 	pw.log.Info("Updating Purchase Watcher")
 
-	purchases, err := pw.api.GetAllBtcOrders()
+	//if the row count is greater than 50, then we only want to get the last 50 purchases
+	purchases, err := pw.api.GetAllBtcOrders(pw.getPurchasesRowCount() > 50)
 	if err != nil {
 		pw.log.Error(err)
 		return
@@ -49,3 +50,13 @@ func (pw *PurchaseWatcher) UpdateData() {
 		}
 	}
 }
+
+func (pw *PurchaseWatcher) getPurchasesRowCount() int {
+	var count int
+	err := timescale.ConnectionPool.QueryRow(timescale.Context, "SELECT COUNT(*) FROM purchases").Scan(&count)
+	if err != nil {
+		pw.log.Error("failed to get row count", err)
+	}
+
+	return count
+}