frostfs-api-go/v2/container/client.go

322 lines
8.6 KiB
Go
Raw Normal View History

package container
import (
"context"
"github.com/nspcc-dev/neofs-api-go/v2/client"
container "github.com/nspcc-dev/neofs-api-go/v2/container/grpc"
"github.com/pkg/errors"
"google.golang.org/grpc"
)
// Client represents universal container
// transport client.
type Client struct {
cPut *putClient
cGet *getClient
cDel *delClient
cList *listClient
cSetEACL *setEACLClient
cGetEACL *getEACLClient
}
// Option represents Client option.
type Option func(*cfg)
type cfg struct {
proto client.Protocol
globalOpts []client.Option
gRPC cfgGRPC
}
type cfgGRPC struct {
serviceClient container.ContainerServiceClient
grpcCallOpts []grpc.CallOption
callOpts []container.Option
client *container.Client
}
type putClient struct {
requestConverter func(*PutRequest) interface{}
caller func(context.Context, interface{}) (interface{}, error)
responseConverter func(interface{}) *PutResponse
}
type getClient struct {
requestConverter func(*GetRequest) interface{}
caller func(context.Context, interface{}) (interface{}, error)
responseConverter func(interface{}) *GetResponse
}
type delClient struct {
requestConverter func(*DeleteRequest) interface{}
caller func(context.Context, interface{}) (interface{}, error)
responseConverter func(interface{}) *DeleteResponse
}
type listClient struct {
requestConverter func(*ListRequest) interface{}
caller func(context.Context, interface{}) (interface{}, error)
responseConverter func(interface{}) *ListResponse
}
type setEACLClient struct {
requestConverter func(*SetExtendedACLRequest) interface{}
caller func(context.Context, interface{}) (interface{}, error)
responseConverter func(interface{}) *SetExtendedACLResponse
}
type getEACLClient struct {
requestConverter func(*GetExtendedACLRequest) interface{}
caller func(context.Context, interface{}) (interface{}, error)
responseConverter func(interface{}) *GetExtendedACLResponse
}
// Put sends PutRequest over the network and returns PutResponse.
//
// It returns any error encountered during the call.
func (c *Client) Put(ctx context.Context, req *PutRequest) (*PutResponse, error) {
resp, err := c.cPut.caller(ctx, c.cPut.requestConverter(req))
if err != nil {
return nil, errors.Wrap(err, "could not send container put request")
}
return c.cPut.responseConverter(resp), nil
}
// Get sends GetRequest over the network and returns GetResponse.
//
// It returns any error encountered during the call.
func (c *Client) Get(ctx context.Context, req *GetRequest) (*GetResponse, error) {
resp, err := c.cGet.caller(ctx, c.cGet.requestConverter(req))
if err != nil {
return nil, errors.Wrap(err, "could not send container get request")
}
return c.cGet.responseConverter(resp), nil
}
// Delete sends GetRequest over the network and returns GetResponse.
//
// It returns any error encountered during the call.
func (c *Client) Delete(ctx context.Context, req *DeleteRequest) (*DeleteResponse, error) {
resp, err := c.cDel.caller(ctx, c.cDel.requestConverter(req))
if err != nil {
return nil, errors.Wrap(err, "could not send container delete request")
}
return c.cDel.responseConverter(resp), nil
}
// List sends ListRequest over the network and returns ListResponse.
//
// It returns any error encountered during the call.
func (c *Client) List(ctx context.Context, req *ListRequest) (*ListResponse, error) {
resp, err := c.cList.caller(ctx, c.cList.requestConverter(req))
if err != nil {
return nil, errors.Wrap(err, "could not send container list request")
}
return c.cList.responseConverter(resp), nil
}
// SetExtendedACL sends SetExtendedACLRequest over the network and returns SetExtendedACLResponse.
//
// It returns any error encountered during the call.
func (c *Client) SetExtendedACL(ctx context.Context, req *SetExtendedACLRequest) (*SetExtendedACLResponse, error) {
resp, err := c.cSetEACL.caller(ctx, c.cSetEACL.requestConverter(req))
if err != nil {
return nil, errors.Wrap(err, "could not send container set EACL request")
}
return c.cSetEACL.responseConverter(resp), nil
}
// GetExtendedACL sends GetExtendedACLRequest over the network and returns GetExtendedACLResponse.
//
// It returns any error encountered during the call.
func (c *Client) GetExtendedACL(ctx context.Context, req *GetExtendedACLRequest) (*GetExtendedACLResponse, error) {
resp, err := c.cGetEACL.caller(ctx, c.cGetEACL.requestConverter(req))
if err != nil {
return nil, errors.Wrap(err, "could not send container get EACL request")
}
return c.cGetEACL.responseConverter(resp), nil
}
func defaultCfg() *cfg {
return &cfg{
proto: client.ProtoGRPC,
}
}
func NewClient(opts ...Option) (*Client, error) {
cfg := defaultCfg()
for i := range opts {
opts[i](cfg)
}
var err error
switch cfg.proto {
case client.ProtoGRPC:
var c *container.Client
if c, err = newGRPCClient(cfg); err != nil {
break
}
return &Client{
cPut: &putClient{
requestConverter: func(req *PutRequest) interface{} {
return PutRequestToGRPCMessage(req)
},
caller: func(ctx context.Context, req interface{}) (interface{}, error) {
return c.Put(ctx, req.(*container.PutRequest))
},
responseConverter: func(resp interface{}) *PutResponse {
return PutResponseFromGRPCMessage(resp.(*container.PutResponse))
},
},
cGet: &getClient{
requestConverter: func(req *GetRequest) interface{} {
return GetRequestToGRPCMessage(req)
},
caller: func(ctx context.Context, req interface{}) (interface{}, error) {
return c.Get(ctx, req.(*container.GetRequest))
},
responseConverter: func(resp interface{}) *GetResponse {
return GetResponseFromGRPCMessage(resp.(*container.GetResponse))
},
},
cDel: &delClient{
requestConverter: func(req *DeleteRequest) interface{} {
return DeleteRequestToGRPCMessage(req)
},
caller: func(ctx context.Context, req interface{}) (interface{}, error) {
return c.Delete(ctx, req.(*container.DeleteRequest))
},
responseConverter: func(resp interface{}) *DeleteResponse {
return DeleteResponseFromGRPCMessage(resp.(*container.DeleteResponse))
},
},
cList: &listClient{
requestConverter: func(req *ListRequest) interface{} {
return ListRequestToGRPCMessage(req)
},
caller: func(ctx context.Context, req interface{}) (interface{}, error) {
return c.List(ctx, req.(*container.ListRequest))
},
responseConverter: func(resp interface{}) *ListResponse {
return ListResponseFromGRPCMessage(resp.(*container.ListResponse))
},
},
cSetEACL: &setEACLClient{
requestConverter: func(req *SetExtendedACLRequest) interface{} {
return SetExtendedACLRequestToGRPCMessage(req)
},
caller: func(ctx context.Context, req interface{}) (interface{}, error) {
return c.SetExtendedACL(ctx, req.(*container.SetExtendedACLRequest))
},
responseConverter: func(resp interface{}) *SetExtendedACLResponse {
return SetExtendedACLResponseFromGRPCMessage(resp.(*container.SetExtendedACLResponse))
},
},
cGetEACL: &getEACLClient{
requestConverter: func(req *GetExtendedACLRequest) interface{} {
return GetExtendedACLRequestToGRPCMessage(req)
},
caller: func(ctx context.Context, req interface{}) (interface{}, error) {
return c.GetExtendedACL(ctx, req.(*container.GetExtendedACLRequest))
},
responseConverter: func(resp interface{}) *GetExtendedACLResponse {
return GetExtendedACLResponseFromGRPCMessage(resp.(*container.GetExtendedACLResponse))
},
},
}, nil
default:
err = client.ErrProtoUnsupported
}
return nil, errors.Wrapf(err, "could not create %s Session client", cfg.proto)
}
func newGRPCClient(cfg *cfg) (*container.Client, error) {
var err error
if cfg.gRPC.client == nil {
if cfg.gRPC.serviceClient == nil {
conn, err := client.NewGRPCClientConn(cfg.globalOpts...)
if err != nil {
return nil, errors.Wrap(err, "could not open gRPC client connection")
}
cfg.gRPC.serviceClient = container.NewContainerServiceClient(conn)
}
cfg.gRPC.client, err = container.NewClient(
cfg.gRPC.serviceClient,
append(
cfg.gRPC.callOpts,
container.WithCallOptions(cfg.gRPC.grpcCallOpts),
)...,
)
}
return cfg.gRPC.client, err
}
func WithGlobalOpts(v ...client.Option) Option {
return func(c *cfg) {
if len(v) > 0 {
c.globalOpts = v
}
}
}
func WithGRPCServiceClient(v container.ContainerServiceClient) Option {
return func(c *cfg) {
c.gRPC.serviceClient = v
}
}
func WithGRPCCallOpts(v []grpc.CallOption) Option {
return func(c *cfg) {
c.gRPC.grpcCallOpts = v
}
}
func WithGRPCClientOpts(v []container.Option) Option {
return func(c *cfg) {
c.gRPC.callOpts = v
}
}
func WithGRPCClient(v *container.Client) Option {
return func(c *cfg) {
c.gRPC.client = v
}
}