mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-22 06:46:46 +00:00
be3a6604d7
* Make kaspawallet store the utxos sorted by amount, so that the bigger utxos are spent first - making it less likely a compound will be required * Start refactor addEntryToUTXOSet * Add GetUTXOsByBalances command to rpc * Store list of addresses, updated with the collectAddresses methods (replacing collectUTXOs methods) * Fix wrong commands in GetBalanceByAddress * Rename: refreshExistingUTXOs -> refreshUTXOs Co-authored-by: Ori Newman <orinewman1@gmail.com>
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/kaspanet/kaspad/cmd/kaspawallet/daemon/pb"
|
|
"github.com/kaspanet/kaspad/cmd/kaspawallet/libkaspawallet"
|
|
)
|
|
|
|
type balancesType struct{ available, pending uint64 }
|
|
type balancesMapType map[*walletAddress]*balancesType
|
|
|
|
func (s *server) GetBalance(_ context.Context, _ *pb.GetBalanceRequest) (*pb.GetBalanceResponse, error) {
|
|
s.lock.RLock()
|
|
defer s.lock.RUnlock()
|
|
|
|
dagInfo, err := s.rpcClient.GetBlockDAGInfo()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
daaScore := dagInfo.VirtualDAAScore
|
|
maturity := s.params.BlockCoinbaseMaturity
|
|
|
|
balancesMap := make(balancesMapType, 0)
|
|
for _, entry := range s.utxosSortedByAmount {
|
|
amount := entry.UTXOEntry.Amount()
|
|
address := entry.address
|
|
balances, ok := balancesMap[address]
|
|
if !ok {
|
|
balances = new(balancesType)
|
|
balancesMap[address] = balances
|
|
}
|
|
if isUTXOSpendable(entry, daaScore, maturity) {
|
|
balances.available += amount
|
|
} else {
|
|
balances.pending += amount
|
|
}
|
|
}
|
|
|
|
addressBalances := make([]*pb.AddressBalances, len(balancesMap))
|
|
i := 0
|
|
var available, pending uint64
|
|
for walletAddress, balances := range balancesMap {
|
|
address, err := libkaspawallet.Address(s.params, s.keysFile.ExtendedPublicKeys, s.keysFile.MinimumSignatures, s.walletAddressPath(walletAddress), s.keysFile.ECDSA)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
addressBalances[i] = &pb.AddressBalances{
|
|
Address: address.String(),
|
|
Available: balances.available,
|
|
Pending: balances.pending,
|
|
}
|
|
i++
|
|
available += balances.available
|
|
pending += balances.pending
|
|
}
|
|
|
|
return &pb.GetBalanceResponse{
|
|
Available: available,
|
|
Pending: pending,
|
|
AddressBalances: addressBalances,
|
|
}, nil
|
|
}
|
|
|
|
func isUTXOSpendable(entry *walletUTXO, virtualDAAScore uint64, coinbaseMaturity uint64) bool {
|
|
if !entry.UTXOEntry.IsCoinbase() {
|
|
return true
|
|
}
|
|
return entry.UTXOEntry.BlockDAAScore()+coinbaseMaturity < virtualDAAScore
|
|
}
|