frostfs-node/cmd/neofs-node/object.go

package main

import (
	"context"
	"io"

	"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 objectExecutor struct {
	count int
}

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 (*objectExecutor) Put(context.Context) (object.PutObjectStreamer, error) {
	panic("implement me")
}

func (*objectExecutor) Head(context.Context, *object.HeadRequestBody) (*object.HeadResponseBody, error) {
	panic("implement me")
}

func (s *objectExecutor) Search(ctx context.Context, body *object.SearchRequestBody) (objectService.SearchObjectBodyStreamer, error) {
	return new(simpleSearchBodyStreamer), nil
}

func (*objectExecutor) Delete(context.Context, *object.DeleteRequestBody) (*object.DeleteResponseBody, error) {
	panic("implement me")
}

func (*objectExecutor) GetRange(context.Context, *object.GetRangeRequestBody) (objectService.GetRangeObjectBodyStreamer, error) {
	panic("implement me")
}

func (*objectExecutor) GetRangeHash(context.Context, *object.GetRangeHashRequestBody) (*object.GetRangeHashResponseBody, error) {
	panic("implement me")
}

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 initObjectService(c *cfg) {
	metaHdr := new(session.ResponseMetaHeader)
	xHdr := new(session.XHeader)
	xHdr.SetKey("test X-Header key for Object service")
	xHdr.SetValue("test X-Header value for Object service")
	metaHdr.SetXHeaders([]*session.XHeader{xHdr})

	objectGRPC.RegisterObjectServiceServer(c.cfgGRPC.server,
		objectTransportGRPC.New(
			objectService.NewSignService(
				c.key,
				objectService.NewExecutionService(
					new(objectExecutor),
					metaHdr,
				),
			),
		),
	)
}