link程序 ， 复杂包头， 如何进行拆包？ about link HOT 9 CLOSED

package link

import (
    "github.com/funny/binary"
    "io"
)

type YarMessage struct {
    Id       uint32
    Version  uint16
    MagicNum uint32
    Provider [32]byte
    Token    [32]byte
    Message  interface{}
}

type YarCodecType struct {
    CodecType CodecType
}

func Yar(codecType CodecType) YarCodecType {
    return YarCodecType{codecType}
}

func (codecType YarCodecType) NewCodec(r io.Reader, w io.Writer) Codec {
    br := binary.NewReader(r)
    bw := binary.NewWriter(w)
    pr := binary.NewPacketReader(binary.SplitByUint32BE, br)
    pw := binary.NewPacketWriter(binary.SplitByUint32BE, bw)
    return &YarCodec{
        codec:   codecType.CodecType.NewCodec(pr, pw),
        reader:  br,
        writer:  bw,
        pwriter: pw,
    }
}

type YarCodec struct {
    codec   Codec
    reader  *binary.Reader
    writer  *binary.Writer
    pwriter *binary.PacketWriter
}

func (codec *YarCodec) Decode(msg interface{}) error {
    ym := msg.(*YarMessage)

    ym.Id = codec.reader.ReadUint32BE()
    ym.Version = codec.reader.ReadUint16BE()
    ym.MagicNum = codec.reader.ReadUint32BE()
    _ = codec.reader.ReadUint32BE()
    codec.reader.ReadFull(ym.Provider[:])
    codec.reader.ReadFull(ym.Token[:])

    return codec.codec.Decode(ym.Message)
}

func (codec *YarCodec) Encode(msg interface{}) error {
    ym := msg.(*YarMessage)

    codec.writer.WriteUint32BE(ym.Id)
    codec.writer.WriteUint16BE(ym.Version)
    codec.writer.WriteUint32BE(ym.MagicNum)
    codec.writer.WriteUint32BE(0)
    codec.writer.Write(ym.Provider[:])
    codec.writer.Write(ym.Token[:])

    if err := codec.codec.Encode(ym.Message); err != nil {
        return err
    }
    return codec.pwriter.Flush()
}

func Test_Yar(t *testing.T) {
    SessionTest(t, Bufio(Yar(Json())), func(t *testing.T, session *Session) {
        for i := 0; i < 2000; i++ {
            msg1 := RandObject()
            msg2 := &YarMessage{Message: &msg1}
            copy(msg2.Provider[:], RandBytes(32))
            copy(msg2.Token[:], RandBytes(32))
            err := session.Send(msg2)
            unitest.NotError(t, err)

            var msg3 TestObject
            msg4 := &YarMessage{Message: &msg3}
            err = session.Receive(msg4)
            unitest.NotError(t, err)
            unitest.Pass(t, msg2.Provider == msg4.Provider)
            unitest.Pass(t, msg2.Token == msg4.Token)
            unitest.Pass(t, msg1 == msg3)
        }
    })
}