@@ -997,16 +997,19 @@ func (s *Server) ChannelFinalized(pid funding.PendingChanID) error {
997997//
998998// NOTE: This method is part of the routing.TlvTrafficShaper interface.
999999func (s * Server ) ShouldHandleTraffic (cid lnwire.ShortChannelID ,
1000- fundingBlob lfn.Option [tlv.Blob ]) (bool , error ) {
1000+ fundingBlob , htlcBlob lfn.Option [tlv.Blob ]) (bool , error ) {
10011001
1002- srvrLog .Debugf ("HandleTraffic called (cid=%v, fundingBlob=%x)" , cid ,
1003- fundingBlob .UnwrapOr (tlv.Blob {}))
1002+ srvrLog .Debugf ("HandleTraffic called (cid=%v, fundingBlob=%v, " +
1003+ "htlcBlob=%v)" , cid , fundingBlob .UnwrapOr (tlv.Blob {}),
1004+ htlcBlob .UnwrapOr (tlv.Blob {}))
10041005
10051006 if err := s .waitForReady (); err != nil {
10061007 return false , err
10071008 }
10081009
1009- return s .cfg .AuxTrafficShaper .ShouldHandleTraffic (cid , fundingBlob )
1010+ return s .cfg .AuxTrafficShaper .ShouldHandleTraffic (
1011+ cid , fundingBlob , htlcBlob ,
1012+ )
10101013}
10111014
10121015// PaymentBandwidth returns the available bandwidth for a custom channel decided
@@ -1016,20 +1019,22 @@ func (s *Server) ShouldHandleTraffic(cid lnwire.ShortChannelID,
10161019// called first.
10171020//
10181021// NOTE: This method is part of the routing.TlvTrafficShaper interface.
1019- func (s * Server ) PaymentBandwidth (htlcBlob , commitmentBlob lfn.Option [tlv.Blob ],
1020- linkBandwidth , htlcAmt lnwire.MilliSatoshi ,
1022+ func (s * Server ) PaymentBandwidth (fundingBlob , htlcBlob ,
1023+ commitmentBlob lfn.Option [tlv.Blob ], linkBandwidth ,
1024+ htlcAmt lnwire.MilliSatoshi ,
10211025 htlcView lnwallet.AuxHtlcView ) (lnwire.MilliSatoshi , error ) {
10221026
1023- srvrLog .Debugf ("PaymentBandwidth called, htlcBlob=%v, " +
1024- "commitmentBlob=%v" , spew .Sdump (htlcBlob ),
1025- spew .Sdump (commitmentBlob ))
1027+ srvrLog .Debugf ("PaymentBandwidth called, fundingBlob=%v, htlcBlob=%v, " +
1028+ "commitmentBlob=%v" , spew .Sdump (fundingBlob ),
1029+ spew .Sdump (htlcBlob ), spew . Sdump ( commitmentBlob ))
10261030
10271031 if err := s .waitForReady (); err != nil {
10281032 return 0 , err
10291033 }
10301034
10311035 return s .cfg .AuxTrafficShaper .PaymentBandwidth (
1032- htlcBlob , commitmentBlob , linkBandwidth , htlcAmt , htlcView ,
1036+ fundingBlob , htlcBlob , commitmentBlob , linkBandwidth , htlcAmt ,
1037+ htlcView ,
10331038 )
10341039}
10351040
0 commit comments