golang/codec: debug flags

progrium · Jul 21, 2021 · 475935a · 475935a
1 parent f574df1
commit 475935a
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/golang/codec/doc.go → golang/codec/codec.go b/golang/codec/doc.go → golang/codec/codec.go
@@ -1,2 +1,9 @@
 // Package codec implements encoding and decoding of qmux messages.
 package codec
+
+import "io"
+
+var (
+	DebugMessages io.Writer
+	DebugBytes    io.Writer
+)
diff --git a/golang/codec/decoder.go b/golang/codec/decoder.go
@@ -28,6 +28,10 @@ func (dec *Decoder) Decode() (Message, error) {
 		return nil, err
 	}
 
+	if DebugBytes != nil {
+		fmt.Fprintln(DebugBytes, ">>DEC", packet)
+	}
+
 	return decode(packet)
 }
 
@@ -87,6 +91,9 @@ func decode(packet []byte) (Message, error) {
 	if err := Unmarshal(packet, msg); err != nil {
 		return nil, err
 	}
+	if DebugMessages != nil {
+		fmt.Fprintln(DebugMessages, ">>DEC", msg)
+	}
 	return msg, nil
 }
 

diff --git a/golang/codec/encoder.go b/golang/codec/encoder.go
@@ -19,11 +19,19 @@ func (enc *Encoder) Encode(msg interface{}) error {
 	enc.Lock()
 	defer enc.Unlock()
 
+	if DebugMessages != nil {
+		fmt.Fprintln(DebugMessages, "<<ENC", msg)
+	}
+
 	b, err := Marshal(msg)
 	if err != nil {
 		return err
 	}
 
+	if DebugBytes != nil {
+		fmt.Fprintln(DebugBytes, "<<ENC", b)
+	}
+
 	_, err = enc.w.Write(b)
 	return err
 }