[#225] v2/object: Support more SplitInfo structures

- SplitInfo structure in object.HeadResponse
- SplitInfo structure in object.GetRangeResponse
- Raw flag in object.GetRangeRequest

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin 2020-12-07 18:59:01 +03:00 committed by Alex Vanin
parent 42cd897b25
commit ded06674ac
4 changed files with 223 additions and 28 deletions

View file

@ -1040,6 +1040,10 @@ func HeadResponseBodyToGRPCMessage(r *HeadResponseBody) *object.HeadResponse_Bod
m.SetShortHeader( m.SetShortHeader(
ShortHeaderToGRPCMessage(t), ShortHeaderToGRPCMessage(t),
) )
case *SplitInfo:
m.SetSplitInfo(
SplitInfoToGRPCMessage(t),
)
default: default:
panic(fmt.Sprintf("unknown header part %T", t)) panic(fmt.Sprintf("unknown header part %T", t))
} }
@ -1064,6 +1068,10 @@ func HeadResponseBodyFromGRPCMessage(m *object.HeadResponse_Body) *HeadResponseB
r.SetHeaderPart( r.SetHeaderPart(
ShortHeaderFromGRPCMessage(v.ShortHeader), ShortHeaderFromGRPCMessage(v.ShortHeader),
) )
case *object.HeadResponse_Body_SplitInfo:
r.SetHeaderPart(
SplitInfoFromGRPCMessage(v.SplitInfo),
)
default: default:
panic(fmt.Sprintf("unknown header part %T", v)) panic(fmt.Sprintf("unknown header part %T", v))
} }
@ -1332,6 +1340,8 @@ func GetRangeRequestBodyToGRPCMessage(r *GetRangeRequestBody) *object.GetRangeRe
RangeToGRPCMessage(r.GetRange()), RangeToGRPCMessage(r.GetRange()),
) )
m.SetRaw(r.GetRaw())
return m return m
} }
@ -1350,6 +1360,8 @@ func GetRangeRequestBodyFromGRPCMessage(m *object.GetRangeRequest_Body) *GetRang
RangeFromGRPCMessage(m.GetRange()), RangeFromGRPCMessage(m.GetRange()),
) )
r.SetRaw(m.GetRaw())
return r return r
} }
@ -1385,6 +1397,30 @@ func GetRangeRequestFromGRPCMessage(m *object.GetRangeRequest) *GetRangeRequest
return r return r
} }
func GetRangePartChunkToGRPCMessage(r *GetRangePartChunk) *object.GetRangeResponse_Body_Chunk {
if r == nil {
return nil
}
m := new(object.GetRangeResponse_Body_Chunk)
m.SetChunk(r.GetChunk())
return m
}
func GetRangePartChunkFromGRPCMessage(m *object.GetRangeResponse_Body_Chunk) *GetRangePartChunk {
if m == nil {
return nil
}
r := new(GetRangePartChunk)
r.SetChunk(m.GetChunk())
return r
}
func GetRangeResponseBodyToGRPCMessage(r *GetRangeResponseBody) *object.GetRangeResponse_Body { func GetRangeResponseBodyToGRPCMessage(r *GetRangeResponseBody) *object.GetRangeResponse_Body {
if r == nil { if r == nil {
return nil return nil
@ -1392,7 +1428,19 @@ func GetRangeResponseBodyToGRPCMessage(r *GetRangeResponseBody) *object.GetRange
m := new(object.GetRangeResponse_Body) m := new(object.GetRangeResponse_Body)
m.SetChunk(r.GetChunk()) switch v := r.GetRangePart(); t := v.(type) {
case nil:
case *GetRangePartChunk:
m.SetChunk(
GetRangePartChunkToGRPCMessage(t),
)
case *SplitInfo:
m.SetSplitInfo(
SplitInfoToGRPCMessage(t),
)
default:
panic(fmt.Sprintf("unknown get range part %T", t))
}
return m return m
} }
@ -1404,7 +1452,19 @@ func GetRangeResponseBodyFromGRPCMessage(m *object.GetRangeResponse_Body) *GetRa
r := new(GetRangeResponseBody) r := new(GetRangeResponseBody)
r.SetChunk(m.GetChunk()) switch v := m.GetRangePart().(type) {
case nil:
case *object.GetRangeResponse_Body_Chunk:
r.SetRangePart(
GetRangePartChunkFromGRPCMessage(v),
)
case *object.GetRangeResponse_Body_SplitInfo:
r.SetRangePart(
SplitInfoFromGRPCMessage(v.SplitInfo),
)
default:
panic(fmt.Sprintf("unknown get range part %T", v))
}
return r return r
} }

