Add stable marshal for owner id
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
696e263e4d
commit
18df156e61
2 changed files with 93 additions and 0 deletions
49
refs/v2/marshal.go
Normal file
49
refs/v2/marshal.go
Normal file
|
@ -0,0 +1,49 @@
|
|||
package v2
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"math/bits"
|
||||
)
|
||||
|
||||
func (m *OwnerID) StableMarshal(buf []byte) ([]byte, error) {
|
||||
if m == nil {
|
||||
return []byte{}, nil
|
||||
}
|
||||
|
||||
if buf == nil {
|
||||
buf = make([]byte, m.StableSize())
|
||||
}
|
||||
|
||||
var (
|
||||
i, n, offset int
|
||||
)
|
||||
|
||||
// Write key field.
|
||||
|
||||
buf[i] = 0x0A // id:0x1 << 3 | wiretype:0x2
|
||||
offset = binary.PutUvarint(buf[i+1:], uint64(len(m.Value)))
|
||||
n = copy(buf[i+1+offset:], m.Value)
|
||||
i += 1 + offset + n
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func (m *OwnerID) StableSize() int {
|
||||
if m == nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
var (
|
||||
ln, size int
|
||||
)
|
||||
|
||||
ln = len(m.Value) // size of key field
|
||||
size += 1 + uvarIntSize(uint64(ln)) + ln // wiretype + size of string + string
|
||||
|
||||
return size
|
||||
}
|
||||
|
||||
// uvarIntSize returns length of varint byte sequence for uint64 value 'x'.
|
||||
func uvarIntSize(x uint64) int {
|
||||
return (bits.Len64(x|1) + 6) / 7
|
||||
}
|
44
refs/v2/marshal_test.go
Normal file
44
refs/v2/marshal_test.go
Normal file
|
@ -0,0 +1,44 @@
|
|||
package v2
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestOwnerID_StableMarshal(t *testing.T) {
|
||||
owner := make([]byte, 25)
|
||||
_, err := rand.Read(owner)
|
||||
require.NoError(t, err)
|
||||
|
||||
expectedOwner := new(OwnerID)
|
||||
expectedOwner.Value = owner
|
||||
|
||||
gotOwner := new(OwnerID)
|
||||
|
||||
t.Run("small buffer", func(t *testing.T) {
|
||||
_, err = expectedOwner.StableMarshal(make([]byte, 1))
|
||||
require.Error(t, err)
|
||||
})
|
||||
|
||||
t.Run("empty owner", func(t *testing.T) {
|
||||
data, err := new(OwnerID).StableMarshal(nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = gotOwner.Unmarshal(data)
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Len(t, gotOwner.Value, 0)
|
||||
})
|
||||
|
||||
t.Run("non empty owner", func(t *testing.T) {
|
||||
data, err := expectedOwner.StableMarshal(nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = gotOwner.Unmarshal(data)
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Equal(t, expectedOwner, gotOwner)
|
||||
})
|
||||
}
|
Loading…
Reference in a new issue