mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-02-23 14:40:30 +01:00
routing: remove redundant GetHash
This commit is contained in:
parent
ce8cde6911
commit
46eb811543
2 changed files with 8 additions and 50 deletions
|
@ -488,17 +488,9 @@ func (p *paymentLifecycle) collectResult(attempt *channeldb.HTLCAttempt) (
|
||||||
|
|
||||||
log.Tracef("Collecting result for attempt %v", spew.Sdump(attempt))
|
log.Tracef("Collecting result for attempt %v", spew.Sdump(attempt))
|
||||||
|
|
||||||
// We'll retrieve the hash specific to this shard from the
|
|
||||||
// shardTracker, since it will be needed to regenerate the circuit
|
|
||||||
// below.
|
|
||||||
hash, err := p.shardTracker.GetHash(attempt.AttemptID)
|
|
||||||
if err != nil {
|
|
||||||
return p.failAttempt(attempt.AttemptID, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Regenerate the circuit for this attempt.
|
// Regenerate the circuit for this attempt.
|
||||||
_, circuit, err := generateSphinxPacket(
|
_, circuit, err := generateSphinxPacket(
|
||||||
&attempt.Route, hash[:], attempt.SessionKey(),
|
&attempt.Route, attempt.Hash[:], attempt.SessionKey(),
|
||||||
)
|
)
|
||||||
// TODO(yy): We generate this circuit to create the error decryptor,
|
// TODO(yy): We generate this circuit to create the error decryptor,
|
||||||
// which is then used in htlcswitch as the deobfuscator to decode the
|
// which is then used in htlcswitch as the deobfuscator to decode the
|
||||||
|
|
|
@ -260,10 +260,15 @@ func createDummyRoute(t *testing.T, amt lnwire.MilliSatoshi) *route.Route {
|
||||||
func makeSettledAttempt(t *testing.T, total int,
|
func makeSettledAttempt(t *testing.T, total int,
|
||||||
preimage lntypes.Preimage) *channeldb.HTLCAttempt {
|
preimage lntypes.Preimage) *channeldb.HTLCAttempt {
|
||||||
|
|
||||||
return &channeldb.HTLCAttempt{
|
a := &channeldb.HTLCAttempt{
|
||||||
HTLCAttemptInfo: makeAttemptInfo(t, total),
|
HTLCAttemptInfo: makeAttemptInfo(t, total),
|
||||||
Settle: &channeldb.HTLCSettleInfo{Preimage: preimage},
|
Settle: &channeldb.HTLCSettleInfo{Preimage: preimage},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
hash := preimage.Hash()
|
||||||
|
a.Hash = &hash
|
||||||
|
|
||||||
|
return a
|
||||||
}
|
}
|
||||||
|
|
||||||
func makeFailedAttempt(t *testing.T, total int) *channeldb.HTLCAttempt {
|
func makeFailedAttempt(t *testing.T, total int) *channeldb.HTLCAttempt {
|
||||||
|
@ -279,6 +284,7 @@ func makeAttemptInfo(t *testing.T, amt int) channeldb.HTLCAttemptInfo {
|
||||||
rt := createDummyRoute(t, lnwire.MilliSatoshi(amt))
|
rt := createDummyRoute(t, lnwire.MilliSatoshi(amt))
|
||||||
return channeldb.HTLCAttemptInfo{
|
return channeldb.HTLCAttemptInfo{
|
||||||
Route: *rt,
|
Route: *rt,
|
||||||
|
Hash: &lntypes.Hash{1, 2, 3},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1303,11 +1309,6 @@ func TestCollectResultExitOnErr(t *testing.T) {
|
||||||
paymentAmt := 10_000
|
paymentAmt := 10_000
|
||||||
attempt := makeFailedAttempt(t, paymentAmt)
|
attempt := makeFailedAttempt(t, paymentAmt)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a dummy error.
|
// Mock the htlcswitch to return a dummy error.
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
attempt.AttemptID, p.identifier, mock.Anything,
|
attempt.AttemptID, p.identifier, mock.Anything,
|
||||||
|
@ -1348,11 +1349,6 @@ func TestCollectResultExitOnResultErr(t *testing.T) {
|
||||||
paymentAmt := 10_000
|
paymentAmt := 10_000
|
||||||
attempt := makeFailedAttempt(t, paymentAmt)
|
attempt := makeFailedAttempt(t, paymentAmt)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
@ -1399,11 +1395,6 @@ func TestCollectResultExitOnSwitchQuit(t *testing.T) {
|
||||||
paymentAmt := 10_000
|
paymentAmt := 10_000
|
||||||
attempt := makeFailedAttempt(t, paymentAmt)
|
attempt := makeFailedAttempt(t, paymentAmt)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
@ -1431,11 +1422,6 @@ func TestCollectResultExitOnRouterQuit(t *testing.T) {
|
||||||
paymentAmt := 10_000
|
paymentAmt := 10_000
|
||||||
attempt := makeFailedAttempt(t, paymentAmt)
|
attempt := makeFailedAttempt(t, paymentAmt)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
@ -1462,11 +1448,6 @@ func TestCollectResultExitOnLifecycleQuit(t *testing.T) {
|
||||||
paymentAmt := 10_000
|
paymentAmt := 10_000
|
||||||
attempt := makeFailedAttempt(t, paymentAmt)
|
attempt := makeFailedAttempt(t, paymentAmt)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
@ -1495,11 +1476,6 @@ func TestCollectResultExitOnSettleErr(t *testing.T) {
|
||||||
preimage := lntypes.Preimage{1}
|
preimage := lntypes.Preimage{1}
|
||||||
attempt := makeSettledAttempt(t, paymentAmt, preimage)
|
attempt := makeSettledAttempt(t, paymentAmt, preimage)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
@ -1542,11 +1518,6 @@ func TestCollectResultSuccess(t *testing.T) {
|
||||||
preimage := lntypes.Preimage{1}
|
preimage := lntypes.Preimage{1}
|
||||||
attempt := makeSettledAttempt(t, paymentAmt, preimage)
|
attempt := makeSettledAttempt(t, paymentAmt, preimage)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
@ -1590,11 +1561,6 @@ func TestCollectResultAsyncSuccess(t *testing.T) {
|
||||||
preimage := lntypes.Preimage{1}
|
preimage := lntypes.Preimage{1}
|
||||||
attempt := makeSettledAttempt(t, paymentAmt, preimage)
|
attempt := makeSettledAttempt(t, paymentAmt, preimage)
|
||||||
|
|
||||||
// Mock shardTracker to return the payment hash.
|
|
||||||
m.shardTracker.On("GetHash",
|
|
||||||
attempt.AttemptID,
|
|
||||||
).Return(p.identifier, nil).Once()
|
|
||||||
|
|
||||||
// Mock the htlcswitch to return a the result chan.
|
// Mock the htlcswitch to return a the result chan.
|
||||||
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
resultChan := make(chan *htlcswitch.PaymentResult, 1)
|
||||||
m.payer.On("GetAttemptResult",
|
m.payer.On("GetAttemptResult",
|
||||||
|
|
Loading…
Add table
Reference in a new issue