mock: support late registration of spend ntfns

This commit is contained in:
Conner Fromknecht 2019-03-19 19:22:47 -07:00
parent 997aa3ecf0
commit 6f06c30421
No known key found for this signature in database
GPG Key ID: E7D737B67FA592C7

46
mock.go

@ -123,6 +123,7 @@ func (m *mockNotfier) RegisterSpendNtfn(outpoint *wire.OutPoint, _ []byte,
type mockSpendNotifier struct { type mockSpendNotifier struct {
*mockNotfier *mockNotfier
spendMap map[wire.OutPoint][]chan *chainntnfs.SpendDetail spendMap map[wire.OutPoint][]chan *chainntnfs.SpendDetail
spends map[wire.OutPoint]*chainntnfs.SpendDetail
mtx sync.Mutex mtx sync.Mutex
} }
@ -132,6 +133,7 @@ func makeMockSpendNotifier() *mockSpendNotifier {
confChannel: make(chan *chainntnfs.TxConfirmation), confChannel: make(chan *chainntnfs.TxConfirmation),
}, },
spendMap: make(map[wire.OutPoint][]chan *chainntnfs.SpendDetail), spendMap: make(map[wire.OutPoint][]chan *chainntnfs.SpendDetail),
spends: make(map[wire.OutPoint]*chainntnfs.SpendDetail),
} }
} }
@ -140,8 +142,22 @@ func (m *mockSpendNotifier) RegisterSpendNtfn(outpoint *wire.OutPoint,
m.mtx.Lock() m.mtx.Lock()
defer m.mtx.Unlock() defer m.mtx.Unlock()
spendChan := make(chan *chainntnfs.SpendDetail) spendChan := make(chan *chainntnfs.SpendDetail, 1)
m.spendMap[*outpoint] = append(m.spendMap[*outpoint], spendChan) if detail, ok := m.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.
m.spendMap[*outpoint] = append(m.spendMap[*outpoint], spendChan)
}
return &chainntnfs.SpendEvent{ return &chainntnfs.SpendEvent{
Spend: spendChan, Spend: spendChan,
Cancel: func() { Cancel: func() {
@ -156,16 +172,30 @@ func (m *mockSpendNotifier) Spend(outpoint *wire.OutPoint, height int32,
m.mtx.Lock() m.mtx.Lock()
defer m.mtx.Unlock() defer m.mtx.Unlock()
txnHash := txn.TxHash()
details := &chainntnfs.SpendDetail{
SpentOutPoint: outpoint,
SpendingHeight: height,
SpendingTx: txn,
SpenderTxHash: &txnHash,
SpenderInputIndex: outpoint.Index,
}
// Cache details in case of late registration.
if _, ok := m.spends[*outpoint]; !ok {
m.spends[*outpoint] = details
}
// Deliver any backlogged spend notifications.
if spendChans, ok := m.spendMap[*outpoint]; ok { if spendChans, ok := m.spendMap[*outpoint]; ok {
delete(m.spendMap, *outpoint) delete(m.spendMap, *outpoint)
for _, spendChan := range spendChans { for _, spendChan := range spendChans {
txnHash := txn.TxHash()
spendChan <- &chainntnfs.SpendDetail{ spendChan <- &chainntnfs.SpendDetail{
SpentOutPoint: outpoint, SpentOutPoint: details.SpentOutPoint,
SpendingHeight: height, SpendingHeight: details.SpendingHeight,
SpendingTx: txn, SpendingTx: details.SpendingTx,
SpenderTxHash: &txnHash, SpenderTxHash: details.SpenderTxHash,
SpenderInputIndex: outpoint.Index, SpenderInputIndex: details.SpenderInputIndex,
} }
} }
} }