View file

@ -76,6 +76,7 @@ const (
headRespBodyHeaderField = 1 headRespBodyHeaderField = 1
headRespBodyShortHeaderField = 2 headRespBodyShortHeaderField = 2
headRespBodySplitInfoField = 3
searchFilterMatchField = 1 searchFilterMatchField = 1
searchFilterNameField = 2 searchFilterNameField = 2
@ -92,8 +93,10 @@ const (
getRangeReqBodyAddressField = 1 getRangeReqBodyAddressField = 1
getRangeReqBodyRangeField = 2 getRangeReqBodyRangeField = 2
getRangeReqBodyRawField = 3
getRangeRespChunkField = 1 getRangeRespChunkField = 1
getRangeRespSplitInfoField = 2
getRangeHashReqBodyAddressField = 1 getRangeHashReqBodyAddressField = 1
getRangeHashReqBodyRangesField = 2 getRangeHashReqBodyRangesField = 2
@ -1030,6 +1033,13 @@ func (r *HeadResponseBody) StableMarshal(buf []byte) ([]byte, error) {
return nil, err return nil, err
} }
} }
case *SplitInfo:
if v != nil {
_, err := proto.NestedStructureMarshal(headRespBodySplitInfoField, buf, v)
if err != nil {
return nil, err
}
}
default: default:
panic("unknown one of object put request body type") panic("unknown one of object put request body type")
} }
@ -1053,6 +1063,10 @@ func (r *HeadResponseBody) StableSize() (size int) {
if v != nil { if v != nil {
size += proto.NestedStructureSize(headRespBodyShortHeaderField, v) size += proto.NestedStructureSize(headRespBodyShortHeaderField, v)
} }
case *SplitInfo:
if v != nil {
size += proto.NestedStructureSize(headRespBodySplitInfoField, v)
}
default: default:
panic("unknown one of object put request body type") panic("unknown one of object put request body type")
} }
@ -1263,7 +1277,14 @@ func (r *GetRangeRequestBody) StableMarshal(buf []byte) ([]byte, error) {
offset += n offset += n
_, err = proto.NestedStructureMarshal(getRangeReqBodyRangeField, buf[offset:], r.rng) n, err = proto.NestedStructureMarshal(getRangeReqBodyRangeField, buf[offset:], r.rng)
if err != nil {
return nil, err
}
offset += n
_, err = proto.BoolMarshal(getRangeReqBodyRawField, buf[offset:], r.raw)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -1278,6 +1299,7 @@ func (r *GetRangeRequestBody) StableSize() (size int) {
size += proto.NestedStructureSize(getRangeReqBodyAddressField, r.addr) size += proto.NestedStructureSize(getRangeReqBodyAddressField, r.addr)
size += proto.NestedStructureSize(getRangeReqBodyRangeField, r.rng) size += proto.NestedStructureSize(getRangeReqBodyRangeField, r.rng)
size += proto.BoolSize(getRangeReqBodyRawField, r.raw)
return size return size
} }
@ -1291,10 +1313,26 @@ func (r *GetRangeResponseBody) StableMarshal(buf []byte) ([]byte, error) {
buf = make([]byte, r.StableSize()) buf = make([]byte, r.StableSize())
} }
_, err := proto.BytesMarshal(getRangeRespChunkField, buf, r.chunk) if r.rngPart != nil {
switch v := r.rngPart.(type) {
case *GetRangePartChunk:
if v != nil {
_, err := proto.BytesMarshal(getRangeRespChunkField, buf, v.chunk)
if err != nil { if err != nil {
return nil, err return nil, err
} }
}
case *SplitInfo:
if v != nil {
_, err := proto.NestedStructureMarshal(getRangeRespSplitInfoField, buf, v)
if err != nil {
return nil, err
}
}
default:
panic("unknown one of object get range request body type")
}
}
return buf, nil return buf, nil
} }
@ -1304,7 +1342,20 @@ func (r *GetRangeResponseBody) StableSize() (size int) {
return 0 return 0
} }
size += proto.BytesSize(getRangeRespChunkField, r.chunk) if r.rngPart != nil {
switch v := r.rngPart.(type) {
case *GetRangePartChunk:
if v != nil {
size += proto.BytesSize(getRangeRespChunkField, v.chunk)
}
case *SplitInfo:
if v != nil {
size = proto.NestedStructureSize(getRangeRespSplitInfoField, v)
}
default:
panic("unknown one of object get range request body type")
}
}
return size return size
} }

