@@ -40,6 +40,7 @@ import (
4040 "github.com/lightningnetwork/lnd/lnwire"
4141 "github.com/lightningnetwork/lnd/macaroons"
4242 "github.com/lightningnetwork/lnd/msgmux"
43+ "github.com/lightningnetwork/lnd/routing/route"
4344 "github.com/lightningnetwork/lnd/sweep"
4445 "github.com/lightningnetwork/lnd/tlv"
4546 "google.golang.org/grpc"
@@ -1045,13 +1046,14 @@ func (s *Server) ShouldHandleTraffic(cid lnwire.ShortChannelID,
10451046// NOTE: This method is part of the routing.TlvTrafficShaper interface.
10461047func (s * Server ) PaymentBandwidth (fundingBlob , htlcBlob ,
10471048 commitmentBlob lfn.Option [tlv.Blob ], linkBandwidth ,
1048- htlcAmt lnwire.MilliSatoshi ,
1049- htlcView lnwallet. AuxHtlcView ) (lnwire.MilliSatoshi , error ) {
1049+ htlcAmt lnwire.MilliSatoshi , htlcView lnwallet. AuxHtlcView ,
1050+ peer route. Vertex ) (lnwire.MilliSatoshi , error ) {
10501051
10511052 srvrLog .Debugf ("PaymentBandwidth called, fundingBlob=%v, htlcBlob=%v, " +
1052- "commitmentBlob=%v" , lnutils .SpewLogClosure (fundingBlob ),
1053+ "commitmentBlob=%v, peer=%x" ,
1054+ lnutils .SpewLogClosure (fundingBlob ),
10531055 lnutils .SpewLogClosure (htlcBlob ),
1054- lnutils .SpewLogClosure (commitmentBlob ))
1056+ lnutils .SpewLogClosure (commitmentBlob ), peer [:] )
10551057
10561058 if err := s .waitForReady (); err != nil {
10571059 return 0 , err
@@ -1069,12 +1071,12 @@ func (s *Server) PaymentBandwidth(fundingBlob, htlcBlob,
10691071//
10701072// NOTE: This method is part of the routing.TlvTrafficShaper interface.
10711073func (s * Server ) ProduceHtlcExtraData (totalAmount lnwire.MilliSatoshi ,
1072- htlcCustomRecords lnwire.CustomRecords ) (lnwire. MilliSatoshi ,
1073- lnwire.CustomRecords , error ) {
1074+ htlcCustomRecords lnwire.CustomRecords ,
1075+ peer route. Vertex ) (lnwire. MilliSatoshi , lnwire.CustomRecords , error ) {
10741076
10751077 srvrLog .Debugf ("ProduceHtlcExtraData called, totalAmount=%d, " +
1076- "htlcBlob=%v" , totalAmount ,
1077- lnutils .SpewLogClosure (htlcCustomRecords ))
1078+ "htlcBlob=%v, peer=%x " , totalAmount ,
1079+ lnutils .SpewLogClosure (htlcCustomRecords ), peer [:] )
10781080
10791081 if err := s .waitForReady (); err != nil {
10801082 return 0 , nil , err
0 commit comments