poanetwork
/
gecko
mirror of https://github.com/poanetwork/gecko.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

don't try to sort balances since maps aren't sorted by key. simplify balance calculation since zero value of uint64 is 0.

This commit is contained in:
Dan Laine 2020-05-01 13:28:38 -04:00
parent a0bd9cdcfb
commit 10aa724d30
1 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
vms/avm/service.go
View File

@ -271,22 +271,17 @@ func (service *Service) GetAllBalances(r *http.Request, args *GetAllBalancesArgs
}
assetID := utxo.AssetID()
assetIDs.Add(assetID)
if balance, ok := balances[assetID.Key()]; ok {
balance, err := safemath.Add64(transferable.Amount(), balance)
if err != nil {
balances[assetID.Key()] = math.MaxUint64
} else {
balances[assetID.Key()] = balance
}
balance := balances[assetID.Key()] // 0 if key doesn't exist
balance, err := safemath.Add64(transferable.Amount(), balance)
if err != nil {
balances[assetID.Key()] = math.MaxUint64
} else {
balances[assetID.Key()] = transferable.Amount()
balances[assetID.Key()] = balance
}
}
sortedAssetIDs := assetIDs.List() // sort so response is always in same order
ids.SortIDs(sortedAssetIDs)
reply.Balances = make(map[string]json.Uint64, len(sortedAssetIDs))
for _, assetID := range sortedAssetIDs {
reply.Balances = make(map[string]json.Uint64, assetIDs.Len())
for _, assetID := range assetIDs.List() {
reply.Balances[assetID.String()] = json.Uint64(balances[assetID.Key()])
}