router: use SetID as identifier for SendToRoute AMP-payments

satindergrewal · Apr 27, 2021 · 56a2c65 · 56a2c65
1 parent f07c9d0
commit 56a2c65
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/routing/router.go b/routing/router.go
@@ -1970,6 +1970,13 @@ func (r *ChannelRouter) SendToRoute(htlcHash lntypes.Hash, rt *route.Route) (
 	// hash as used for this HTLC.
 	paymentIdentifier := htlcHash
 
+	// For AMP-payments, we'll use the setID as the unique ID for the
+	// overall payment.
+	amp := finalHop.AMP
+	if amp != nil {
+		paymentIdentifier = amp.SetID()
+	}
+
 	// Record this payment hash with the ControlTower, ensuring it is not
 	// already in-flight.
 	info := &channeldb.PaymentCreationInfo{