wtxmgr: add ListLeases

This commit is contained in:
Joost Jager 2021-03-12 09:26:37 +01:00
parent 39cbb7bdd9
commit ff2577a0e7
3 changed files with 52 additions and 8 deletions

View file

@ -2659,6 +2659,19 @@ func (w *Wallet) ListUnspent(minconf, maxconf int32,
return results, err return results, err
} }
// ListLeasedOutputs returns a list of objects representing the currently locked
// utxos.
func (w *Wallet) ListLeasedOutputs() ([]*wtxmgr.LockedOutput, error) {
var outputs []*wtxmgr.LockedOutput
err := walletdb.View(w.db, func(tx walletdb.ReadTx) error {
ns := tx.ReadBucket(wtxmgrNamespaceKey)
var err error
outputs, err = w.TxStore.ListLockedOutputs(ns)
return err
})
return outputs, err
}
// DumpPrivKeys returns the WIF-encoded private keys for all addresses with // DumpPrivKeys returns the WIF-encoded private keys for all addresses with
// private keys in a wallet. // private keys in a wallet.
func (w *Wallet) DumpPrivKeys() ([]string, error) { func (w *Wallet) DumpPrivKeys() ([]string, error) {

View file

@ -125,6 +125,14 @@ type TxRecord struct {
SerializedTx []byte // Optional: may be nil SerializedTx []byte // Optional: may be nil
} }
// LockedOutput is a type that contains an outpoint of an UTXO and its lock
// lease information.
type LockedOutput struct {
Outpoint wire.OutPoint
LockID LockID
Expiration time.Time
}
// NewTxRecord creates a new transaction record that may be inserted into the // NewTxRecord creates a new transaction record that may be inserted into the
// store. It uses memoization to save the transaction hash and the serialized // store. It uses memoization to save the transaction hash and the serialized
// transaction. // transaction.
@ -1226,3 +1234,25 @@ func (s *Store) DeleteExpiredLockedOutputs(ns walletdb.ReadWriteBucket) error {
return nil return nil
} }
// ListLockedOutputs returns a list of objects representing the currently locked
// utxos.
func (s *Store) ListLockedOutputs(ns walletdb.ReadBucket) ([]*LockedOutput,
error) {
var outputs []*LockedOutput
err := forEachLockedOutput(
ns, func(op wire.OutPoint, id LockID, expiration time.Time) {
outputs = append(outputs, &LockedOutput{
Outpoint: op,
LockID: id,
Expiration: expiration,
})
},
)
if err != nil {
return nil, err
}
return outputs, nil
}

View file

@ -2425,19 +2425,20 @@ func assertOutputLocksExist(t *testing.T, s *Store, ns walletdb.ReadBucket,
t.Helper() t.Helper()
var found []wire.OutPoint outputs, err := s.ListLockedOutputs(ns)
forEachLockedOutput(ns, func(op wire.OutPoint, _ LockID, _ time.Time) { if err != nil {
found = append(found, op) t.Fatal(err)
}) }
if len(found) != len(exp) {
if len(outputs) != len(exp) {
t.Fatalf("expected to find %v locked output(s), found %v", t.Fatalf("expected to find %v locked output(s), found %v",
len(exp), len(found)) len(exp), len(outputs))
} }
for _, expOp := range exp { for _, expOp := range exp {
exists := false exists := false
for _, foundOp := range found { for _, found := range outputs {
if expOp == foundOp { if expOp == found.Outpoint {
exists = true exists = true
break break
} }