Reuse raw CBOR from message if present

Fixes #47
blinklabs-io · Mar 20, 2022 · a631882 · a631882
1 parent 1670124
commit a631882
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/protocol/protocol.go b/protocol/protocol.go
@@ -90,9 +90,15 @@ func (p *Protocol) SendMessage(msg Message, isResponse bool) error {
 	if err != nil {
 		return fmt.Errorf("%s: error sending message: %s", p.config.Name, err)
 	}
-	data, err := utils.CborEncode(msg)
-	if err != nil {
-		return err
+	// Get raw CBOR from message
+	data := msg.Cbor()
+	// If message has no raw CBOR, encode the message
+	if data == nil {
+		var err error
+		data, err = utils.CborEncode(msg)
+		if err != nil {
+			return err
+		}
 	}
 	segment := muxer.NewSegment(p.config.ProtocolId, data, isResponse)
 	p.sendChan <- segment