mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2025-09-19 12:17:59 +08:00
add flatbuffer serde for message
This commit is contained in:
47
weed/mq/segment/message_serde_test.go
Normal file
47
weed/mq/segment/message_serde_test.go
Normal file
@@ -0,0 +1,47 @@
|
||||
package segment
|
||||
|
||||
import (
|
||||
"github.com/chrislusf/seaweedfs/weed/pb/message_fbs"
|
||||
flatbuffers "github.com/google/flatbuffers/go"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestMessageSerde(t *testing.T) {
|
||||
b := flatbuffers.NewBuilder(1024)
|
||||
|
||||
prop := make(map[string]string)
|
||||
prop["n1"] = "v1"
|
||||
prop["n2"] = "v2"
|
||||
|
||||
CreateMessage(b, 1, 2, 3, 4, 5, 6, prop,
|
||||
[]byte("the primary key"), []byte("body is here"))
|
||||
|
||||
buf := b.FinishedBytes()
|
||||
|
||||
println("serialized size", len(buf))
|
||||
|
||||
m := message_fbs.GetRootAsMessage(buf, 0)
|
||||
|
||||
assert.Equal(t, int32(1), m.ProducerId())
|
||||
assert.Equal(t, int64(2), m.ProducerSeq())
|
||||
assert.Equal(t, int32(3), m.SegmentId())
|
||||
assert.Equal(t, int64(4), m.SegmentSeq())
|
||||
assert.Equal(t, int64(5), m.EventTsNs())
|
||||
assert.Equal(t, int64(6), m.RecvTsNs())
|
||||
|
||||
assert.Equal(t, 2, m.PropertiesLength())
|
||||
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())
|
||||
|
||||
m.MutateSegmentSeq(123)
|
||||
assert.Equal(t, int64(123), m.SegmentSeq())
|
||||
|
||||
}
|
Reference in New Issue
Block a user