View file

@ -239,8 +239,9 @@ func TestSplitHeaderFromGRPCMessage(t *testing.T) {
} }
func TestHeadResponseBody_StableMarshal(t *testing.T) { func TestHeadResponseBody_StableMarshal(t *testing.T) {
shortFrom := generateHeadResponseBody(true) shortFrom := generateHeadResponseBody(0)
fullFrom := generateHeadResponseBody(false) fullFrom := generateHeadResponseBody(1)
splitInfoFrom := generateHeadResponseBody(2)
transport := new(grpc.HeadResponse_Body) transport := new(grpc.HeadResponse_Body)
t.Run("short header non empty", func(t *testing.T) { t.Run("short header non empty", func(t *testing.T) {
@ -264,6 +265,17 @@ func TestHeadResponseBody_StableMarshal(t *testing.T) {
to := object.HeadResponseBodyFromGRPCMessage(transport) to := object.HeadResponseBodyFromGRPCMessage(transport)
require.Equal(t, fullFrom, to) require.Equal(t, fullFrom, to)
}) })
t.Run("split info non empty", func(t *testing.T) {
wire, err := splitInfoFrom.StableMarshal(nil)
require.NoError(t, err)
err = goproto.Unmarshal(wire, transport)
require.NoError(t, err)
to := object.HeadResponseBodyFromGRPCMessage(transport)
require.Equal(t, splitInfoFrom, to)
})
} }
func TestSearchRequestBody_StableMarshal(t *testing.T) { func TestSearchRequestBody_StableMarshal(t *testing.T) {
@ -315,18 +327,30 @@ func TestGetRangeRequestBody_StableMarshal(t *testing.T) {
} }
func TestGetRangeResponseBody_StableMarshal(t *testing.T) { func TestGetRangeResponseBody_StableMarshal(t *testing.T) {
from := generateRangeResponseBody("some data") dataFrom := generateRangeResponseBody("some data", true)
splitInfoFrom := generateRangeResponseBody("some data", false)
transport := new(grpc.GetRangeResponse_Body) transport := new(grpc.GetRangeResponse_Body)
t.Run("non empty", func(t *testing.T) { t.Run("data non empty", func(t *testing.T) {
wire, err := from.StableMarshal(nil) wire, err := dataFrom.StableMarshal(nil)
require.NoError(t, err) require.NoError(t, err)
err = goproto.Unmarshal(wire, transport) err = goproto.Unmarshal(wire, transport)
require.NoError(t, err) require.NoError(t, err)
to := object.GetRangeResponseBodyFromGRPCMessage(transport) to := object.GetRangeResponseBodyFromGRPCMessage(transport)
require.Equal(t, from, to) require.Equal(t, dataFrom, to)
})
t.Run("split info non empty", func(t *testing.T) {
wire, err := splitInfoFrom.StableMarshal(nil)
require.NoError(t, err)
err = goproto.Unmarshal(wire, transport)
require.NoError(t, err)
to := object.GetRangeResponseBodyFromGRPCMessage(transport)
require.Equal(t, splitInfoFrom, to)
}) })
} }
@ -546,11 +570,7 @@ func generateGetResponseBody(i int) *object.GetResponseBody {
chunk.SetChunk([]byte("Some data chunk")) chunk.SetChunk([]byte("Some data chunk"))
part = chunk part = chunk
default: default:
splitInfo := new(object.SplitInfo) part = generateSplitInfo()
splitInfo.SetSplitID([]byte("splitID"))
splitInfo.SetLastPart(generateObjectID("Right ID"))
splitInfo.SetLink(generateObjectID("Link ID"))
part = splitInfo
} }
resp.SetObjectPart(part) resp.SetObjectPart(part)
@ -606,14 +626,17 @@ func generateHeadRequestBody(cid, oid string) *object.HeadRequestBody {
return req return req
} }
func generateHeadResponseBody(flag bool) *object.HeadResponseBody { func generateHeadResponseBody(flag int) *object.HeadResponseBody {
req := new(object.HeadResponseBody) req := new(object.HeadResponseBody)
var part object.GetHeaderPart var part object.GetHeaderPart
if flag { switch flag {
case 0:
part = generateShortHeader("short id") part = generateShortHeader("short id")
} else { case 1:
part = generateHeaderWithSignature() part = generateHeaderWithSignature()
default:
part = generateSplitInfo()
} }
req.SetHeaderPart(part) req.SetHeaderPart(part)
@ -677,13 +700,21 @@ func generateRangeRequestBody(cid, oid string) *object.GetRangeRequestBody {
req := new(object.GetRangeRequestBody) req := new(object.GetRangeRequestBody)
req.SetAddress(generateAddress(cid, oid)) req.SetAddress(generateAddress(cid, oid))
req.SetRange(generateRange(10, 20)) req.SetRange(generateRange(10, 20))
req.SetRaw(true)
return req return req
} }
func generateRangeResponseBody(data string) *object.GetRangeResponseBody { func generateRangeResponseBody(data string, flag bool) *object.GetRangeResponseBody {
resp := new(object.GetRangeResponseBody) resp := new(object.GetRangeResponseBody)
resp.SetChunk([]byte(data))
if flag {
p := new(object.GetRangePartChunk)
p.SetChunk([]byte(data))
resp.SetRangePart(p)
} else {
resp.SetRangePart(generateSplitInfo())
}
return resp return resp
} }
@ -729,3 +760,12 @@ func TestObject_StableUnmarshal(t *testing.T) {
require.Equal(t, obj, obj2) require.Equal(t, obj, obj2)
} }
func generateSplitInfo() *object.SplitInfo {
splitInfo := new(object.SplitInfo)
splitInfo.SetSplitID([]byte("splitID"))
splitInfo.SetLastPart(generateObjectID("Right ID"))
splitInfo.SetLink(generateObjectID("Link ID"))
return splitInfo
}

View file

@ -181,10 +181,20 @@ type GetRangeRequestBody struct {
addr *refs.Address addr *refs.Address
rng *Range rng *Range
raw bool
}
type GetRangePart interface {
getRangePart()
}
type GetRangePartChunk struct {
chunk []byte
} }
type GetRangeResponseBody struct { type GetRangeResponseBody struct {
chunk []byte rngPart GetRangePart
} }
type GetRangeHashRequestBody struct { type GetRangeHashRequestBody struct {
@ -680,7 +690,11 @@ func (s *SplitInfo) SetLink(v *refs.ObjectID) {
} }
} }
func (s *SplitInfo) getObjectPart() {} // implement interface to be one of response body func (s *SplitInfo) getObjectPart() {}
func (s *SplitInfo) getHeaderPart() {}
func (s *SplitInfo) getRangePart() {}
func (r *GetRequestBody) GetAddress() *refs.Address { func (r *GetRequestBody) GetAddress() *refs.Address {
if r != nil { if r != nil {
@ -1530,6 +1544,20 @@ func (r *GetRangeRequestBody) SetRange(v *Range) {
} }
} }
func (r *GetRangeRequestBody) GetRaw() bool {
if r != nil {
return r.raw
}
return false
}
func (r *GetRangeRequestBody) SetRaw(v bool) {
if r != nil {
r.raw = v
}
}
func (r *GetRangeRequest) GetBody() *GetRangeRequestBody { func (r *GetRangeRequest) GetBody() *GetRangeRequestBody {
if r != nil { if r != nil {
return r.body return r.body
@ -1572,7 +1600,7 @@ func (r *GetRangeRequest) SetVerificationHeader(v *session.RequestVerificationHe
} }
} }
func (r *GetRangeResponseBody) GetChunk() []byte { func (r *GetRangePartChunk) GetChunk() []byte {
if r != nil { if r != nil {
return r.chunk return r.chunk
} }
@ -1580,12 +1608,28 @@ func (r *GetRangeResponseBody) GetChunk() []byte {
return nil return nil
} }
func (r *GetRangeResponseBody) SetChunk(v []byte) { func (r *GetRangePartChunk) SetChunk(v []byte) {
if r != nil { if r != nil {
r.chunk = v r.chunk = v
} }
} }
func (r *GetRangePartChunk) getRangePart() {}
func (r *GetRangeResponseBody) GetRangePart() GetRangePart {
if r != nil {
return r.rngPart
}
return nil
}
func (r *GetRangeResponseBody) SetRangePart(v GetRangePart) {
if r != nil {
r.rngPart = v
}
}
func (r *GetRangeResponse) GetBody() *GetRangeResponseBody { func (r *GetRangeResponse) GetBody() *GetRangeResponseBody {
if r != nil { if r != nil {
return r.body return r.body