mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-29 10:16:45 +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>
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
package rpchandlers
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/rpc/rpccontext"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/txscript"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/kaspanet/kaspad/util"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// HandleGetBalanceByAddress handles the respectively named RPC command
|
|
func HandleGetBalanceByAddress(context *rpccontext.Context, _ *router.Router, request appmessage.Message) (appmessage.Message, error) {
|
|
if !context.Config.UTXOIndex {
|
|
errorMessage := &appmessage.GetUTXOsByAddressesResponseMessage{}
|
|
errorMessage.Error = appmessage.RPCErrorf("Method unavailable when kaspad is run without --utxoindex")
|
|
return errorMessage, nil
|
|
}
|
|
|
|
getBalanceByAddressRequest := request.(*appmessage.GetBalanceByAddressRequestMessage)
|
|
|
|
balance, err := getBalanceByAddress(context, getBalanceByAddressRequest.Address)
|
|
if err != nil {
|
|
rpcError := &appmessage.RPCError{}
|
|
if !errors.As(err, rpcError) {
|
|
return nil, err
|
|
}
|
|
errorMessage := &appmessage.GetUTXOsByAddressesResponseMessage{}
|
|
errorMessage.Error = rpcError
|
|
return errorMessage, nil
|
|
}
|
|
|
|
response := appmessage.NewGetBalanceByAddressResponse(balance)
|
|
return response, nil
|
|
}
|
|
|
|
func getBalanceByAddress(context *rpccontext.Context, addressString string) (uint64, error) {
|
|
address, err := util.DecodeAddress(addressString, context.Config.ActiveNetParams.Prefix)
|
|
if err != nil {
|
|
return 0, appmessage.RPCErrorf("Couldn't decode address '%s': %s", addressString, err)
|
|
}
|
|
|
|
scriptPublicKey, err := txscript.PayToAddrScript(address)
|
|
if err != nil {
|
|
return 0, appmessage.RPCErrorf("Could not create a scriptPublicKey for address '%s': %s", addressString, err)
|
|
}
|
|
utxoOutpointEntryPairs, err := context.UTXOIndex.UTXOs(scriptPublicKey)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
balance := uint64(0)
|
|
for _, utxoOutpointEntryPair := range utxoOutpointEntryPairs {
|
|
balance += utxoOutpointEntryPair.Amount()
|
|
}
|
|
return balance, nil
|
|
}
|