Add meta data type to record serialization

This commit is contained in:
Daniel 2019-07-19 21:30:51 +02:00
parent 3b68c8ea4c
commit 3bd7b8558f
5 changed files with 67 additions and 26 deletions

View file

@ -70,7 +70,7 @@ func (b *Base) SetMeta(meta *Meta) {
b.meta = meta b.meta = meta
} }
// Marshal marshals the object, without the database key or metadata // Marshal marshals the object, without the database key or metadata. It returns nil if the record is deleted.
func (b *Base) Marshal(self Record, format uint8) ([]byte, error) { func (b *Base) Marshal(self Record, format uint8) ([]byte, error) {
if b.Meta() == nil { if b.Meta() == nil {
return nil, errors.New("missing meta") return nil, errors.New("missing meta")
@ -96,15 +96,15 @@ func (b *Base) MarshalRecord(self Record) ([]byte, error) {
// version // version
c := container.New([]byte{1}) c := container.New([]byte{1})
// meta // meta encoding
metaSection, err := b.meta.GenCodeMarshal(nil) metaSection, err := dsd.Dump(b.meta, GenCode)
if err != nil { if err != nil {
return nil, err return nil, err
} }
c.AppendAsBlock(metaSection) c.AppendAsBlock(metaSection)
// data // data
dataSection, err := b.Marshal(self, dsd.JSON) dataSection, err := b.Marshal(self, JSON)
if err != nil { if err != nil {
return nil, err return nil, err
} }

View file

@ -11,5 +11,5 @@ const (
BYTES = dsd.BYTES // X BYTES = dsd.BYTES // X
JSON = dsd.JSON // J JSON = dsd.JSON // J
BSON = dsd.BSON // B BSON = dsd.BSON // B
GenCode = dsd.GenCode // G (reserved) GenCode = dsd.GenCode // G
) )

View file

@ -432,7 +432,7 @@ func BenchmarkMetaUnserializeWithCodegen(b *testing.B) {
func BenchmarkMetaSerializeWithDSDJSON(b *testing.B) { func BenchmarkMetaSerializeWithDSDJSON(b *testing.B) {
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
_, err := dsd.Dump(testMeta, dsd.JSON) _, err := dsd.Dump(testMeta, JSON)
if err != nil { if err != nil {
b.Errorf("failed to serialize with DSD/JSON: %s", err) b.Errorf("failed to serialize with DSD/JSON: %s", err)
return return
@ -444,7 +444,7 @@ func BenchmarkMetaSerializeWithDSDJSON(b *testing.B) {
func BenchmarkMetaUnserializeWithDSDJSON(b *testing.B) { func BenchmarkMetaUnserializeWithDSDJSON(b *testing.B) {
// Setup // Setup
encodedData, err := dsd.Dump(testMeta, dsd.JSON) encodedData, err := dsd.Dump(testMeta, JSON)
if err != nil { if err != nil {
b.Errorf("failed to serialize with DSD/JSON: %s", err) b.Errorf("failed to serialize with DSD/JSON: %s", err)
return return

View file

@ -37,9 +37,20 @@ func NewRawWrapper(database, key string, data []byte) (*Wrapper, error) {
offset += n offset += n
newMeta := &Meta{} newMeta := &Meta{}
_, err = newMeta.GenCodeUnmarshal(metaSection) if len(metaSection) == 34 && metaSection[4] == 0 {
if err != nil { // TODO: remove in 2020
return nil, fmt.Errorf("could not unmarshal meta section: %s", err) // backward compatibility:
// format would byte shift and populate metaSection[4] with value > 0 (would naturally populate >0 at 07.02.2106 07:28:15)
// this must be gencode without format
_, err = newMeta.GenCodeUnmarshal(metaSection)
if err != nil {
return nil, fmt.Errorf("could not unmarshal meta section: %s", err)
}
} else {
_, err = dsd.Load(metaSection, newMeta)
if err != nil {
return nil, fmt.Errorf("could not unmarshal meta section: %s", err)
}
} }
format, n, err := varint.Unpack8(data[offset:]) format, n, err := varint.Unpack8(data[offset:])
@ -86,7 +97,7 @@ func (w *Wrapper) Marshal(r Record, format uint8) ([]byte, error) {
return nil, nil return nil, nil
} }
if format != dsd.AUTO && format != w.Format { if format != AUTO && format != w.Format {
return nil, errors.New("could not dump model, wrapped object format mismatch") return nil, errors.New("could not dump model, wrapped object format mismatch")
} }
@ -109,14 +120,14 @@ func (w *Wrapper) MarshalRecord(r Record) ([]byte, error) {
c := container.New([]byte{1}) c := container.New([]byte{1})
// meta // meta
metaSection, err := w.meta.GenCodeMarshal(nil) metaSection, err := dsd.Dump(w.meta, GenCode)
if err != nil { if err != nil {
return nil, err return nil, err
} }
c.AppendAsBlock(metaSection) c.AppendAsBlock(metaSection)
// data // data
dataSection, err := w.Marshal(r, dsd.JSON) dataSection, err := w.Marshal(r, JSON)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -125,16 +136,6 @@ func (w *Wrapper) MarshalRecord(r Record) ([]byte, error) {
return c.CompileData(), nil return c.CompileData(), nil
} }
// // Lock locks the record.
// func (w *Wrapper) Lock() {
// w.lock.Lock()
// }
//
// // Unlock unlocks the record.
// func (w *Wrapper) Unlock() {
// w.lock.Unlock()
// }
// IsWrapped returns whether the record is a Wrapper. // IsWrapped returns whether the record is a Wrapper.
func (w *Wrapper) IsWrapped() bool { func (w *Wrapper) IsWrapped() bool {
return true return true

View file

@ -2,9 +2,10 @@ package record
import ( import (
"bytes" "bytes"
"errors"
"testing" "testing"
"github.com/safing/portbase/formats/dsd" "github.com/safing/portbase/container"
) )
func TestWrapper(t *testing.T) { func TestWrapper(t *testing.T) {
@ -24,14 +25,14 @@ func TestWrapper(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if wrapper.Format != dsd.JSON { if wrapper.Format != JSON {
t.Error("format mismatch") t.Error("format mismatch")
} }
if !bytes.Equal(testData, wrapper.Data) { if !bytes.Equal(testData, wrapper.Data) {
t.Error("data mismatch") t.Error("data mismatch")
} }
encoded, err := wrapper.Marshal(wrapper, dsd.JSON) encoded, err := wrapper.Marshal(wrapper, JSON)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -40,6 +41,7 @@ func TestWrapper(t *testing.T) {
} }
wrapper.SetMeta(&Meta{}) wrapper.SetMeta(&Meta{})
wrapper.meta.Update()
raw, err := wrapper.MarshalRecord(wrapper) raw, err := wrapper.MarshalRecord(wrapper)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
@ -53,4 +55,42 @@ func TestWrapper(t *testing.T) {
t.Error("marshal mismatch") t.Error("marshal mismatch")
} }
// test new format
oldRaw, err := oldWrapperMarshalRecord(wrapper, wrapper)
if err != nil {
t.Fatal(err)
}
wrapper3, err := NewRawWrapper("test", "a", oldRaw)
if err != nil {
t.Fatal(err)
}
if !bytes.Equal(testData, wrapper3.Data) {
t.Error("marshal mismatch")
}
}
func oldWrapperMarshalRecord(w *Wrapper, r Record) ([]byte, error) {
if w.Meta() == nil {
return nil, errors.New("missing meta")
}
// version
c := container.New([]byte{1})
// meta
metaSection, err := w.meta.GenCodeMarshal(nil)
if err != nil {
return nil, err
}
c.AppendAsBlock(metaSection)
// data
dataSection, err := w.Marshal(r, JSON)
if err != nil {
return nil, err
}
c.Append(dataSection)
return c.CompileData(), nil
} }