lnwire: add new SerializedSize method to all wire messages

This'll be useful for the bandwidth based rate limiting we'll implement
in the next commit.
This commit is contained in:
Olaoluwa Osuntokun committed 2025-03-20 18:27:52 -07:00
1 parent e8875e06fe
commit 56a100123b
40 files changed
+453 -4

No files matched your search

+11
View File
@@ -91,6 +91,10 @@ func NewQueryShortChanIDs(h chainhash.Hash, e QueryEncoding,
// lnwire.Message interface.
var _ Message = (*QueryShortChanIDs)(nil)
// A compile time check to ensure QueryShortChanIDs implements the lnwire.SizeableMessage
// interface.
var _ SizeableMessage = (*QueryShortChanIDs)(nil)
// Decode deserializes a serialized QueryShortChanIDs message stored in the
// passed io.Reader observing the specified protocol version.
//
@@ -427,3 +431,10 @@ func encodeShortChanIDs(w *bytes.Buffer, encodingType QueryEncoding,
func (q *QueryShortChanIDs) MsgType() MessageType {
return MsgQueryShortChanIDs
}
// SerializedSize returns the serialized size of the message in bytes.
//
// This is part of the lnwire.SizeableMessage interface.
func (q *QueryShortChanIDs) SerializedSize() (uint32, error) {
return MessageSerializedSize(q)
}