forked from TrueCloudLab/frostfs-node
126 lines
3 KiB
Go
126 lines
3 KiB
Go
package getsvc
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ecdsa"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/client"
|
|
objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
|
|
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/engine"
|
|
"github.com/nspcc-dev/neofs-node/pkg/network/cache"
|
|
headsvc "github.com/nspcc-dev/neofs-node/pkg/services/object/head"
|
|
"github.com/nspcc-dev/neofs-node/pkg/services/object/util"
|
|
"github.com/nspcc-dev/neofs-node/pkg/services/object_manager/placement"
|
|
"github.com/nspcc-dev/neofs-node/pkg/util/logger"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// Service utility serving requests of Object.Get service.
|
|
type Service struct {
|
|
*cfg
|
|
}
|
|
|
|
// Option is a Service's constructor option.
|
|
type Option func(*cfg)
|
|
|
|
type getClient interface {
|
|
GetObject(context.Context, Prm) (*objectSDK.Object, error)
|
|
}
|
|
|
|
type cfg struct {
|
|
assembly bool
|
|
|
|
log *logger.Logger
|
|
|
|
headSvc interface {
|
|
head(context.Context, Prm) (*object.Object, error)
|
|
}
|
|
|
|
localStorage interface {
|
|
Get(*objectSDK.Address) (*object.Object, error)
|
|
}
|
|
|
|
clientCache interface {
|
|
get(*ecdsa.PrivateKey, string) (getClient, error)
|
|
}
|
|
|
|
traverserGenerator interface {
|
|
GenerateTraverser(*objectSDK.Address) (*placement.Traverser, error)
|
|
}
|
|
}
|
|
|
|
func defaultCfg() *cfg {
|
|
return &cfg{
|
|
assembly: true,
|
|
log: zap.L(),
|
|
headSvc: new(headSvcWrapper),
|
|
localStorage: new(storageEngineWrapper),
|
|
clientCache: new(clientCacheWrapper),
|
|
}
|
|
}
|
|
|
|
// New creates, initializes and returns utility serving
|
|
// Object.Get service requests.
|
|
func New(opts ...Option) *Service {
|
|
c := defaultCfg()
|
|
|
|
for i := range opts {
|
|
opts[i](c)
|
|
}
|
|
|
|
return &Service{
|
|
cfg: c,
|
|
}
|
|
}
|
|
|
|
// WithLogger returns option to specify Get service's logger.
|
|
func WithLogger(l *logger.Logger) Option {
|
|
return func(c *cfg) {
|
|
c.log = l.With(zap.String("component", "Object.Get service"))
|
|
}
|
|
}
|
|
|
|
// WithoutAssembly returns option to disable object assembling.
|
|
func WithoutAssembly() Option {
|
|
return func(c *cfg) {
|
|
c.assembly = false
|
|
}
|
|
}
|
|
|
|
// WithLocalStorageEngine returns option to set local storage
|
|
// instance.
|
|
func WithLocalStorageEngine(e *engine.StorageEngine) Option {
|
|
return func(c *cfg) {
|
|
c.localStorage.(*storageEngineWrapper).engine = e
|
|
}
|
|
}
|
|
|
|
// WithClientCache returns option to set cache of remote node clients.
|
|
func WithClientCache(v *cache.ClientCache) Option {
|
|
return func(c *cfg) {
|
|
c.clientCache.(*clientCacheWrapper).cache = v
|
|
}
|
|
}
|
|
|
|
// WithClientOptions returns option to specify options of remote node clients.
|
|
func WithClientOptions(opts ...client.Option) Option {
|
|
return func(c *cfg) {
|
|
c.clientCache.(*clientCacheWrapper).opts = opts
|
|
}
|
|
}
|
|
|
|
// WithTraverserGenerator returns option to set generator of
|
|
// placement traverser to get the objects from containers.
|
|
func WithTraverserGenerator(t *util.TraverserGenerator) Option {
|
|
return func(c *cfg) {
|
|
c.traverserGenerator = t
|
|
}
|
|
}
|
|
|
|
// WithHeadService returns option to set the utility serving object.Head.
|
|
func WithHeadService(svc *headsvc.Service) Option {
|
|
return func(c *cfg) {
|
|
c.headSvc.(*headSvcWrapper).svc = svc
|
|
}
|
|
}
|