[#45] api-go: Add PutSingle response wrappers

Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
This commit is contained in:
Dmitrii Stepanov 2023-06-30 16:05:37 +03:00
parent 2cb57a8835
commit d5b55d0926
7 changed files with 124 additions and 0 deletions

View file

@ -2114,3 +2114,58 @@ func (r *PutSingleRequest) FromGRPCMessage(m grpc.Message) error {
return r.RequestHeaders.FromMessage(v) return r.RequestHeaders.FromMessage(v)
} }
func (r *PutSingleResponseBody) ToGRPCMessage() grpc.Message {
var b *object.PutSingleResponse_Body
if r != nil {
b = new(object.PutSingleResponse_Body)
}
return b
}
func (r *PutSingleResponseBody) FromGRPCMessage(m grpc.Message) error {
v, ok := m.(*object.PutSingleResponse_Body)
if !ok {
return message.NewUnexpectedMessageType(m, v)
}
return nil
}
func (r *PutSingleResponse) ToGRPCMessage() grpc.Message {
var m *object.PutSingleResponse
if r != nil {
m = new(object.PutSingleResponse)
m.SetBody(r.body.ToGRPCMessage().(*object.PutSingleResponse_Body))
r.ResponseHeaders.ToMessage(m)
}
return m
}
func (r *PutSingleResponse) FromGRPCMessage(m grpc.Message) error {
v, ok := m.(*object.PutSingleResponse)
if !ok {
return message.NewUnexpectedMessageType(m, v)
}
var err error
body := v.GetBody()
if body == nil {
r.body = nil
} else {
if r.body == nil {
r.body = new(PutSingleResponseBody)
}
err = r.body.FromGRPCMessage(body)
if err != nil {
return err
}
}
return r.ResponseHeaders.FromMessage(v)
}

View file

@ -520,3 +520,18 @@ func (m *PutSingleRequest) SetMetaHeader(v *session.RequestMetaHeader) {
func (m *PutSingleRequest) SetVerifyHeader(v *session.RequestVerificationHeader) { func (m *PutSingleRequest) SetVerifyHeader(v *session.RequestVerificationHeader) {
m.VerifyHeader = v m.VerifyHeader = v
} }
// SetBody sets body of the response.
func (m *PutSingleResponse) SetBody(b *PutSingleResponse_Body) {
m.Body = b
}
// SetMetaHeader sets meta header of the response.
func (m *PutSingleResponse) SetMetaHeader(v *session.ResponseMetaHeader) {
m.MetaHeader = v
}
// SetVerifyHeader sets verification header of the response.
func (m *PutSingleResponse) SetVerifyHeader(v *session.ResponseVerificationHeader) {
m.VerifyHeader = v
}

View file

@ -1098,3 +1098,23 @@ func (r *PutSingleRequestBody) StableSize() int {
func (r *PutSingleRequestBody) Unmarshal(data []byte) error { func (r *PutSingleRequestBody) Unmarshal(data []byte) error {
return message.Unmarshal(r, data, new(object.PutSingleRequest_Body)) return message.Unmarshal(r, data, new(object.PutSingleRequest_Body))
} }
func (r *PutSingleResponseBody) StableMarshal(buf []byte) []byte {
if r == nil {
return []byte{}
}
if buf == nil {
buf = make([]byte, r.StableSize())
}
return buf
}
func (r *PutSingleResponseBody) StableSize() int {
return 0
}
func (r *PutSingleResponseBody) Unmarshal(data []byte) error {
return message.Unmarshal(r, data, new(object.PutSingleResponse_Body))
}

View file

@ -52,5 +52,6 @@ func TestMessageConvert(t *testing.T) {
func(empty bool) message.Message { return objecttest.GenerateGetRangeHashResponse(empty) }, func(empty bool) message.Message { return objecttest.GenerateGetRangeHashResponse(empty) },
func(empty bool) message.Message { return objecttest.GenerateLock(empty) }, func(empty bool) message.Message { return objecttest.GenerateLock(empty) },
func(empty bool) message.Message { return objecttest.GeneratePutSingleRequest(empty) }, func(empty bool) message.Message { return objecttest.GeneratePutSingleRequest(empty) },
func(empty bool) message.Message { return objecttest.GeneratePutSingleResponse(empty) },
) )
} }

View file

@ -632,3 +632,13 @@ func GeneratePutSingleRequestBody(empty bool) *object.PutSingleRequestBody {
} }
return b return b
} }
func GeneratePutSingleResponse(empty bool) *object.PutSingleResponse {
m := new(object.PutSingleResponse)
if !empty {
m.SetBody(new(object.PutSingleResponseBody))
}
m.SetMetaHeader(sessiontest.GenerateResponseMetaHeader(empty))
m.SetVerificationHeader(sessiontest.GenerateResponseVerificationHeader(empty))
return m
}

View file

@ -312,6 +312,15 @@ type PutSingleRequest struct {
session.RequestHeaders session.RequestHeaders
} }
type PutSingleResponseBody struct {
}
type PutSingleResponse struct {
body *PutSingleResponseBody
session.ResponseHeaders
}
const ( const (
TypeRegular Type = iota TypeRegular Type = iota
TypeTombstone TypeTombstone
@ -1450,3 +1459,15 @@ func (b *PutSingleRequestBody) GetCopiesNumber() []uint32 {
func (b *PutSingleRequestBody) SetCopiesNumber(v []uint32) { func (b *PutSingleRequestBody) SetCopiesNumber(v []uint32) {
b.copyNum = v b.copyNum = v
} }
func (r *PutSingleResponse) GetBody() *PutSingleResponseBody {
if r != nil {
return r.body
}
return nil
}
func (r *PutSingleResponse) SetBody(v *PutSingleResponseBody) {
r.body = v
}

View file

@ -89,6 +89,8 @@ func serviceMessageBody(req any) stableMarshaler {
return v.GetBody() return v.GetBody()
case *object.PutSingleRequest: case *object.PutSingleRequest:
return v.GetBody() return v.GetBody()
case *object.PutSingleResponse:
return v.GetBody()
/* Netmap */ /* Netmap */
case *netmap.LocalNodeInfoRequest: case *netmap.LocalNodeInfoRequest: