highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-06-26 00:31:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

routing: update TLV traffic shaper interface

Browse Source
This commit is contained in:
Oliver Gugger 2024-05-24 17:18:29 +02:00
parent 1404468299
commit b628483120
No known key found for this signature in database
GPG Key ID: 8E4256593F177720
2 changed files with 9 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
routing/bandwidth.go
View File

@ -3,7 +3,6 @@ package routing
import (
"fmt"
"github.com/btcsuite/btcd/btcutil"
"github.com/lightningnetwork/lnd/channeldb"
"github.com/lightningnetwork/lnd/fn"
"github.com/lightningnetwork/lnd/htlcswitch"
@ -55,7 +54,8 @@ type AuxHtlcModifier interface {
// data blob of an HTLC, may produce a different blob or modify the
// amount of bitcoin this htlc should carry.
ProduceHtlcExtraData(totalAmount lnwire.MilliSatoshi,
htlcBlob tlv.Blob) (btcutil.Amount, tlv.Blob, error)
htlcCustomRecords lnwire.CustomRecords) (lnwire.MilliSatoshi,
tlv.Blob, error)
}
// getLinkQuery is the function signature used to lookup a link.

20
routing/payment_lifecycle.go
View File

@ -679,23 +679,13 @@ func (p *paymentLifecycle) sendAttempt(
CustomRecords: lnwire.CustomRecords(p.firstHopTLVs),
}
// If we had custom records in the HTLC, then we'll encode that here
// now. We allow the traffic shaper (if there is one) to overwrite the
// custom records below. But if there is no traffic shaper, we still
// want to forward these custom records.
encodedRecords, err := htlcAdd.CustomRecords.Serialize()
if err != nil {
return nil, fmt.Errorf("unable to encode first hop TLVs: %w",
err)
}
// If a hook exists that may affect our outgoing message, we call it now
// and apply its side effects to the UpdateAddHTLC message.
err = fn.MapOptionZ(
err := fn.MapOptionZ(
p.router.cfg.TrafficShaper,
func(ts TlvTrafficShaper) error {
newAmt, newData, err := ts.ProduceHtlcExtraData(
rt.TotalAmount, encodedRecords,
rt.TotalAmount, htlcAdd.CustomRecords,
)
if err != nil {
return err
@ -706,8 +696,12 @@ func (p *paymentLifecycle) sendAttempt(
return err
}
log.Debugf("TLV traffic shaper returned custom "+
"records %v and amount %d msat for HTLC",
spew.Sdump(customRecords), newAmt)
htlcAdd.CustomRecords = customRecords
htlcAdd.Amount = lnwire.NewMSatFromSatoshis(newAmt)
htlcAdd.Amount = newAmt
return nil
},