mc root source folder cleanup

mc/proto/replay/writer.go

@@ -0,0 +1,51 @@
+package replay
+
+import (
+	"bytes"
+	"encoding/binary"
+	"io"
+
+	proto2 "github.com/admin-else/strom/mc/proto"
+	"github.com/admin-else/strom/mc/proto_base"
+)
+
+type Writer struct {
+	w io.Writer
+}
+
+func NewWriter(w io.Writer) *Writer {
+	return &Writer{w: w}
+}
+
+func (w *Writer) WriteRaw(p []byte, timestamp uint32) (err error) {
+	err = binary.Write(w.w, binary.BigEndian, timestamp)
+	if err != nil {
+		return
+	}
+	err = binary.Write(w.w, binary.BigEndian, uint32(len(p)))
+	if err != nil {
+		return
+	}
+	_, err = w.w.Write(p)
+	return
+}
+
+func (w *Writer) WritePacket(packet proto_base.EncodeDecodeAble, timestamp uint32) (err error) {
+	b := bytes.NewBuffer(nil)
+
+	info, ok := proto2.LookupPacketInfoByType(packet)
+	if !ok {
+		return proto2.BadPacketIdErr
+	}
+
+	err = proto_base.EncodeVarInt(b, info.PacketId)
+	if err != nil {
+		return
+	}
+
+	err = packet.Encode(b)
+	if err != nil {
+		return
+	}
+	return w.WriteRaw(b.Bytes(), timestamp)
+}