package object import ( "context" "github.com/nspcc-dev/neofs-api-go/v2/object" "github.com/nspcc-dev/neofs-api-go/v2/session" "github.com/pkg/errors" ) type GetObjectBodyStreamer interface { Recv() (*object.GetResponseBody, error) } type PutObjectBodyStreamer interface { Send(*object.PutRequestBody) error CloseAndRecv() (*object.PutResponseBody, error) } type SearchObjectBodyStreamer interface { Recv() (*object.SearchResponseBody, error) } type GetRangeObjectBodyStreamer interface { Recv() (*object.GetRangeResponseBody, error) } type ServiceExecutor interface { Get(context.Context, *object.GetRequestBody) (GetObjectBodyStreamer, error) Put(context.Context) (object.PutObjectStreamer, error) Head(context.Context, *object.HeadRequestBody) (*object.HeadResponseBody, error) Search(context.Context, *object.SearchRequestBody) (SearchObjectBodyStreamer, error) Delete(context.Context, *object.DeleteRequestBody) (*object.DeleteResponseBody, error) GetRange(context.Context, *object.GetRangeRequestBody) (GetRangeObjectBodyStreamer, error) GetRangeHash(context.Context, *object.GetRangeHashRequestBody) (*object.GetRangeHashResponseBody, error) } type executorSvc struct { exec ServiceExecutor metaHeader *session.ResponseMetaHeader } type searchStreamer struct { bodyStreamer SearchObjectBodyStreamer metaHdr *session.ResponseMetaHeader } // NewExecutionService wraps ServiceExecutor and returns Object Service interface. // // Passed meta header is attached to all responses. func NewExecutionService(exec ServiceExecutor, metaHdr *session.ResponseMetaHeader) object.Service { return &executorSvc{ exec: exec, metaHeader: metaHdr, } } func (s *executorSvc) Get(context.Context, *object.GetRequest) (object.GetObjectStreamer, error) { panic("implement me") } func (*executorSvc) Put(context.Context) (object.PutObjectStreamer, error) { panic("implement me") } func (*executorSvc) Head(context.Context, *object.HeadRequest) (*object.HeadResponse, error) { panic("implement me") } func (s *searchStreamer) Recv() (*object.SearchResponse, error) { body, err := s.bodyStreamer.Recv() if err != nil { return nil, errors.Wrap(err, "could not receive response body") } resp := new(object.SearchResponse) resp.SetBody(body) resp.SetMetaHeader(s.metaHdr) return resp, nil } func (s *executorSvc) Search(ctx context.Context, req *object.SearchRequest) (object.SearchObjectStreamer, error) { bodyStream, err := s.exec.Search(ctx, req.GetBody()) if err != nil { return nil, errors.Wrap(err, "could not execute Balance request") } return &searchStreamer{ bodyStreamer: bodyStream, metaHdr: s.metaHeader, }, nil } func (*executorSvc) Delete(context.Context, *object.DeleteRequest) (*object.DeleteResponse, error) { panic("implement me") } func (*executorSvc) GetRange(context.Context, *object.GetRangeRequest) (object.GetRangeObjectStreamer, error) { panic("implement me") } func (*executorSvc) GetRangeHash(context.Context, *object.GetRangeHashRequest) (*object.GetRangeHashResponse, error) { panic("implement me") }