seaweedfs/weed/mq/segment/message_serde_test.go

57 lines
1.5 KiB
Go
Raw Normal View History

2022-07-18 07:25:14 +00:00
package segment
import (
flatbuffers "github.com/google/flatbuffers/go"
"github.com/seaweedfs/seaweedfs/weed/pb/message_fbs"
2022-07-18 07:25:14 +00:00
"github.com/stretchr/testify/assert"
"testing"
)
func TestMessageSerde(t *testing.T) {
b := flatbuffers.NewBuilder(1024)
2022-07-22 08:12:32 +00:00
prop := make(map[string][]byte)
prop["n1"] = []byte("v1")
prop["n2"] = []byte("v2")
2022-07-18 07:25:14 +00:00
2022-07-22 08:12:32 +00:00
bb := NewMessageBatchBuilder(b, 1, 2, 3, 4)
2022-07-18 07:25:14 +00:00
2022-07-22 08:12:32 +00:00
bb.AddMessage(5, 6, prop, []byte("the primary key"), []byte("body is here"))
bb.BuildMessageBatch()
buf := bb.GetBytes()
2022-07-18 07:25:14 +00:00
println("serialized size", len(buf))
2022-07-22 08:12:32 +00:00
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(0), mb.TsMsMaxDelta())
assert.Equal(t, 1, mb.MessagesLength())
2022-07-18 07:25:14 +00:00
2022-07-22 08:12:32 +00:00
m := &message_fbs.Message{}
mb.Messages(m, 0)
2022-07-18 07:25:14 +00:00
nv := &message_fbs.NameValue{}
m.Properties(nv, 0)
assert.Equal(t, "n1", string(nv.Name()))
assert.Equal(t, "v1", string(nv.Value()))
m.Properties(nv, 1)
assert.Equal(t, "n2", string(nv.Name()))
assert.Equal(t, "v2", string(nv.Value()))
assert.Equal(t, []byte("the primary key"), m.Key())
assert.Equal(t, []byte("body is here"), m.Data())
2022-07-22 08:12:32 +00:00
assert.Equal(t, int32(0), m.SeqDelta())
assert.Equal(t, int32(0), m.TsMsDelta())
2022-07-18 07:25:14 +00:00
}