117 lines
2.8 KiB
Go
117 lines
2.8 KiB
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,
|
|
),
|
|
),
|
|
),
|
|
)
|
|
}
|