@@ -35,7 +35,7 @@ import (
35
35
//
36
36
// The default configuration of the interceptor is to not retry *at all*. This behaviour can be
37
37
// changed through options (e.g. WithMax) on creation of the interceptor or on call (through grpc.CallOptions).
38
- func (c * Client ) unaryClientInterceptor (logger * zap. Logger , optFuncs ... retryOption ) grpc.UnaryClientInterceptor {
38
+ func (c * Client ) unaryClientInterceptor (optFuncs ... retryOption ) grpc.UnaryClientInterceptor {
39
39
intOpts := reuseOrNewWithCallOptions (defaultOptions , optFuncs )
40
40
return func (ctx context.Context , method string , req , reply interface {}, cc * grpc.ClientConn , invoker grpc.UnaryInvoker , opts ... grpc.CallOption ) error {
41
41
ctx = withVersion (ctx )
@@ -50,7 +50,7 @@ func (c *Client) unaryClientInterceptor(logger *zap.Logger, optFuncs ...retryOpt
50
50
if err := waitRetryBackoff (ctx , attempt , callOpts ); err != nil {
51
51
return err
52
52
}
53
- logger .Debug (
53
+ c . GetLogger () .Debug (
54
54
"retrying of unary invoker" ,
55
55
zap .String ("target" , cc .Target ()),
56
56
zap .Uint ("attempt" , attempt ),
@@ -59,7 +59,7 @@ func (c *Client) unaryClientInterceptor(logger *zap.Logger, optFuncs ...retryOpt
59
59
if lastErr == nil {
60
60
return nil
61
61
}
62
- logger .Warn (
62
+ c . GetLogger () .Warn (
63
63
"retrying of unary invoker failed" ,
64
64
zap .String ("target" , cc .Target ()),
65
65
zap .Uint ("attempt" , attempt ),
@@ -82,7 +82,7 @@ func (c *Client) unaryClientInterceptor(logger *zap.Logger, optFuncs ...retryOpt
82
82
83
83
gterr := c .getToken (ctx )
84
84
if gterr != nil {
85
- logger .Warn (
85
+ c . GetLogger () .Warn (
86
86
"retrying of unary invoker failed to fetch new auth token" ,
87
87
zap .String ("target" , cc .Target ()),
88
88
zap .Error (gterr ),
@@ -107,7 +107,7 @@ func (c *Client) unaryClientInterceptor(logger *zap.Logger, optFuncs ...retryOpt
107
107
// Retry logic is available *only for ServerStreams*, i.e. 1:n streams, as the internal logic needs
108
108
// to buffer the messages sent by the client. If retry is enabled on any other streams (ClientStreams,
109
109
// BidiStreams), the retry interceptor will fail the call.
110
- func (c * Client ) streamClientInterceptor (logger * zap. Logger , optFuncs ... retryOption ) grpc.StreamClientInterceptor {
110
+ func (c * Client ) streamClientInterceptor (optFuncs ... retryOption ) grpc.StreamClientInterceptor {
111
111
intOpts := reuseOrNewWithCallOptions (defaultOptions , optFuncs )
112
112
return func (ctx context.Context , desc * grpc.StreamDesc , cc * grpc.ClientConn , method string , streamer grpc.Streamer , opts ... grpc.CallOption ) (grpc.ClientStream , error ) {
113
113
ctx = withVersion (ctx )
@@ -117,7 +117,7 @@ func (c *Client) streamClientInterceptor(logger *zap.Logger, optFuncs ...retryOp
117
117
// equal to c.Username != "" && c.Password != ""
118
118
err := c .getToken (ctx )
119
119
if err != nil && rpctypes .Error (err ) != rpctypes .ErrAuthNotEnabled {
120
- logger .Error ("clientv3/retry_interceptor: getToken failed" , zap .Error (err ))
120
+ c . GetLogger () .Error ("clientv3/retry_interceptor: getToken failed" , zap .Error (err ))
121
121
return nil , err
122
122
}
123
123
}
@@ -132,7 +132,7 @@ func (c *Client) streamClientInterceptor(logger *zap.Logger, optFuncs ...retryOp
132
132
}
133
133
newStreamer , err := streamer (ctx , desc , cc , method , grpcOpts ... )
134
134
if err != nil {
135
- logger .Error ("streamer failed to create ClientStream" , zap .Error (err ))
135
+ c . GetLogger () .Error ("streamer failed to create ClientStream" , zap .Error (err ))
136
136
return nil , err // TODO(mwitkow): Maybe dial and transport errors should be retriable?
137
137
}
138
138
retryingStreamer := & serverStreamingRetryingStream {
0 commit comments