seaweedfs/weed/mq/segment/message_serde_test.go

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)

	/*
		// the vector seems not consistent
		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())

	assert.Equal(t, int32(0), m.SeqDelta())
	assert.Equal(t, int32(0), m.TsMsDelta())

}