123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- package segment
- import (
- flatbuffers "github.com/google/flatbuffers/go"
- "github.com/seaweedfs/seaweedfs/weed/pb/message_fbs"
- "github.com/stretchr/testify/assert"
- "testing"
- )
- func TestMessageSerde(t *testing.T) {
- b := flatbuffers.NewBuilder(1024)
- prop := make(map[string][]byte)
- prop["n1"] = []byte("v1")
- prop["n2"] = []byte("v2")
- bb := NewMessageBatchBuilder(b, 1, 2, 3, 4)
- bb.AddMessage(5, 6, prop, []byte("the primary key"), []byte("body is here"))
- bb.AddMessage(5, 7, prop, []byte("the primary 2"), []byte("body is 2"))
- bb.BuildMessageBatch()
- buf := bb.GetBytes()
- println("serialized size", len(buf))
- mb := message_fbs.GetRootAsMessageBatch(buf, 0)
- assert.Equal(t, int32(1), mb.ProducerId())
- assert.Equal(t, int32(2), mb.ProducerEpoch())
- assert.Equal(t, int32(3), mb.SegmentId())
- assert.Equal(t, int32(4), mb.Flags())
- assert.Equal(t, int64(5), mb.SegmentSeqBase())
- assert.Equal(t, int32(0), mb.SegmentSeqMaxDelta())
- assert.Equal(t, int64(6), mb.TsMsBase())
- assert.Equal(t, int32(1), mb.TsMsMaxDelta())
- assert.Equal(t, 2, mb.MessagesLength())
- m := &message_fbs.Message{}
- mb.Messages(m, 0)
-
- assert.Equal(t, []byte("the primary key"), m.Key())
- assert.Equal(t, []byte("body is here"), m.Data())
- assert.Equal(t, int32(0), m.SeqDelta())
- assert.Equal(t, int32(0), m.TsMsDelta())
- }
|