13566028875

Committed 27 Feb 2025 12:09PM UTC coverage: 49.396% (-9.4%) from 58.748%

Build # 13566028875

Build Type

Pull #9555

github

Committed by

ellemouton

Commit Message

graph/db: populate the graph cache in Start instead of during construction

In this commit, we move the graph cache population logic out of the
ChannelGraph constructor and into its Start method instead.

Pull Request Pull Request #9555: graph: extract cache from CRUD [6]

Run Details

34 of 54 new or added lines in 4 files covered. (62.96%)

27464 existing lines in 436 files now uncovered.

101095 of 204664 relevant lines covered (49.4%)

1.54 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/lntest/mock/spendnotifier.go

package mock

import (
        "sync"

        "github.com/btcsuite/btcd/wire"
        "github.com/lightningnetwork/lnd/chainntnfs"
)

// SpendNotifier extends the mock.ChainNotifier so that spend
// notifications can be triggered and delivered to subscribers.
type SpendNotifier struct {
        *ChainNotifier
        spendMap map[wire.OutPoint][]chan *chainntnfs.SpendDetail
        spends   map[wire.OutPoint]*chainntnfs.SpendDetail
        mtx      sync.Mutex
}

// MakeMockSpendNotifier creates a SpendNotifier.
func MakeMockSpendNotifier() *SpendNotifier {
        return &SpendNotifier{
                ChainNotifier: &ChainNotifier{
                        SpendChan: make(chan *chainntnfs.SpendDetail),
                        EpochChan: make(chan *chainntnfs.BlockEpoch),
                        ConfChan:  make(chan *chainntnfs.TxConfirmation),
                },
                spendMap: make(map[wire.OutPoint][]chan *chainntnfs.SpendDetail),
                spends:   make(map[wire.OutPoint]*chainntnfs.SpendDetail),
        }
}

// RegisterSpendNtfn registers a spend notification for a specified outpoint.
func (s *SpendNotifier) RegisterSpendNtfn(outpoint *wire.OutPoint,
        _ []byte, heightHint uint32) (*chainntnfs.SpendEvent, error) {

        s.mtx.Lock()
        defer s.mtx.Unlock()

        spendChan := make(chan *chainntnfs.SpendDetail, 1)
        if detail, ok := s.spends[*outpoint]; ok {
                // Deliver spend immediately if details are already known.
                spendChan <- &chainntnfs.SpendDetail{
                        SpentOutPoint:     detail.SpentOutPoint,
                        SpendingHeight:    detail.SpendingHeight,
                        SpendingTx:        detail.SpendingTx,
                        SpenderTxHash:     detail.SpenderTxHash,
                        SpenderInputIndex: detail.SpenderInputIndex,
                }
        } else {
                // Otherwise, queue the notification for delivery if the spend
                // is ever received.
                s.spendMap[*outpoint] = append(s.spendMap[*outpoint], spendChan)
        }

        return &chainntnfs.SpendEvent{
                Spend:  spendChan,
                Cancel: func() {},
        }, nil
}

// Spend dispatches SpendDetails to all subscribers of the outpoint. The details
// will includethe transaction and height provided by the caller.
func (s *SpendNotifier) Spend(outpoint *wire.OutPoint, height int32,
        txn *wire.MsgTx) {

        s.mtx.Lock()
        defer s.mtx.Unlock()

        var inputIndex uint32
        for i, in := range txn.TxIn {
                if in.PreviousOutPoint == *outpoint {
                        inputIndex = uint32(i)
                }
        }

        txnHash := txn.TxHash()
        details := &chainntnfs.SpendDetail{
                SpentOutPoint:     outpoint,
                SpendingHeight:    height,
                SpendingTx:        txn,
                SpenderTxHash:     &txnHash,
                SpenderInputIndex: inputIndex,
        }

        // Cache details in case of late registration.
        if _, ok := s.spends[*outpoint]; !ok {
                s.spends[*outpoint] = details
        }

        // Deliver any backlogged spend notifications.
        if spendChans, ok := s.spendMap[*outpoint]; ok {
                delete(s.spendMap, *outpoint)
                for _, spendChan := range spendChans {
                        spendChan <- &chainntnfs.SpendDetail{
                                SpentOutPoint:     details.SpentOutPoint,
                                SpendingHeight:    details.SpendingHeight,
                                SpendingTx:        details.SpendingTx,
                                SpenderTxHash:     details.SpenderTxHash,
                                SpenderInputIndex: details.SpenderInputIndex,
                        }
                }
        }
}

