_

services/wallet/concurrent.go

@@ -45,8 +45,10 @@ type TransferDownloader interface {
 	GetTransfersByNumber(context.Context, *big.Int) ([]Transfer, error)
 }
 
-var balancesCache = map[common.Address]map[*big.Int]*big.Int{}
-var balanaceMapLock = sync.RWMutex{}
+var (
+	balancesCache   = map[common.Address]map[*big.Int]*big.Int{}
+	balanaceMapLock = sync.RWMutex{}
+)
 
 func readCachedBalance(account common.Address, blockNumber *big.Int) (*big.Int, bool) {
 	balanaceMapLock.RLock()
@@ -56,13 +58,13 @@ func readCachedBalance(account common.Address, blockNumber *big.Int) (*big.Int,
 	return balance, exists
 }
 
-func writeCachedBalance(account common.Address, blockNumber *big.Int, balance *big.Int) {
+func addBalanceToCache(account common.Address, blockNumber *big.Int, balance *big.Int) {
 	balanaceMapLock.Lock()
 	defer balanaceMapLock.Unlock()
 
 	_, exists := balancesCache[account]
 	if !exists {
-		balancesCache[account] = map[*big.Int]*big.Int{}
+		balancesCache[account] = make(map[*big.Int]*big.Int)
 	}
 	balancesCache[account][blockNumber] = balance
 }
@@ -76,7 +78,7 @@ func getBalanceAt(client BalanceReader, ctx context.Context, account common.Addr
 		if err != nil {
 			return nil, err
 		}
-		writeCachedBalance(account, blockNumber, balance)
+		addBalanceToCache(account, blockNumber, balance)
 
 		return balance, nil
 	}