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
@@ -73,6 +73,10 @@ type Custom struct {
// interface.
var _ Message = (*Custom)(nil)
// A compile time check to ensure Custom implements the lnwire.SizeableMessage
// interface.
var _ SizeableMessage = (*Custom)(nil)
// NewCustom instantiates a new custom message.
func NewCustom(msgType MessageType, data []byte) (*Custom, error) {
if msgType < CustomTypeStart && !IsCustomOverride(msgType) {
@@ -117,3 +121,10 @@ func (c *Custom) Decode(r io.Reader, pver uint32) error {
func (c *Custom) MsgType() MessageType {
return c.Type
}
// SerializedSize returns the serialized size of the message in bytes.
//
// This is part of the lnwire.SizeableMessage interface.
func (c *Custom) SerializedSize() (uint32, error) {
return MessageSerializedSize(c)
}