From caedef82afa36e33ba650b831806a0d32a030933 Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Wed, 23 Sep 2020 16:28:42 +0300 Subject: [PATCH] [#43] cmd/neofs-node: Change object executor mocked calls Signed-off-by: Leonard Lyubich --- cmd/neofs-node/object.go | 129 ++++----------------------------------- 1 file changed, 11 insertions(+), 118 deletions(-) diff --git a/cmd/neofs-node/object.go b/cmd/neofs-node/object.go index bfb44073..4d943174 100644 --- a/cmd/neofs-node/object.go +++ b/cmd/neofs-node/object.go @@ -2,150 +2,43 @@ package main import ( "context" - "fmt" - "io" + "errors" "github.com/nspcc-dev/neofs-api-go/v2/object" objectGRPC "github.com/nspcc-dev/neofs-api-go/v2/object/grpc" - "github.com/nspcc-dev/neofs-api-go/v2/refs" "github.com/nspcc-dev/neofs-api-go/v2/session" objectTransportGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/object/grpc" objectService "github.com/nspcc-dev/neofs-node/pkg/services/object" ) -type simpleSearchBodyStreamer struct { - count int -} - -type simpleGetBodyStreamer struct { - count int -} - -type simplePutBodyStreamer struct{} - -type simpleRangeBodyStreamer struct { - count int -} - type objectExecutor struct{} -func (s *simpleGetBodyStreamer) Recv() (*object.GetResponseBody, error) { - body := new(object.GetResponseBody) - - id := new(refs.ObjectID) - id.SetValue([]byte{1, 2, 3}) - - if s.count == 0 { - in := new(object.GetObjectPartInit) - in.SetObjectID(id) - - body.SetObjectPart(in) - } else if s.count == 1 { - c := new(object.GetObjectPartChunk) - c.SetChunk([]byte{8, 8, 0, 0, 5, 5, 5, 3, 5, 3, 5}) - - body.SetObjectPart(c) - } else { - return nil, io.EOF - } - - s.count++ - - return body, nil -} - func (*objectExecutor) Get(context.Context, *object.GetRequestBody) (objectService.GetObjectBodyStreamer, error) { - return new(simpleGetBodyStreamer), nil -} - -func (s *simplePutBodyStreamer) Send(body *object.PutRequestBody) error { - fmt.Printf("object part received %T\n", body.GetObjectPart()) - return nil -} - -func (s *simplePutBodyStreamer) CloseAndRecv() (*object.PutResponseBody, error) { - body := new(object.PutResponseBody) - oid := new(refs.ObjectID) - - body.SetObjectID(oid) - - oid.SetValue([]byte{6, 7, 8}) - - return body, nil + return nil, errors.New("unimplemented service call") } func (*objectExecutor) Put(context.Context) (objectService.PutObjectBodyStreamer, error) { - return new(simplePutBodyStreamer), nil + return nil, errors.New("unimplemented service call") } -func (*objectExecutor) Head(_ context.Context, body *object.HeadRequestBody) (*object.HeadResponseBody, error) { - res := new(object.HeadResponseBody) - - hdrPart := new(object.GetHeaderPartShort) - shHdr := new(object.ShortHeader) - hdrPart.SetShortHeader(shHdr) - - shHdr.SetPayloadLength(100) - - res.SetHeaderPart(hdrPart) - - return res, nil +func (*objectExecutor) Head(context.Context, *object.HeadRequestBody) (*object.HeadResponseBody, error) { + return nil, errors.New("unimplemented service call") } -func (s *objectExecutor) Search(ctx context.Context, body *object.SearchRequestBody) (objectService.SearchObjectBodyStreamer, error) { - return new(simpleSearchBodyStreamer), nil +func (s *objectExecutor) Search(context.Context, *object.SearchRequestBody) (objectService.SearchObjectBodyStreamer, error) { + return nil, errors.New("unimplemented service call") } func (*objectExecutor) Delete(_ context.Context, body *object.DeleteRequestBody) (*object.DeleteResponseBody, error) { - return new(object.DeleteResponseBody), nil -} - -func (s *simpleRangeBodyStreamer) Recv() (*object.GetRangeResponseBody, error) { - body := new(object.GetRangeResponseBody) - - if s.count == 0 { - body.SetChunk([]byte{1, 2, 2, 1}) - } else if s.count == 1 { - body.SetChunk([]byte{4, 2, 4, 2}) - } else { - return nil, io.EOF - } - - s.count++ - - return body, nil + return nil, errors.New("unimplemented service call") } func (*objectExecutor) GetRange(_ context.Context, body *object.GetRangeRequestBody) (objectService.GetRangeObjectBodyStreamer, error) { - return new(simpleRangeBodyStreamer), nil + return nil, errors.New("unimplemented service call") } -func (*objectExecutor) GetRangeHash(_ context.Context, body *object.GetRangeHashRequestBody) (*object.GetRangeHashResponseBody, error) { - fmt.Println(body.GetRanges()[0]) - - res := new(object.GetRangeHashResponseBody) - res.SetHashList([][]byte{{1, 2, 3}, {4, 5, 6}}) - - return res, nil -} - -func (s *simpleSearchBodyStreamer) Recv() (*object.SearchResponseBody, error) { - body := new(object.SearchResponseBody) - - id := new(refs.ObjectID) - body.SetIDList([]*refs.ObjectID{id}) - - if s.count == 0 { - id.SetValue([]byte{1}) - } else if s.count == 1 { - id.SetValue([]byte{2}) - } else { - return nil, io.EOF - } - - s.count++ - - return body, nil +func (*objectExecutor) GetRangeHash(context.Context, *object.GetRangeHashRequestBody) (*object.GetRangeHashResponseBody, error) { + return nil, errors.New("unimplemented service call") } func initObjectService(c *cfg) {