Merge pull request #9623 from Roasbeef/size-msg-test-msg

Size msg test msg

Author: Roasbeef

.gitignore

@@ -80,3 +80,5 @@ coverage.txt
 # Release build directory (to avoid build.vcs.modified Golang build tag to be
 # set to true by having untracked files in the working directory).
 /lnd-*/
+
+.aider*

lnwire/accept_channel.go

@@ -128,6 +128,10 @@ type AcceptChannel struct {
 // interface.
 var _ Message = (*AcceptChannel)(nil)
 
+// A compile time check to ensure AcceptChannel implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*AcceptChannel)(nil)
+
 // Encode serializes the target AcceptChannel into the passed io.Writer
 // implementation. Serialization will observe the rules defined by the passed
 // protocol version.
@@ -281,3 +285,10 @@ func (a *AcceptChannel) Decode(r io.Reader, pver uint32) error {
 func (a *AcceptChannel) MsgType() MessageType {
 	return MsgAcceptChannel
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (a *AcceptChannel) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(a)
+}

lnwire/announcement_signatures.go

@@ -47,6 +47,10 @@ type AnnounceSignatures1 struct {
 // lnwire.Message interface.
 var _ Message = (*AnnounceSignatures1)(nil)
 
+// A compile time check to ensure AnnounceSignatures1 implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*AnnounceSignatures1)(nil)
+
 // A compile time check to ensure AnnounceSignatures1 implements the
 // lnwire.AnnounceSignatures interface.
 var _ AnnounceSignatures = (*AnnounceSignatures1)(nil)
@@ -97,6 +101,13 @@ func (a *AnnounceSignatures1) MsgType() MessageType {
 	return MsgAnnounceSignatures
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (a *AnnounceSignatures1) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(a)
+}
+
 // SCID returns the ShortChannelID of the channel.
 //
 // This is part of the lnwire.AnnounceSignatures interface.

lnwire/announcement_signatures_2.go

@@ -41,6 +41,10 @@ type AnnounceSignatures2 struct {
 // lnwire.Message interface.
 var _ Message = (*AnnounceSignatures2)(nil)
 
+// A compile time check to ensure AnnounceSignatures2 implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*AnnounceSignatures2)(nil)
+
 // Decode deserializes a serialized AnnounceSignatures2 stored in the passed
 // io.Reader observing the specified protocol version.
 //
@@ -82,6 +86,13 @@ func (a *AnnounceSignatures2) MsgType() MessageType {
 	return MsgAnnounceSignatures2
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (a *AnnounceSignatures2) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(a)
+}
+
 // SCID returns the ShortChannelID of the channel.
 //
 // NOTE: this is part of the AnnounceSignatures interface.

lnwire/channel_announcement.go

@@ -62,6 +62,10 @@ type ChannelAnnouncement1 struct {
 // lnwire.Message interface.
 var _ Message = (*ChannelAnnouncement1)(nil)
 
+// A compile time check to ensure ChannelAnnouncement1 implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ChannelAnnouncement1)(nil)
+
 // Decode deserializes a serialized ChannelAnnouncement stored in the passed
 // io.Reader observing the specified protocol version.
 //
@@ -143,6 +147,13 @@ func (a *ChannelAnnouncement1) MsgType() MessageType {
 	return MsgChannelAnnouncement
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (a *ChannelAnnouncement1) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(a)
+}
+
 // DataToSign is used to retrieve part of the announcement message which should
 // be signed.
 func (a *ChannelAnnouncement1) DataToSign() ([]byte, error) {

lnwire/channel_announcement_2.go

@@ -194,10 +194,21 @@ func (c *ChannelAnnouncement2) MsgType() MessageType {
 	return MsgChannelAnnouncement2
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *ChannelAnnouncement2) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}
+
 // A compile time check to ensure ChannelAnnouncement2 implements the
 // lnwire.Message interface.
 var _ Message = (*ChannelAnnouncement2)(nil)
 
+// A compile time check to ensure ChannelAnnouncement2 implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ChannelAnnouncement2)(nil)
+
 // Node1KeyBytes returns the bytes representing the public key of node 1 in the
 // channel.
 //

lnwire/channel_ready.go

@@ -63,6 +63,10 @@ func NewChannelReady(cid ChannelID, npcp *btcec.PublicKey) *ChannelReady {
 // interface.
 var _ Message = (*ChannelReady)(nil)
 
+// A compile time check to ensure ChannelReady implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ChannelReady)(nil)
+
 // Decode deserializes the serialized ChannelReady message stored in the
 // passed io.Reader into the target ChannelReady using the deserialization
 // rules defined by the passed protocol version.
@@ -170,3 +174,10 @@ func (c *ChannelReady) Encode(w *bytes.Buffer, _ uint32) error {
 func (c *ChannelReady) MsgType() MessageType {
 	return MsgChannelReady
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *ChannelReady) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}

lnwire/channel_reestablish.go

@@ -99,6 +99,10 @@ type ChannelReestablish struct {
 // lnwire.Message interface.
 var _ Message = (*ChannelReestablish)(nil)
 
+// A compile time check to ensure ChannelReestablish implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ChannelReestablish)(nil)
+
 // Encode serializes the target ChannelReestablish into the passed io.Writer
 // observing the protocol version specified.
 //
@@ -234,3 +238,10 @@ func (a *ChannelReestablish) Decode(r io.Reader, pver uint32) error {
 func (a *ChannelReestablish) MsgType() MessageType {
 	return MsgChannelReestablish
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (a *ChannelReestablish) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(a)
+}

lnwire/channel_update.go

@@ -124,6 +124,10 @@ type ChannelUpdate1 struct {
 // interface.
 var _ Message = (*ChannelUpdate1)(nil)
 
+// A compile time check to ensure ChannelUpdate1 implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ChannelUpdate1)(nil)
+
 // Decode deserializes a serialized ChannelUpdate stored in the passed
 // io.Reader observing the specified protocol version.
 //
@@ -367,3 +371,10 @@ func (a *ChannelUpdate1) SetSCID(scid ShortChannelID) {
 // A compile time assertion to ensure ChannelUpdate1 implements the
 // ChannelUpdate interface.
 var _ ChannelUpdate = (*ChannelUpdate1)(nil)
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (a *ChannelUpdate1) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(a)
+}

lnwire/channel_update_2.go

@@ -241,6 +241,13 @@ func (c *ChannelUpdate2) MsgType() MessageType {
 	return MsgChannelUpdate2
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *ChannelUpdate2) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}
+
 func (c *ChannelUpdate2) ExtraData() ExtraOpaqueData {
 	return c.ExtraOpaqueData
 }

lnwire/closing_complete.go

@@ -169,6 +169,17 @@ func (c *ClosingComplete) MsgType() MessageType {
 	return MsgClosingComplete
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *ClosingComplete) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}
+
 // A compile time check to ensure ClosingComplete implements the lnwire.Message
 // interface.
 var _ Message = (*ClosingComplete)(nil)
+
+// A compile time check to ensure ClosingComplete implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ClosingComplete)(nil)

lnwire/closing_sig.go

@@ -107,6 +107,17 @@ func (c *ClosingSig) MsgType() MessageType {
 	return MsgClosingSig
 }
 
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *ClosingSig) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}
+
 // A compile time check to ensure ClosingSig implements the lnwire.Message
 // interface.
 var _ Message = (*ClosingSig)(nil)
+
+// A compile time check to ensure ClosingSig implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ClosingSig)(nil)

lnwire/closing_signed.go

@@ -59,6 +59,10 @@ func NewClosingSigned(cid ChannelID, fs btcutil.Amount,
 // interface.
 var _ Message = (*ClosingSigned)(nil)
 
+// A compile time check to ensure ClosingSigned implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*ClosingSigned)(nil)
+
 // Decode deserializes a serialized ClosingSigned message stored in the passed
 // io.Reader observing the specified protocol version.
 //
@@ -130,3 +134,10 @@ func (c *ClosingSigned) Encode(w *bytes.Buffer, pver uint32) error {
 func (c *ClosingSigned) MsgType() MessageType {
 	return MsgClosingSigned
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *ClosingSigned) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}

lnwire/commit_sig.go

@@ -64,6 +64,10 @@ func NewCommitSig() *CommitSig {
 // interface.
 var _ Message = (*CommitSig)(nil)
 
+// A compile time check to ensure CommitSig implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*CommitSig)(nil)
+
 // Decode deserializes a serialized CommitSig message stored in the
 // passed io.Reader observing the specified protocol version.
 //
@@ -151,3 +155,10 @@ func (c *CommitSig) MsgType() MessageType {
 func (c *CommitSig) TargetChanID() ChannelID {
 	return c.ChanID
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (c *CommitSig) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(c)
+}

lnwire/custom.go

@@ -69,10 +69,14 @@ type Custom struct {
 	Data []byte
 }
 
-// A compile time check to ensure FundingCreated implements the lnwire.Message
+// A compile time check to ensure Custom implements the lnwire.Message
 // 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)
+}

lnwire/dyn_ack.go

@@ -41,6 +41,10 @@ type DynAck struct {
 // interface.
 var _ Message = (*DynAck)(nil)
 
+// A compile time check to ensure DynAck implements the lnwire.SizeableMessage
+// interface.
+var _ SizeableMessage = (*DynAck)(nil)
+
 // Encode serializes the target DynAck into the passed io.Writer. Serialization
 // will observe the rules defined by the passed protocol version.
 //
@@ -136,3 +140,10 @@ func (da *DynAck) Decode(r io.Reader, _ uint32) error {
 func (da *DynAck) MsgType() MessageType {
 	return MsgDynAck
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (da *DynAck) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(da)
+}

lnwire/dyn_propose.go

@@ -105,6 +105,10 @@ type DynPropose struct {
 // interface.
 var _ Message = (*DynPropose)(nil)
 
+// A compile time check to ensure DynPropose implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*DynPropose)(nil)
+
 // Encode serializes the target DynPropose into the passed io.Writer.
 // Serialization will observe the rules defined by the passed protocol version.
 //
@@ -317,3 +321,10 @@ func (dp *DynPropose) Decode(r io.Reader, _ uint32) error {
 func (dp *DynPropose) MsgType() MessageType {
 	return MsgDynPropose
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (dp *DynPropose) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(dp)
+}

lnwire/dyn_reject.go

@@ -30,6 +30,10 @@ type DynReject struct {
 // interface.
 var _ Message = (*DynReject)(nil)
 
+// A compile time check to ensure DynReject implements the
+// lnwire.SizeableMessage interface.
+var _ SizeableMessage = (*DynReject)(nil)
+
 // Encode serializes the target DynReject into the passed io.Writer.
 // Serialization will observe the rules defined by the passed protocol version.
 //
@@ -74,3 +78,10 @@ func (dr *DynReject) Decode(r io.Reader, _ uint32) error {
 func (dr *DynReject) MsgType() MessageType {
 	return MsgDynReject
 }
+
+// SerializedSize returns the serialized size of the message in bytes.
+//
+// This is part of the lnwire.SizeableMessage interface.
+func (dr *DynReject) SerializedSize() (uint32, error) {
+	return MessageSerializedSize(dr)
+}