highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-05-15 22:30:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

routing: add htlcAmt to PaymentBandwidth method of TlvTrafficShaper

Browse Source 
This commit was added to the 0-19-staging branch recently and therefore
didn't make it into a previous part yet. So it's unrelated to the
changes in this part but is required for the whole custom channel saga.
This commit is contained in:
George Tsagkarelis 2024-09-12 14:39:51 +02:00 committed by Olaoluwa Osuntokun
parent 6485a816d1
commit d1a2dd6bd0
No known key found for this signature in database
GPG Key ID: 3BBD59E99B280306
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
routing/bandwidth.go
View File

@ -47,7 +47,8 @@ type TlvTrafficShaper interface {
// is a custom channel that should be handled by the traffic shaper, the // is a custom channel that should be handled by the traffic shaper, the
// HandleTraffic method should be called first. // HandleTraffic method should be called first.
PaymentBandwidth(htlcBlob, commitmentBlob fn.Option[tlv.Blob], PaymentBandwidth(htlcBlob, commitmentBlob fn.Option[tlv.Blob],
linkBandwidth lnwire.MilliSatoshi) (lnwire.MilliSatoshi, error) linkBandwidth,
htlcAmt lnwire.MilliSatoshi) (lnwire.MilliSatoshi, error)
} }
// AuxHtlcModifier is an interface that allows the sender to modify the outgoing // AuxHtlcModifier is an interface that allows the sender to modify the outgoing
@ -191,6 +192,7 @@ func (b *bandwidthManager) getBandwidth(cid lnwire.ShortChannelID,
commitmentBlob := link.CommitmentCustomBlob() commitmentBlob := link.CommitmentCustomBlob()
auxBandwidth, err := ts.PaymentBandwidth( auxBandwidth, err := ts.PaymentBandwidth(
b.firstHopBlob, commitmentBlob, linkBandwidth, b.firstHopBlob, commitmentBlob, linkBandwidth,
amount,
) )
if err != nil { if err != nil {
return bandwidthErr(fmt.Errorf("failed to get "+ return bandwidthErr(fmt.Errorf("failed to get "+

2
routing/bandwidth_test.go
View File

@ -148,7 +148,7 @@ func (*mockTrafficShaper) ShouldHandleTraffic(_ lnwire.ShortChannelID,
// is a custom channel that should be handled by the traffic shaper, the // is a custom channel that should be handled by the traffic shaper, the
// HandleTraffic method should be called first. // HandleTraffic method should be called first.
func (*mockTrafficShaper) PaymentBandwidth(_, _ fn.Option[tlv.Blob], func (*mockTrafficShaper) PaymentBandwidth(_, _ fn.Option[tlv.Blob],
linkBandwidth lnwire.MilliSatoshi) (lnwire.MilliSatoshi, error) { linkBandwidth, _ lnwire.MilliSatoshi) (lnwire.MilliSatoshi, error) {
return linkBandwidth, nil return linkBandwidth, nil
} }