From 486aeef5df85677709837628acb90615731a4ee4 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Tue, 25 Aug 2020 17:44:09 +0300
Subject: [PATCH] [#13] services/object: Define and implement Object executor

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
---
 pkg/services/object/executor.go | 107 ++++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 pkg/services/object/executor.go

diff --git a/pkg/services/object/executor.go b/pkg/services/object/executor.go
new file mode 100644
index 000000000..a6b3f354d
--- /dev/null
+++ b/pkg/services/object/executor.go
@@ -0,0 +1,107 @@
+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")
+}