make balances faster

deso-protocol · Sep 1, 2021 · 5520ae4 · 5520ae4
1 parent 300af30
commit 5520ae4
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/block_view.go b/lib/block_view.go
@@ -8693,10 +8693,15 @@ func (bav *UtxoView) Preload(bitcloutBlock *MsgBitCloutBlock) error {
 		}
 
 		// Set real entries for all the profiles that actually exist
-		result := bav.Postgres.GetProfilesForPublicKeys(publicKeys)
-		for _, profile := range result {
+		profiles := bav.Postgres.GetProfilesForPublicKeys(publicKeys)
+		for _, profile := range profiles {
 			bav.setProfileMappings(profile)
 		}
+
+		balances := bav.Postgres.GetBalancesBatch(publicKeys)
+		for _, balance := range balances {
+			bav.PublicKeyToBitcloutBalanceNanos[MakePkMapKey(balance.PublicKey.ToBytes())] = balance.BalanceNanos
+		}
 	}
 
 	// One iteration for everything else