1	package mock
2
3	import (
4	"sync"
5
6	"github.com/btcsuite/btcd/wire"
7	"github.com/lightningnetwork/lnd/chainntnfs"
8	)
9
10	// SpendNotifier extends the mock.ChainNotifier so that spend
11	// notifications can be triggered and delivered to subscribers.
12	type SpendNotifier struct {
13	*ChainNotifier
14	spendMap map[wire.OutPoint][]chan *chainntnfs.SpendDetail
15	spends map[wire.OutPoint]*chainntnfs.SpendDetail
16	mtx sync.Mutex
17	}
18
19	// MakeMockSpendNotifier creates a SpendNotifier.
UNCOV 20	func MakeMockSpendNotifier() *SpendNotifier {	×
UNCOV 21	return &SpendNotifier{	×
UNCOV 22	ChainNotifier: &ChainNotifier{	×
UNCOV 23	SpendChan: make(chan *chainntnfs.SpendDetail),	×
UNCOV 24	EpochChan: make(chan *chainntnfs.BlockEpoch),	×
UNCOV 25	ConfChan: make(chan *chainntnfs.TxConfirmation),	×
UNCOV 26	},	×
UNCOV 27	spendMap: make(map[wire.OutPoint][]chan *chainntnfs.SpendDetail),	×
UNCOV 28	spends: make(map[wire.OutPoint]*chainntnfs.SpendDetail),	×
UNCOV 29	}	×
UNCOV 30	}	×
31
32	// RegisterSpendNtfn registers a spend notification for a specified outpoint.
33	func (s SpendNotifier) RegisterSpendNtfn(outpoint wire.OutPoint,
UNCOV 34	_ []byte, heightHint uint32) (*chainntnfs.SpendEvent, error) {	×
UNCOV 35		×
UNCOV 36	s.mtx.Lock()	×
UNCOV 37	defer s.mtx.Unlock()	×
UNCOV 38		×
UNCOV 39	spendChan := make(chan *chainntnfs.SpendDetail, 1)	×
UNCOV 40	if detail, ok := s.spends[*outpoint]; ok {	×
41	// Deliver spend immediately if details are already known.	×
42	spendChan <- &chainntnfs.SpendDetail{	×
43	SpentOutPoint: detail.SpentOutPoint,	×
44	SpendingHeight: detail.SpendingHeight,	×
45	SpendingTx: detail.SpendingTx,	×
46	SpenderTxHash: detail.SpenderTxHash,	×
47	SpenderInputIndex: detail.SpenderInputIndex,	×
48	}	×
UNCOV 49	} else {	×
UNCOV 50	// Otherwise, queue the notification for delivery if the spend	×
UNCOV 51	// is ever received.	×
UNCOV 52	s.spendMap[outpoint] = append(s.spendMap[outpoint], spendChan)	×
UNCOV 53	}	×
54
UNCOV 55	return &chainntnfs.SpendEvent{	×
UNCOV 56	Spend: spendChan,	×
UNCOV 57	Cancel: func() {},	×
58	}, nil
59	}
60
61	// Spend dispatches SpendDetails to all subscribers of the outpoint. The details
62	// will includethe transaction and height provided by the caller.
63	func (s SpendNotifier) Spend(outpoint wire.OutPoint, height int32,
UNCOV 64	txn *wire.MsgTx) {	×
UNCOV 65		×
UNCOV 66	s.mtx.Lock()	×
UNCOV 67	defer s.mtx.Unlock()	×
UNCOV 68		×
UNCOV 69	var inputIndex uint32	×
UNCOV 70	for i, in := range txn.TxIn {	×
UNCOV 71	if in.PreviousOutPoint == *outpoint {	×
UNCOV 72	inputIndex = uint32(i)	×
UNCOV 73	}	×
74	}
75
UNCOV 76	txnHash := txn.TxHash()	×
UNCOV 77	details := &chainntnfs.SpendDetail{	×
UNCOV 78	SpentOutPoint: outpoint,	×
UNCOV 79	SpendingHeight: height,	×
UNCOV 80	SpendingTx: txn,	×
UNCOV 81	SpenderTxHash: &txnHash,	×
UNCOV 82	SpenderInputIndex: inputIndex,	×
UNCOV 83	}	×
UNCOV 84		×
UNCOV 85	// Cache details in case of late registration.	×
UNCOV 86	if _, ok := s.spends[*outpoint]; !ok {	×
UNCOV 87	s.spends[*outpoint] = details	×
UNCOV 88	}	×
89
90	// Deliver any backlogged spend notifications.
UNCOV 91	if spendChans, ok := s.spendMap[*outpoint]; ok {	×
UNCOV 92	delete(s.spendMap, *outpoint)	×
UNCOV 93	for _, spendChan := range spendChans {	×
UNCOV 94	spendChan <- &chainntnfs.SpendDetail{	×
UNCOV 95	SpentOutPoint: details.SpentOutPoint,	×
UNCOV 96	SpendingHeight: details.SpendingHeight,	×
UNCOV 97	SpendingTx: details.SpendingTx,	×
UNCOV 98	SpenderTxHash: details.SpenderTxHash,	×
UNCOV 99	SpenderInputIndex: details.SpenderInputIndex,	×
UNCOV 100	}	×
UNCOV 101	}	×
102	}
103	}

lightningnetwork / lnd / 13566028875

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous