FailFast(false), return helpful error message when context is time out

grpc · Apr 25, 2019 · 27a8866 · 27a8866
1 parent 3fc7430
commit 27a8866
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/picker_wrapper.go b/picker_wrapper.go
@@ -120,6 +120,14 @@ func (bp *pickerWrapper) pick(ctx context.Context, failfast bool, opts balancer.
  bp.mu.Unlock()
  select {
  case <-ctx.Done():
+ if bp.connectionError() != nil {
+ switch ctx.Err() {
+ case context.DeadlineExceeded:
+ return nil, nil, status.Errorf(codes.DeadlineExceeded, "latest connection error: %v", bp.connectionError())
+ case context.Canceled:
+ return nil, nil, status.Errorf(codes.Canceled, "latest connection error: %v", bp.connectionError())
+ }
+ }
  return nil, nil, ctx.Err()
  case <-ch:
  }