mirror of
https://github.com/luxfi/node.git
synced 2026-10-30 15:24:05 +00:00
zeekay
7cf87b230b
The module already called itself github.com/luxfi/node while containing no node in Go: cmd/luxd was twenty-one lines that printed the size of a few types, and the Go column of the matrix measured a VM plugin. A repo that fetches the thing it claims to be has not replaced it, so the Go sources move in — 1515 files, the daemon and everything under it — and the build reaches outside this module for nothing. Rust and C++ sit beside it unchanged and are still measured against it by the differential. That is the point of one repo: the reference implementation and the two ports are the same checkout, so parity is a thing you can run rather than a thing you assert. Verified: builds, and the closure carries no lux-private — checked at build time rather than believed. The daemon starts twelve chains on a local network (P X C Q Z A B G K M F, plus D through its plugin) and reports healthy.
111 lines
2.9 KiB
Go
111 lines
2.9 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package lux
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"math"
|
|
|
|
"github.com/luxfi/ids"
|
|
"github.com/luxfi/math/set"
|
|
"github.com/luxfi/util"
|
|
|
|
safemath "github.com/luxfi/math"
|
|
)
|
|
|
|
// GetBalance returns the current balance of [addrs]
|
|
func GetBalance(db UTXOReader, addrs set.Set[ids.ShortID]) (uint64, error) {
|
|
utxos, err := GetAllUTXOs(db, addrs)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("couldn't get UTXOs: %w", err)
|
|
}
|
|
balance := uint64(0)
|
|
for _, utxo := range utxos {
|
|
if out, ok := utxo.Out.(Amounter); ok {
|
|
balance, err = safemath.Add64(out.Amount(), balance)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
}
|
|
return balance, nil
|
|
}
|
|
|
|
func GetAllUTXOs(db UTXOReader, addrs set.Set[ids.ShortID]) ([]*UTXO, error) {
|
|
utxos, _, _, err := GetPaginatedUTXOs(
|
|
db,
|
|
addrs,
|
|
ids.ShortEmpty,
|
|
ids.Empty,
|
|
math.MaxInt,
|
|
)
|
|
return utxos, err
|
|
}
|
|
|
|
// GetPaginatedUTXOs returns UTXOs such that at least one of the addresses in
|
|
// [addrs] is referenced.
|
|
//
|
|
// Returns at most [limit] UTXOs.
|
|
//
|
|
// Only returns UTXOs associated with addresses >= [startAddr].
|
|
//
|
|
// For address [startAddr], only returns UTXOs whose IDs are greater than
|
|
// [startUTXOID].
|
|
//
|
|
// Returns:
|
|
// * The fetched UTXOs
|
|
// * The address associated with the last UTXO fetched
|
|
// * The ID of the last UTXO fetched
|
|
func GetPaginatedUTXOs(
|
|
db UTXOReader,
|
|
addrs set.Set[ids.ShortID],
|
|
lastAddr ids.ShortID,
|
|
lastUTXOID ids.ID,
|
|
limit int,
|
|
) ([]*UTXO, ids.ShortID, ids.ID, error) {
|
|
var (
|
|
utxos []*UTXO
|
|
seen = set.NewSet[ids.ID](limit) // IDs of UTXOs already in the list
|
|
searchSize = limit // the limit diminishes which can impact the expected return
|
|
addrsList = addrs.List()
|
|
)
|
|
utils.Sort(addrsList) // enforces the same ordering for pagination
|
|
for _, addr := range addrsList {
|
|
start := ids.Empty
|
|
if comp := bytes.Compare(addr.Bytes(), lastAddr.Bytes()); comp == -1 { // Skip addresses before [startAddr]
|
|
continue
|
|
} else if comp == 0 {
|
|
start = lastUTXOID
|
|
}
|
|
|
|
lastAddr = addr // The last address searched
|
|
|
|
utxoIDs, err := db.UTXOIDs(addr.Bytes(), start, searchSize) // Get UTXOs associated with [addr]
|
|
if err != nil {
|
|
return nil, ids.ShortID{}, ids.Empty, fmt.Errorf("couldn't get UTXOs for address %s: %w", addr, err)
|
|
}
|
|
for _, utxoID := range utxoIDs {
|
|
lastUTXOID = utxoID // The last searched UTXO - not the last found
|
|
|
|
if seen.Contains(utxoID) { // Already have this UTXO in the list
|
|
continue
|
|
}
|
|
|
|
utxo, err := db.GetUTXO(utxoID)
|
|
if err != nil {
|
|
return nil, ids.ShortID{}, ids.Empty, fmt.Errorf("couldn't get UTXO %s: %w", utxoID, err)
|
|
}
|
|
|
|
utxos = append(utxos, utxo)
|
|
seen.Add(utxoID)
|
|
limit--
|
|
if limit <= 0 {
|
|
return utxos, lastAddr, lastUTXOID, nil // Found [limit] utxos; stop.
|
|
}
|
|
}
|
|
}
|
|
return utxos, lastAddr, lastUTXOID, nil // Didn't reach the [limit] utxos; no more were found
|
|
}
|