From 3eb422afd0771e615898bdf5ae66e1c58edce3ed Mon Sep 17 00:00:00 2001 From: Matt Joiner Date: Thu, 31 Aug 2017 14:30:53 +1000 Subject: [PATCH] Add peer_protocol.Message.MustMarshalBinary --- peer_protocol/protocol.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/peer_protocol/protocol.go b/peer_protocol/protocol.go index f1f392f9..19db34fb 100644 --- a/peer_protocol/protocol.go +++ b/peer_protocol/protocol.go @@ -68,6 +68,14 @@ type Message struct { Port uint16 } +func (msg Message) MustMarshalBinary() []byte { + b, err := msg.MarshalBinary() + if err != nil { + panic(err) + } + return b +} + func (msg Message) MarshalBinary() (data []byte, err error) { buf := &bytes.Buffer{} if !msg.Keepalive { -- 2.48.1