status-im/status-go
2
0
Fork 0
mirror of https://github.com/status-im/status-go.git synced 2025-01-21 20:20:29 +00:00
Code Issues Projects Releases Wiki Activity

feat: allow to get wallet token for specific addresses

Browse Source
This commit is contained in:
Anthony Laibe 2022-12-02 11:55:44 +01:00 committed by Anthony Laibe
parent f25ac111db
commit d7bf19fdbb
2 changed files with 7 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
services/wallet/api.go
View File

@ -34,8 +34,8 @@ func (api *API) StartWallet(ctx context.Context) error {
return api.reader.Start()
}
func (api *API) GetWalletToken(ctx context.Context) (map[common.Address][]Token, error) {
return api.reader.GetWalletToken(ctx)
func (api *API) GetWalletToken(ctx context.Context, addresses []common.Address) (map[common.Address][]Token, error) {
return api.reader.GetWalletToken(ctx, addresses)
}
type DerivedAddress struct {

24
services/wallet/reader.go
View File

@ -57,14 +57,6 @@ type Token struct {
CurrencyPrice float64 `json:"currencyPrice"`
}
func getAddresses(accounts []*accounts.Account) []common.Address {
addresses := make([]common.Address, len(accounts))
for _, account := range accounts {
addresses = append(addresses, common.Address(account.Address))
}
return addresses
}
func getTokenBySymbols(tokens []*token.Token) map[string][]*token.Token {
res := make(map[string][]*token.Token)
@ -127,7 +119,7 @@ func (r *Reader) Stop() {
}
}
func (r *Reader) GetWalletToken(ctx context.Context) (map[common.Address][]Token, error) {
func (r *Reader) GetWalletToken(ctx context.Context, addresses []common.Address) (map[common.Address][]Token, error) {
networks, err := r.rpcClient.NetworkManager.Get(false)
if err != nil {
return nil, err
@ -154,11 +146,6 @@ func (r *Reader) GetWalletToken(ctx context.Context) (map[common.Address][]Token
tokenSymbols := getTokenSymbols(allTokens)
tokenAddresses := getTokenAddresses(allTokens)
accounts, err := r.accountsDB.GetAccounts()
if err != nil {
return nil, err
}
var (
group = async.NewAtomicGroup(ctx)
prices = map[string]float64{}
@ -197,7 +184,7 @@ func (r *Reader) GetWalletToken(ctx context.Context) (map[common.Address][]Token
return err
}
balances, err = r.tokenManager.GetBalancesByChain(ctx, clients, getAddresses(accounts), tokenAddresses)
balances, err = r.tokenManager.GetBalancesByChain(ctx, clients, addresses, tokenAddresses)
if err != nil {
return err
}
@ -214,13 +201,12 @@ func (r *Reader) GetWalletToken(ctx context.Context) (map[common.Address][]Token
return nil, err
}
result := make(map[common.Address][]Token)
for _, account := range accounts {
commonAddress := common.Address(account.Address)
for _, address := range addresses {
for symbol, tokens := range getTokenBySymbols(allTokens) {
balancesPerChain := make(map[uint64]ChainBalance)
decimals := tokens[0].Decimals
for _, token := range tokens {
hexBalance := balances[token.ChainID][commonAddress][token.Address]
hexBalance := balances[token.ChainID][address][token.Address]
balance := big.NewFloat(0.0)
if hexBalance != nil {
balance = new(big.Float).Quo(
@ -254,7 +240,7 @@ func (r *Reader) GetWalletToken(ctx context.Context) (map[common.Address][]Token
CurrencyPrice: prices[symbol],
}
result[commonAddress] = append(result[commonAddress], walletToken)
result[address] = append(result[address], walletToken)
}
}
return result, nil