[#419] rpc/client: Use provided context for client dial
In previous implementation `Client` passed `context.Background()` to `grpc.DialContext` function. This didn't allow to abort dial stage by the given context. Base dial context on the one provided with `WithContext` option. Fall back to using `context.Background` if context is not specified. Signed-off-by: Leonard Lyubich <ctulhurider@gmail.com>
This commit is contained in:
parent
2b89b7e798
commit
3a91383f24
3 changed files with 19 additions and 13 deletions
|
@ -2,26 +2,28 @@ package client
|
|||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/nspcc-dev/neofs-api-go/v2/rpc/grpc"
|
||||
)
|
||||
|
||||
// CallOption is a messaging session option within Protobuf RPC.
|
||||
type CallOption func(*callParameters)
|
||||
|
||||
type callParameters struct {
|
||||
callOpts []grpc.CallOption
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
func defaultCallParameters() *callParameters {
|
||||
return &callParameters{
|
||||
callOpts: make([]grpc.CallOption, 0, 1),
|
||||
ctx: context.Background(),
|
||||
}
|
||||
}
|
||||
|
||||
// WithContext return options to specify call context.
|
||||
// WithContext returns option to specify call context. If provided, all network
|
||||
// communications will be based on this context. Otherwise, context.Background()
|
||||
// is used.
|
||||
//
|
||||
// Context SHOULD NOT be nil.
|
||||
func WithContext(ctx context.Context) CallOption {
|
||||
return func(prm *callParameters) {
|
||||
prm.callOpts = append(prm.callOpts, grpc.WithContext(ctx))
|
||||
prm.ctx = ctx
|
||||
}
|
||||
}
|
||||
|
|
|
@ -13,9 +13,10 @@ import (
|
|||
"google.golang.org/grpc/credentials/insecure"
|
||||
)
|
||||
|
||||
func (c *Client) createGRPCClient() (err error) {
|
||||
func (c *Client) createGRPCClient(ctx context.Context) (err error) {
|
||||
c.gRPCClientOnce.Do(func() {
|
||||
if err = c.openGRPCConn(); err != nil {
|
||||
if err = c.openGRPCConn(ctx); err != nil {
|
||||
err = fmt.Errorf("open gRPC connection: %w", err)
|
||||
return
|
||||
}
|
||||
|
||||
|
@ -30,7 +31,7 @@ func (c *Client) createGRPCClient() (err error) {
|
|||
|
||||
var errInvalidEndpoint = errors.New("invalid endpoint options")
|
||||
|
||||
func (c *Client) openGRPCConn() error {
|
||||
func (c *Client) openGRPCConn(ctx context.Context) error {
|
||||
if c.conn != nil {
|
||||
return nil
|
||||
}
|
||||
|
@ -47,15 +48,18 @@ func (c *Client) openGRPCConn() error {
|
|||
creds = insecure.NewCredentials()
|
||||
}
|
||||
|
||||
dialCtx, cancel := context.WithTimeout(context.Background(), c.dialTimeout)
|
||||
dialCtx, cancel := context.WithTimeout(ctx, c.dialTimeout)
|
||||
var err error
|
||||
|
||||
c.conn, err = grpcstd.DialContext(dialCtx, c.addr,
|
||||
grpcstd.WithTransportCredentials(creds),
|
||||
grpcstd.WithBlock(),
|
||||
)
|
||||
|
||||
cancel()
|
||||
|
||||
if err != nil {
|
||||
return fmt.Errorf("open gRPC client connection: %w", err)
|
||||
return fmt.Errorf("gRPC dial: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
|
|
|
@ -68,11 +68,11 @@ func (g rwGRPC) WriteMessage(m message.Message) error {
|
|||
}
|
||||
|
||||
func (c *Client) initGRPC(info common.CallMethodInfo, prm *callParameters) (MessageReadWriter, error) {
|
||||
if err := c.createGRPCClient(); err != nil {
|
||||
if err := c.createGRPCClient(prm.ctx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
rw, err := c.gRPCClient.Init(info, prm.callOpts...)
|
||||
rw, err := c.gRPCClient.Init(info, grpc.WithContext(prm.ctx))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue