test(wallet)_: moved all balance related code from token.Manager to
balance_fetcher.BalanceFetcher Split methods into smaller ones for testability
This commit is contained in:
Ivan Belyakov
IvanBelyakoff
parent
135ab9a5cf
commit
d180e19fa8
|
|
@ -15,6 +15,11 @@ import (
|
|||
"github.com/status-im/status-go/rpc"
|
||||
)
|
||||
|
||||
type ContractMakerIface interface {
|
||||
NewEthScan(chainID uint64) (*ethscan.BalanceScanner, uint, error)
|
||||
// TODO extend with other contracts
|
||||
}
|
||||
|
||||
type ContractMaker struct {
|
||||
RPCClient *rpc.Client
|
||||
}
|
||||
|
|
|
|||
|
|
@ -0,0 +1,321 @@
|
|||
package balancefetcher
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"math/big"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/common/hexutil"
|
||||
"github.com/ethereum/go-ethereum/log"
|
||||
"github.com/status-im/status-go/contracts"
|
||||
"github.com/status-im/status-go/contracts/ethscan"
|
||||
"github.com/status-im/status-go/contracts/ierc20"
|
||||
"github.com/status-im/status-go/rpc/chain"
|
||||
"github.com/status-im/status-go/services/wallet/async"
|
||||
)
|
||||
|
||||
var nativeChainAddress = common.HexToAddress("0x")
|
||||
var requestTimeout = 20 * time.Second
|
||||
|
||||
const (
|
||||
tokenChunkSize = 500
|
||||
)
|
||||
|
||||
type BalanceFetcher interface {
|
||||
FetchBalancesForChain(parent context.Context, client chain.ClientInterface, accounts, tokens []common.Address, atBlocks map[uint64]*big.Int) (map[common.Address]map[common.Address]*hexutil.Big, error)
|
||||
GetTokenBalanceAt(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address, blockNumber *big.Int) (*big.Int, error)
|
||||
GetBalancesAtByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address, atBlocks map[uint64]*big.Int) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error)
|
||||
GetBalancesByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error)
|
||||
GetBalance(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address) (*big.Int, error)
|
||||
GetChainBalance(ctx context.Context, client chain.ClientInterface, account common.Address) (*big.Int, error)
|
||||
}
|
||||
|
||||
type DefaultBalanceFetcher struct {
|
||||
contractMaker contracts.ContractMakerIface
|
||||
}
|
||||
|
||||
func NewDefaultBalanceFetcher(contractMaker contracts.ContractMakerIface) *DefaultBalanceFetcher {
|
||||
return &DefaultBalanceFetcher{
|
||||
contractMaker: contractMaker,
|
||||
}
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) FetchBalancesForChain(parent context.Context, client chain.ClientInterface, accounts, tokens []common.Address, atBlocks map[uint64]*big.Int) (map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
var (
|
||||
group = async.NewAtomicGroup(parent)
|
||||
mu sync.Mutex
|
||||
)
|
||||
|
||||
balances := make(map[common.Address]map[common.Address]*hexutil.Big)
|
||||
updateBalance := func(accTokenBalance map[common.Address]map[common.Address]*hexutil.Big) {
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
for account, tokenBalance := range accTokenBalance {
|
||||
if _, ok := balances[account]; !ok {
|
||||
balances[account] = make(map[common.Address]*hexutil.Big)
|
||||
}
|
||||
|
||||
for token, balance := range tokenBalance {
|
||||
if _, ok := balances[account][token]; !ok {
|
||||
zeroHex := hexutil.Big(*big.NewInt(0))
|
||||
balances[account][token] = &zeroHex
|
||||
}
|
||||
|
||||
sum := big.NewInt(0).Add(balances[account][token].ToInt(), balance.ToInt())
|
||||
sumHex := hexutil.Big(*sum)
|
||||
balances[account][token] = &sumHex
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ethScanContract, availableAtBlock, err := bf.contractMaker.NewEthScan(client.NetworkID())
|
||||
if err != nil {
|
||||
log.Error("error scanning contract", "err", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
atBlock := atBlocks[client.NetworkID()]
|
||||
|
||||
fetchChainBalance := false
|
||||
|
||||
for _, token := range tokens {
|
||||
if token == nativeChainAddress {
|
||||
fetchChainBalance = true
|
||||
}
|
||||
}
|
||||
if fetchChainBalance {
|
||||
group.Add(func(parent context.Context) error {
|
||||
balances, err := bf.FetchChainBalances(parent, accounts, ethScanContract, atBlock)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
updateBalance(balances)
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
tokenChunks := splitTokensToChunks(tokens, tokenChunkSize)
|
||||
for accountIdx := range accounts {
|
||||
// Keep the reference to the account. DO NOT USE A LOOP, the account will be overridden in the coroutine
|
||||
account := accounts[accountIdx]
|
||||
for idx := range tokenChunks {
|
||||
// Keep the reference to the chunk. DO NOT USE A LOOP, the chunk will be overridden in the coroutine
|
||||
chunk := tokenChunks[idx]
|
||||
|
||||
group.Add(func(parent context.Context) error {
|
||||
ctx, cancel := context.WithTimeout(parent, requestTimeout)
|
||||
defer cancel()
|
||||
|
||||
var accTokenBalance map[common.Address]map[common.Address]*hexutil.Big
|
||||
var err error
|
||||
if atBlock == nil || big.NewInt(int64(availableAtBlock)).Cmp(atBlock) < 0 {
|
||||
accTokenBalance, err = bf.FetchTokenBalancesWithScanContract(ctx, ethScanContract, account, chunk, atBlock)
|
||||
} else {
|
||||
accTokenBalance, err = bf.FetchTokenBalancesWithTokenContracts(ctx, client, account, chunk, atBlock)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
updateBalance(accTokenBalance)
|
||||
return nil
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
select {
|
||||
case <-group.WaitAsync():
|
||||
case <-parent.Done():
|
||||
return nil, parent.Err()
|
||||
}
|
||||
return balances, group.Error()
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) FetchChainBalances(parent context.Context, accounts []common.Address, ethScanContract *ethscan.BalanceScanner, atBlock *big.Int) (map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
accTokenBalance := make(map[common.Address]map[common.Address]*hexutil.Big)
|
||||
|
||||
ctx, cancel := context.WithTimeout(parent, requestTimeout)
|
||||
defer cancel()
|
||||
|
||||
res, err := ethScanContract.EtherBalances(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
BlockNumber: atBlock,
|
||||
}, accounts)
|
||||
if err != nil {
|
||||
log.Error("can't fetch chain balance 5", "err", err)
|
||||
return nil, err
|
||||
}
|
||||
for idx, account := range accounts {
|
||||
balance := new(big.Int)
|
||||
balance.SetBytes(res[idx].Data)
|
||||
|
||||
if _, ok := accTokenBalance[account]; !ok {
|
||||
accTokenBalance[account] = make(map[common.Address]*hexutil.Big)
|
||||
}
|
||||
|
||||
accTokenBalance[account][nativeChainAddress] = (*hexutil.Big)(balance)
|
||||
}
|
||||
|
||||
return accTokenBalance, nil
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) FetchTokenBalancesWithScanContract(ctx context.Context, ethScanContract *ethscan.BalanceScanner, account common.Address, chunk []common.Address, atBlock *big.Int) (map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
accTokenBalance := make(map[common.Address]map[common.Address]*hexutil.Big)
|
||||
res, err := ethScanContract.TokensBalance(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
BlockNumber: atBlock,
|
||||
}, account, chunk)
|
||||
if err != nil {
|
||||
log.Error("can't fetch erc20 token balance 6", "account", account, "error", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(res) != len(chunk) {
|
||||
log.Error("can't fetch erc20 token balance 7", "account", account, "error", "response not complete")
|
||||
return nil, errors.New("response not complete")
|
||||
}
|
||||
|
||||
for idx, token := range chunk {
|
||||
if !res[idx].Success {
|
||||
continue
|
||||
}
|
||||
balance := new(big.Int)
|
||||
balance.SetBytes(res[idx].Data)
|
||||
|
||||
if _, ok := accTokenBalance[account]; !ok {
|
||||
accTokenBalance[account] = make(map[common.Address]*hexutil.Big)
|
||||
}
|
||||
|
||||
accTokenBalance[account][token] = (*hexutil.Big)(balance)
|
||||
}
|
||||
return accTokenBalance, nil
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) FetchTokenBalancesWithTokenContracts(ctx context.Context, client chain.ClientInterface, account common.Address, chunk []common.Address, atBlock *big.Int) (map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
accTokenBalance := make(map[common.Address]map[common.Address]*hexutil.Big)
|
||||
for _, token := range chunk {
|
||||
balance, err := bf.GetTokenBalanceAt(ctx, client, account, token, atBlock)
|
||||
if err != nil {
|
||||
if err != bind.ErrNoCode {
|
||||
log.Error("can't fetch erc20 token balance 8", "account", account, "token", token, "error", "on fetching token balance")
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if _, ok := accTokenBalance[account]; !ok {
|
||||
accTokenBalance[account] = make(map[common.Address]*hexutil.Big)
|
||||
}
|
||||
|
||||
accTokenBalance[account][token] = (*hexutil.Big)(balance)
|
||||
}
|
||||
|
||||
return accTokenBalance, nil
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) GetTokenBalanceAt(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address, blockNumber *big.Int) (*big.Int, error) {
|
||||
caller, err := ierc20.NewIERC20Caller(token, client)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
balance, err := caller.BalanceOf(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
BlockNumber: blockNumber,
|
||||
}, account)
|
||||
|
||||
if err != nil {
|
||||
if err != bind.ErrNoCode {
|
||||
return nil, err
|
||||
}
|
||||
balance = big.NewInt(0)
|
||||
}
|
||||
|
||||
return balance, nil
|
||||
}
|
||||
|
||||
func splitTokensToChunks(tokens []common.Address, chunkSize int) [][]common.Address {
|
||||
tokenChunks := make([][]common.Address, 0)
|
||||
for i := 0; i < len(tokens); i += chunkSize {
|
||||
end := i + chunkSize
|
||||
if end > len(tokens) {
|
||||
end = len(tokens)
|
||||
}
|
||||
|
||||
tokenChunks = append(tokenChunks, tokens[i:end])
|
||||
}
|
||||
|
||||
return tokenChunks
|
||||
}
|
||||
|
||||
func (tm *DefaultBalanceFetcher) GetTokenBalance(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address) (*big.Int, error) {
|
||||
caller, err := ierc20.NewIERC20Caller(token, client)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return caller.BalanceOf(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
}, account)
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) GetChainBalance(ctx context.Context, client chain.ClientInterface, account common.Address) (*big.Int, error) {
|
||||
return client.BalanceAt(ctx, account, nil)
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) GetBalance(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address) (*big.Int, error) {
|
||||
if token == nativeChainAddress {
|
||||
return bf.GetChainBalance(ctx, client, account)
|
||||
}
|
||||
|
||||
return bf.GetTokenBalance(ctx, client, account, token)
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) GetBalancesByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
return bf.GetBalancesAtByChain(parent, clients, accounts, tokens, nil)
|
||||
}
|
||||
|
||||
func (bf *DefaultBalanceFetcher) GetBalancesAtByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address, atBlocks map[uint64]*big.Int) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
var (
|
||||
group = async.NewAtomicGroup(parent)
|
||||
mu sync.Mutex
|
||||
response = map[uint64]map[common.Address]map[common.Address]*hexutil.Big{}
|
||||
)
|
||||
|
||||
updateBalance := func(chainID uint64, accTokenBalance map[common.Address]map[common.Address]*hexutil.Big) {
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
if _, ok := response[chainID]; !ok {
|
||||
response[chainID] = map[common.Address]map[common.Address]*hexutil.Big{}
|
||||
}
|
||||
|
||||
for account, tokenBalance := range accTokenBalance {
|
||||
response[chainID][account] = tokenBalance
|
||||
}
|
||||
}
|
||||
|
||||
for clientIdx := range clients {
|
||||
// Keep the reference to the client. DO NOT USE A LOOP, the client will be overridden in the coroutine
|
||||
client := clients[clientIdx]
|
||||
|
||||
balances, err := bf.FetchBalancesForChain(parent, client, accounts, tokens, atBlocks)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
updateBalance(client.NetworkID(), balances)
|
||||
}
|
||||
select {
|
||||
case <-group.WaitAsync():
|
||||
case <-parent.Done():
|
||||
return nil, parent.Err()
|
||||
}
|
||||
return response, group.Error()
|
||||
}
|
||||
|
|
@ -19,22 +19,19 @@ import (
|
|||
"github.com/ethereum/go-ethereum/log"
|
||||
"github.com/status-im/status-go/contracts"
|
||||
"github.com/status-im/status-go/contracts/community-tokens/assets"
|
||||
"github.com/status-im/status-go/contracts/ethscan"
|
||||
"github.com/status-im/status-go/contracts/ierc20"
|
||||
eth_node_types "github.com/status-im/status-go/eth-node/types"
|
||||
"github.com/status-im/status-go/multiaccounts/accounts"
|
||||
"github.com/status-im/status-go/params"
|
||||
"github.com/status-im/status-go/protocol/communities/token"
|
||||
"github.com/status-im/status-go/rpc"
|
||||
"github.com/status-im/status-go/rpc/chain"
|
||||
"github.com/status-im/status-go/rpc/network"
|
||||
"github.com/status-im/status-go/server"
|
||||
"github.com/status-im/status-go/services/accounts/accountsevent"
|
||||
"github.com/status-im/status-go/services/communitytokens/communitytokensdatabase"
|
||||
"github.com/status-im/status-go/services/utils"
|
||||
"github.com/status-im/status-go/services/wallet/async"
|
||||
"github.com/status-im/status-go/services/wallet/bigint"
|
||||
"github.com/status-im/status-go/services/wallet/community"
|
||||
"github.com/status-im/status-go/services/wallet/token/balancefetcher"
|
||||
"github.com/status-im/status-go/services/wallet/walletevent"
|
||||
)
|
||||
|
||||
|
|
@ -42,9 +39,6 @@ const (
|
|||
EventCommunityTokenReceived walletevent.EventType = "wallet-community-token-received"
|
||||
)
|
||||
|
||||
var requestTimeout = 20 * time.Second
|
||||
var nativeChainAddress = common.HexToAddress("0x")
|
||||
|
||||
type Token struct {
|
||||
Address common.Address `json:"address"`
|
||||
Name string `json:"name"`
|
||||
|
|
@ -92,15 +86,16 @@ type addressTokenMap = map[common.Address]*Token
|
|||
type storeMap = map[uint64]addressTokenMap
|
||||
|
||||
type ManagerInterface interface {
|
||||
balancefetcher.BalanceFetcher
|
||||
LookupTokenIdentity(chainID uint64, address common.Address, native bool) *Token
|
||||
LookupToken(chainID *uint64, tokenSymbol string) (token *Token, isNative bool)
|
||||
GetTokensByChainIDs(chainIDs []uint64) ([]*Token, error)
|
||||
GetBalancesByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error)
|
||||
GetTokenHistoricalBalance(account common.Address, chainID uint64, symbol string, timestamp int64) (*big.Int, error)
|
||||
GetTokensByChainIDs(chainIDs []uint64) ([]*Token, error)
|
||||
}
|
||||
|
||||
// Manager is used for accessing token store. It changes the token store based on overridden tokens
|
||||
type Manager struct {
|
||||
balancefetcher.BalanceFetcher
|
||||
db *sql.DB
|
||||
RPCClient *rpc.Client
|
||||
ContractMaker *contracts.ContractMaker
|
||||
|
|
@ -178,9 +173,10 @@ func NewTokenManager(
|
|||
tokens := prepareTokens(networkManager, stores)
|
||||
|
||||
return &Manager{
|
||||
db: db,
|
||||
RPCClient: RPCClient,
|
||||
ContractMaker: maker,
|
||||
BalanceFetcher: balancefetcher.NewDefaultBalanceFetcher(maker),
|
||||
db: db,
|
||||
RPCClient: RPCClient,
|
||||
// ContractMaker: maker,
|
||||
networkManager: networkManager,
|
||||
communityManager: communityManager,
|
||||
stores: stores,
|
||||
|
|
@ -693,195 +689,6 @@ func (tm *Manager) DeleteCustom(chainID uint64, address common.Address) error {
|
|||
return err
|
||||
}
|
||||
|
||||
func (tm *Manager) GetTokenBalance(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address) (*big.Int, error) {
|
||||
caller, err := ierc20.NewIERC20Caller(token, client)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return caller.BalanceOf(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
}, account)
|
||||
}
|
||||
|
||||
func (tm *Manager) GetTokenBalanceAt(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address, blockNumber *big.Int) (*big.Int, error) {
|
||||
caller, err := ierc20.NewIERC20Caller(token, client)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
balance, err := caller.BalanceOf(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
BlockNumber: blockNumber,
|
||||
}, account)
|
||||
|
||||
if err != nil {
|
||||
if err != bind.ErrNoCode {
|
||||
return nil, err
|
||||
}
|
||||
balance = big.NewInt(0)
|
||||
}
|
||||
|
||||
return balance, nil
|
||||
}
|
||||
|
||||
func (tm *Manager) GetChainBalance(ctx context.Context, client chain.ClientInterface, account common.Address) (*big.Int, error) {
|
||||
return client.BalanceAt(ctx, account, nil)
|
||||
}
|
||||
|
||||
func (tm *Manager) GetBalance(ctx context.Context, client chain.ClientInterface, account common.Address, token common.Address) (*big.Int, error) {
|
||||
if token == nativeChainAddress {
|
||||
return tm.GetChainBalance(ctx, client, account)
|
||||
}
|
||||
|
||||
return tm.GetTokenBalance(ctx, client, account, token)
|
||||
}
|
||||
|
||||
func (tm *Manager) GetBalancesByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
return tm.GetBalancesAtByChain(parent, clients, accounts, tokens, nil)
|
||||
}
|
||||
|
||||
func (tm *Manager) GetBalancesAtByChain(parent context.Context, clients map[uint64]chain.ClientInterface, accounts, tokens []common.Address, atBlocks map[uint64]*big.Int) (map[uint64]map[common.Address]map[common.Address]*hexutil.Big, error) {
|
||||
var (
|
||||
group = async.NewAtomicGroup(parent)
|
||||
mu sync.Mutex
|
||||
response = map[uint64]map[common.Address]map[common.Address]*hexutil.Big{}
|
||||
)
|
||||
|
||||
updateBalance := func(chainID uint64, account common.Address, token common.Address, balance *big.Int) {
|
||||
mu.Lock()
|
||||
if _, ok := response[chainID]; !ok {
|
||||
response[chainID] = map[common.Address]map[common.Address]*hexutil.Big{}
|
||||
}
|
||||
|
||||
if _, ok := response[chainID][account]; !ok {
|
||||
response[chainID][account] = map[common.Address]*hexutil.Big{}
|
||||
}
|
||||
|
||||
if _, ok := response[chainID][account][token]; !ok {
|
||||
zeroHex := hexutil.Big(*big.NewInt(0))
|
||||
response[chainID][account][token] = &zeroHex
|
||||
}
|
||||
sum := big.NewInt(0).Add(response[chainID][account][token].ToInt(), balance)
|
||||
sumHex := hexutil.Big(*sum)
|
||||
response[chainID][account][token] = &sumHex
|
||||
mu.Unlock()
|
||||
}
|
||||
|
||||
for clientIdx := range clients {
|
||||
// Keep the reference to the client. DO NOT USE A LOOP, the client will be overridden in the coroutine
|
||||
client := clients[clientIdx]
|
||||
|
||||
ethScanContract, availableAtBlock, err := tm.ContractMaker.NewEthScan(client.NetworkID())
|
||||
if err != nil {
|
||||
log.Error("error scanning contract", "err", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
atBlock := atBlocks[client.NetworkID()]
|
||||
|
||||
fetchChainBalance := false
|
||||
var tokenChunks [][]common.Address
|
||||
chunkSize := 500
|
||||
for i := 0; i < len(tokens); i += chunkSize {
|
||||
end := i + chunkSize
|
||||
if end > len(tokens) {
|
||||
end = len(tokens)
|
||||
}
|
||||
|
||||
tokenChunks = append(tokenChunks, tokens[i:end])
|
||||
}
|
||||
|
||||
for _, token := range tokens {
|
||||
if token == nativeChainAddress {
|
||||
fetchChainBalance = true
|
||||
}
|
||||
}
|
||||
if fetchChainBalance {
|
||||
group.Add(func(parent context.Context) error {
|
||||
ctx, cancel := context.WithTimeout(parent, requestTimeout)
|
||||
defer cancel()
|
||||
res, err := ethScanContract.EtherBalances(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
BlockNumber: atBlock,
|
||||
}, accounts)
|
||||
if err != nil {
|
||||
log.Error("can't fetch chain balance 5", "err", err)
|
||||
return nil
|
||||
}
|
||||
for idx, account := range accounts {
|
||||
balance := new(big.Int)
|
||||
balance.SetBytes(res[idx].Data)
|
||||
updateBalance(client.NetworkID(), account, common.HexToAddress("0x"), balance)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
for accountIdx := range accounts {
|
||||
// Keep the reference to the account. DO NOT USE A LOOP, the account will be overridden in the coroutine
|
||||
account := accounts[accountIdx]
|
||||
for idx := range tokenChunks {
|
||||
// Keep the reference to the chunk. DO NOT USE A LOOP, the chunk will be overridden in the coroutine
|
||||
chunk := tokenChunks[idx]
|
||||
|
||||
group.Add(func(parent context.Context) error {
|
||||
ctx, cancel := context.WithTimeout(parent, requestTimeout)
|
||||
defer cancel()
|
||||
var res []ethscan.BalanceScannerResult
|
||||
if atBlock == nil || big.NewInt(int64(availableAtBlock)).Cmp(atBlock) < 0 {
|
||||
res, err = ethScanContract.TokensBalance(&bind.CallOpts{
|
||||
Context: ctx,
|
||||
BlockNumber: atBlock,
|
||||
}, account, chunk)
|
||||
if err != nil {
|
||||
log.Error("can't fetch erc20 token balance 6", "account", account, "error", err)
|
||||
return nil
|
||||
}
|
||||
|
||||
if len(res) != len(chunk) {
|
||||
log.Error("can't fetch erc20 token balance 7", "account", account, "error", "response not complete")
|
||||
return nil
|
||||
}
|
||||
|
||||
for idx, token := range chunk {
|
||||
if !res[idx].Success {
|
||||
continue
|
||||
}
|
||||
balance := new(big.Int)
|
||||
balance.SetBytes(res[idx].Data)
|
||||
updateBalance(client.NetworkID(), account, token, balance)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
for _, token := range chunk {
|
||||
balance, err := tm.GetTokenBalanceAt(ctx, client, account, token, atBlock)
|
||||
if err != nil {
|
||||
if err != bind.ErrNoCode {
|
||||
log.Error("can't fetch erc20 token balance 8", "account", account, "token", token, "error", "on fetching token balance")
|
||||
|
||||
return nil
|
||||
}
|
||||
}
|
||||
updateBalance(client.NetworkID(), account, token, balance)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
select {
|
||||
case <-group.WaitAsync():
|
||||
case <-parent.Done():
|
||||
return nil, parent.Err()
|
||||
}
|
||||
return response, group.Error()
|
||||
}
|
||||
|
||||
func (tm *Manager) SignalCommunityTokenReceived(address common.Address, txHash common.Hash, value *big.Int, t *Token, isFirst bool) {
|
||||
if tm.walletFeed == nil || t == nil || t.CommunityData == nil {
|
||||
return
|
||||
|
|
|
|||
Loading…
Reference in New Issue