2016-07-11 19:16:16 +00:00
/ *
*
* Copyright 2014 , Google Inc .
* All rights reserved .
*
* Redistribution and use in source and binary forms , with or without
* modification , are permitted provided that the following conditions are
* met :
*
* * Redistributions of source code must retain the above copyright
* notice , this list of conditions and the following disclaimer .
* * Redistributions in binary form must reproduce the above
* copyright notice , this list of conditions and the following disclaimer
* in the documentation and / or other materials provided with the
* distribution .
* * Neither the name of Google Inc . nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission .
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES , INCLUDING , BUT NOT
* LIMITED TO , THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED . IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT , INDIRECT , INCIDENTAL ,
* SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES ( INCLUDING , BUT NOT
* LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE ,
* DATA , OR PROFITS ; OR BUSINESS INTERRUPTION ) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
* ( INCLUDING NEGLIGENCE OR OTHERWISE ) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
*
* /
package grpc
import (
"errors"
"fmt"
"net"
"strings"
"sync"
"time"
"golang.org/x/net/context"
"golang.org/x/net/trace"
"google.golang.org/grpc/credentials"
"google.golang.org/grpc/grpclog"
"google.golang.org/grpc/transport"
)
var (
// ErrClientConnClosing indicates that the operation is illegal because
2016-08-30 22:27:31 +00:00
// the ClientConn is closing.
2016-07-11 19:16:16 +00:00
ErrClientConnClosing = errors . New ( "grpc: the client connection is closing" )
2016-08-30 22:27:31 +00:00
// ErrClientConnTimeout indicates that the ClientConn cannot establish the
// underlying connections within the specified timeout.
ErrClientConnTimeout = errors . New ( "grpc: timed out when dialing" )
// errNoTransportSecurity indicates that there is no transport security
// being set for ClientConn. Users should either set one or explicitly
// call WithInsecure DialOption to disable security.
errNoTransportSecurity = errors . New ( "grpc: no transport security set (use grpc.WithInsecure() explicitly or set credentials)" )
// errTransportCredentialsMissing indicates that users want to transmit security
// information (e.g., oauth2 token) which requires secure connection on an insecure
// connection.
errTransportCredentialsMissing = errors . New ( "grpc: the credentials require transport level security (use grpc.WithTransportCredentials() to set)" )
// errCredentialsConflict indicates that grpc.WithTransportCredentials()
// and grpc.WithInsecure() are both called for a connection.
errCredentialsConflict = errors . New ( "grpc: transport credentials are set for an insecure connection (grpc.WithTransportCredentials() and grpc.WithInsecure() are both called)" )
// errNetworkIO indicates that the connection is down due to some network I/O error.
errNetworkIO = errors . New ( "grpc: failed with network I/O error" )
// errConnDrain indicates that the connection starts to be drained and does not accept any new RPCs.
errConnDrain = errors . New ( "grpc: the connection is drained" )
// errConnClosing indicates that the connection is closing.
errConnClosing = errors . New ( "grpc: the connection is closing" )
// errConnUnavailable indicates that the connection is unavailable.
errConnUnavailable = errors . New ( "grpc: the connection is unavailable" )
errNoAddr = errors . New ( "grpc: there is no address available to dial" )
2016-07-11 19:16:16 +00:00
// minimum time to give a connection to complete
minConnectTimeout = 20 * time . Second
)
// dialOptions configure a Dial call. dialOptions are set by the DialOption
// values passed to Dial.
type dialOptions struct {
codec Codec
2016-08-30 22:27:31 +00:00
cp Compressor
dc Decompressor
bs backoffStrategy
balancer Balancer
2016-07-11 19:16:16 +00:00
block bool
insecure bool
2016-08-30 22:27:31 +00:00
timeout time . Duration
2016-07-11 19:16:16 +00:00
copts transport . ConnectOptions
}
// DialOption configures how we set up the connection.
type DialOption func ( * dialOptions )
// WithCodec returns a DialOption which sets a codec for message marshaling and unmarshaling.
func WithCodec ( c Codec ) DialOption {
return func ( o * dialOptions ) {
o . codec = c
}
}
2016-08-30 22:27:31 +00:00
// WithCompressor returns a DialOption which sets a CompressorGenerator for generating message
// compressor.
func WithCompressor ( cp Compressor ) DialOption {
return func ( o * dialOptions ) {
o . cp = cp
}
}
// WithDecompressor returns a DialOption which sets a DecompressorGenerator for generating
// message decompressor.
func WithDecompressor ( dc Decompressor ) DialOption {
return func ( o * dialOptions ) {
o . dc = dc
}
}
// WithBalancer returns a DialOption which sets a load balancer.
func WithBalancer ( b Balancer ) DialOption {
2016-07-11 19:16:16 +00:00
return func ( o * dialOptions ) {
2016-08-30 22:27:31 +00:00
o . balancer = b
}
}
// WithBackoffMaxDelay configures the dialer to use the provided maximum delay
// when backing off after failed connection attempts.
func WithBackoffMaxDelay ( md time . Duration ) DialOption {
return WithBackoffConfig ( BackoffConfig { MaxDelay : md } )
}
// WithBackoffConfig configures the dialer to use the provided backoff
// parameters after connection failures.
//
// Use WithBackoffMaxDelay until more parameters on BackoffConfig are opened up
// for use.
func WithBackoffConfig ( b BackoffConfig ) DialOption {
// Set defaults to ensure that provided BackoffConfig is valid and
// unexported fields get default values.
setDefaults ( & b )
return withBackoff ( b )
}
// withBackoff sets the backoff strategy used for retries after a
// failed connection attempt.
//
// This can be exported if arbitrary backoff strategies are allowed by gRPC.
func withBackoff ( bs backoffStrategy ) DialOption {
return func ( o * dialOptions ) {
o . bs = bs
2016-07-11 19:16:16 +00:00
}
}
// WithBlock returns a DialOption which makes caller of Dial blocks until the underlying
// connection is up. Without this, Dial returns immediately and connecting the server
// happens in background.
func WithBlock ( ) DialOption {
return func ( o * dialOptions ) {
o . block = true
}
}
2016-08-30 22:27:31 +00:00
// WithInsecure returns a DialOption which disables transport security for this ClientConn.
// Note that transport security is required unless WithInsecure is set.
2016-07-11 19:16:16 +00:00
func WithInsecure ( ) DialOption {
return func ( o * dialOptions ) {
o . insecure = true
}
}
// WithTransportCredentials returns a DialOption which configures a
// connection level security credentials (e.g., TLS/SSL).
2016-08-30 22:27:31 +00:00
func WithTransportCredentials ( creds credentials . TransportCredentials ) DialOption {
2016-07-11 19:16:16 +00:00
return func ( o * dialOptions ) {
2016-08-30 22:27:31 +00:00
o . copts . TransportCredentials = creds
2016-07-11 19:16:16 +00:00
}
}
// WithPerRPCCredentials returns a DialOption which sets
// credentials which will place auth state on each outbound RPC.
2016-08-30 22:27:31 +00:00
func WithPerRPCCredentials ( creds credentials . PerRPCCredentials ) DialOption {
2016-07-11 19:16:16 +00:00
return func ( o * dialOptions ) {
2016-08-30 22:27:31 +00:00
o . copts . PerRPCCredentials = append ( o . copts . PerRPCCredentials , creds )
2016-07-11 19:16:16 +00:00
}
}
2016-08-30 22:27:31 +00:00
// WithTimeout returns a DialOption that configures a timeout for dialing a ClientConn
// initially. This is valid if and only if WithBlock() is present.
2016-07-11 19:16:16 +00:00
func WithTimeout ( d time . Duration ) DialOption {
return func ( o * dialOptions ) {
2016-08-30 22:27:31 +00:00
o . timeout = d
2016-07-11 19:16:16 +00:00
}
}
// WithDialer returns a DialOption that specifies a function to use for dialing network addresses.
2016-08-30 22:27:31 +00:00
func WithDialer ( f func ( string , time . Duration ) ( net . Conn , error ) ) DialOption {
2016-07-11 19:16:16 +00:00
return func ( o * dialOptions ) {
2016-08-30 22:27:31 +00:00
o . copts . Dialer = func ( ctx context . Context , addr string ) ( net . Conn , error ) {
if deadline , ok := ctx . Deadline ( ) ; ok {
return f ( addr , deadline . Sub ( time . Now ( ) ) )
}
return f ( addr , 0 )
}
2016-07-11 19:16:16 +00:00
}
}
// WithUserAgent returns a DialOption that specifies a user agent string for all the RPCs.
func WithUserAgent ( s string ) DialOption {
return func ( o * dialOptions ) {
o . copts . UserAgent = s
}
}
2016-08-30 22:27:31 +00:00
// Dial creates a client connection to the given target.
2016-07-11 19:16:16 +00:00
func Dial ( target string , opts ... DialOption ) ( * ClientConn , error ) {
2016-08-30 22:27:31 +00:00
return DialContext ( context . Background ( ) , target , opts ... )
}
2016-09-17 13:50:35 +00:00
// DialContext creates a client connection to the given target
// using the supplied context.
func DialContext ( ctx context . Context , target string , opts ... DialOption ) ( * ClientConn , error ) {
2016-07-11 19:16:16 +00:00
cc := & ClientConn {
target : target ,
2016-08-30 22:27:31 +00:00
conns : make ( map [ Address ] * addrConn ) ,
2016-07-11 19:16:16 +00:00
}
2016-09-17 13:50:35 +00:00
cc . ctx , cc . cancel = context . WithCancel ( ctx )
2016-07-11 19:16:16 +00:00
for _ , opt := range opts {
opt ( & cc . dopts )
}
2016-08-30 22:27:31 +00:00
// Set defaults.
2016-07-11 19:16:16 +00:00
if cc . dopts . codec == nil {
cc . dopts . codec = protoCodec { }
}
2016-08-30 22:27:31 +00:00
if cc . dopts . bs == nil {
cc . dopts . bs = DefaultBackoffConfig
}
var (
ok bool
addrs [ ] Address
)
if cc . dopts . balancer == nil {
// Connect to target directly if balancer is nil.
addrs = append ( addrs , Address { Addr : target } )
} else {
if err := cc . dopts . balancer . Start ( target ) ; err != nil {
return nil , err
}
ch := cc . dopts . balancer . Notify ( )
if ch == nil {
// There is no name resolver installed.
addrs = append ( addrs , Address { Addr : target } )
} else {
addrs , ok = <- ch
if ! ok || len ( addrs ) == 0 {
return nil , errNoAddr
}
}
}
waitC := make ( chan error , 1 )
go func ( ) {
for _ , a := range addrs {
if err := cc . resetAddrConn ( a , false , nil ) ; err != nil {
waitC <- err
return
}
}
close ( waitC )
} ( )
var timeoutCh <- chan time . Time
if cc . dopts . timeout > 0 {
timeoutCh = time . After ( cc . dopts . timeout )
}
select {
case err := <- waitC :
if err != nil {
2016-09-17 13:50:35 +00:00
cc . Close ( )
2016-08-30 22:27:31 +00:00
return nil , err
2016-07-11 19:16:16 +00:00
}
2016-09-17 13:50:35 +00:00
case <- cc . ctx . Done ( ) :
cc . Close ( )
return nil , cc . ctx . Err ( )
2016-08-30 22:27:31 +00:00
case <- timeoutCh :
2016-09-17 13:50:35 +00:00
cc . Close ( )
2016-08-30 22:27:31 +00:00
return nil , ErrClientConnTimeout
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
// If balancer is nil or balancer.Notify() is nil, ok will be false here.
// The lbWatcher goroutine will not be created.
if ok {
go cc . lbWatcher ( )
2016-07-11 19:16:16 +00:00
}
colonPos := strings . LastIndex ( target , ":" )
if colonPos == - 1 {
colonPos = len ( target )
}
cc . authority = target [ : colonPos ]
return cc , nil
}
// ConnectivityState indicates the state of a client connection.
type ConnectivityState int
const (
// Idle indicates the ClientConn is idle.
Idle ConnectivityState = iota
// Connecting indicates the ClienConn is connecting.
Connecting
// Ready indicates the ClientConn is ready for work.
Ready
// TransientFailure indicates the ClientConn has seen a failure but expects to recover.
TransientFailure
// Shutdown indicates the ClientConn has started shutting down.
Shutdown
)
func ( s ConnectivityState ) String ( ) string {
switch s {
case Idle :
return "IDLE"
case Connecting :
return "CONNECTING"
case Ready :
return "READY"
case TransientFailure :
return "TRANSIENT_FAILURE"
case Shutdown :
return "SHUTDOWN"
default :
panic ( fmt . Sprintf ( "unknown connectivity state: %d" , s ) )
}
}
2016-08-30 22:27:31 +00:00
// ClientConn represents a client connection to an RPC server.
2016-07-11 19:16:16 +00:00
type ClientConn struct {
2016-08-30 22:27:31 +00:00
ctx context . Context
cancel context . CancelFunc
2016-07-11 19:16:16 +00:00
target string
authority string
dopts dialOptions
2016-08-30 22:27:31 +00:00
mu sync . RWMutex
conns map [ Address ] * addrConn
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
func ( cc * ClientConn ) lbWatcher ( ) {
for addrs := range cc . dopts . balancer . Notify ( ) {
var (
add [ ] Address // Addresses need to setup connections.
del [ ] * addrConn // Connections need to tear down.
)
cc . mu . Lock ( )
for _ , a := range addrs {
if _ , ok := cc . conns [ a ] ; ! ok {
add = append ( add , a )
}
}
for k , c := range cc . conns {
var keep bool
for _ , a := range addrs {
if k == a {
keep = true
break
}
}
if ! keep {
del = append ( del , c )
delete ( cc . conns , c . addr )
}
}
cc . mu . Unlock ( )
for _ , a := range add {
cc . resetAddrConn ( a , true , nil )
}
for _ , c := range del {
c . tearDown ( errConnDrain )
}
}
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
// resetAddrConn creates an addrConn for addr and adds it to cc.conns.
// If there is an old addrConn for addr, it will be torn down, using tearDownErr as the reason.
// If tearDownErr is nil, errConnDrain will be used instead.
func ( cc * ClientConn ) resetAddrConn ( addr Address , skipWait bool , tearDownErr error ) error {
ac := & addrConn {
cc : cc ,
addr : addr ,
dopts : cc . dopts ,
}
ac . ctx , ac . cancel = context . WithCancel ( cc . ctx )
ac . stateCV = sync . NewCond ( & ac . mu )
2016-07-11 19:16:16 +00:00
if EnableTracing {
2016-08-30 22:27:31 +00:00
ac . events = trace . NewEventLog ( "grpc.ClientConn" , ac . addr . Addr )
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
if ! ac . dopts . insecure {
if ac . dopts . copts . TransportCredentials == nil {
return errNoTransportSecurity
2016-07-11 19:16:16 +00:00
}
} else {
2016-08-30 22:27:31 +00:00
if ac . dopts . copts . TransportCredentials != nil {
return errCredentialsConflict
}
for _ , cd := range ac . dopts . copts . PerRPCCredentials {
2016-07-11 19:16:16 +00:00
if cd . RequireTransportSecurity ( ) {
2016-08-30 22:27:31 +00:00
return errTransportCredentialsMissing
2016-07-11 19:16:16 +00:00
}
}
}
2016-08-30 22:27:31 +00:00
// Track ac in cc. This needs to be done before any getTransport(...) is called.
cc . mu . Lock ( )
if cc . conns == nil {
cc . mu . Unlock ( )
return ErrClientConnClosing
}
stale := cc . conns [ ac . addr ]
cc . conns [ ac . addr ] = ac
cc . mu . Unlock ( )
if stale != nil {
// There is an addrConn alive on ac.addr already. This could be due to
// 1) a buggy Balancer notifies duplicated Addresses;
// 2) goaway was received, a new ac will replace the old ac.
// The old ac should be deleted from cc.conns, but the
// underlying transport should drain rather than close.
if tearDownErr == nil {
// tearDownErr is nil if resetAddrConn is called by
// 1) Dial
// 2) lbWatcher
// In both cases, the stale ac should drain, not close.
stale . tearDown ( errConnDrain )
} else {
stale . tearDown ( tearDownErr )
}
}
// skipWait may overwrite the decision in ac.dopts.block.
if ac . dopts . block && ! skipWait {
if err := ac . resetTransport ( false ) ; err != nil {
if err != errConnClosing {
// Tear down ac and delete it from cc.conns.
cc . mu . Lock ( )
delete ( cc . conns , ac . addr )
cc . mu . Unlock ( )
ac . tearDown ( err )
}
if e , ok := err . ( transport . ConnectionError ) ; ok && ! e . Temporary ( ) {
return e . Origin ( )
}
return err
2016-07-11 19:16:16 +00:00
}
// Start to monitor the error status of transport.
2016-08-30 22:27:31 +00:00
go ac . transportMonitor ( )
2016-07-11 19:16:16 +00:00
} else {
// Start a goroutine connecting to the server asynchronously.
go func ( ) {
2016-08-30 22:27:31 +00:00
if err := ac . resetTransport ( false ) ; err != nil {
grpclog . Printf ( "Failed to dial %s: %v; please retry." , ac . addr . Addr , err )
if err != errConnClosing {
// Keep this ac in cc.conns, to get the reason it's torn down.
ac . tearDown ( err )
}
2016-07-11 19:16:16 +00:00
return
}
2016-08-30 22:27:31 +00:00
ac . transportMonitor ( )
2016-07-11 19:16:16 +00:00
} ( )
}
2016-08-30 22:27:31 +00:00
return nil
}
func ( cc * ClientConn ) getTransport ( ctx context . Context , opts BalancerGetOptions ) ( transport . ClientTransport , func ( ) , error ) {
var (
ac * addrConn
ok bool
put func ( )
)
if cc . dopts . balancer == nil {
// If balancer is nil, there should be only one addrConn available.
cc . mu . RLock ( )
if cc . conns == nil {
cc . mu . RUnlock ( )
return nil , nil , toRPCErr ( ErrClientConnClosing )
}
for _ , ac = range cc . conns {
// Break after the first iteration to get the first addrConn.
ok = true
break
}
cc . mu . RUnlock ( )
} else {
var (
addr Address
err error
)
addr , put , err = cc . dopts . balancer . Get ( ctx , opts )
if err != nil {
return nil , nil , toRPCErr ( err )
}
cc . mu . RLock ( )
if cc . conns == nil {
cc . mu . RUnlock ( )
return nil , nil , toRPCErr ( ErrClientConnClosing )
}
ac , ok = cc . conns [ addr ]
cc . mu . RUnlock ( )
}
if ! ok {
if put != nil {
put ( )
}
return nil , nil , errConnClosing
}
t , err := ac . wait ( ctx , cc . dopts . balancer != nil , ! opts . BlockingWait )
if err != nil {
if put != nil {
put ( )
}
return nil , nil , err
}
return t , put , nil
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
// Close tears down the ClientConn and all underlying connections.
func ( cc * ClientConn ) Close ( ) error {
cc . cancel ( )
cc . mu . Lock ( )
if cc . conns == nil {
cc . mu . Unlock ( )
return ErrClientConnClosing
}
conns := cc . conns
cc . conns = nil
cc . mu . Unlock ( )
if cc . dopts . balancer != nil {
cc . dopts . balancer . Close ( )
}
for _ , ac := range conns {
ac . tearDown ( ErrClientConnClosing )
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
return nil
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
// addrConn is a network connection to a given address.
type addrConn struct {
ctx context . Context
cancel context . CancelFunc
cc * ClientConn
addr Address
dopts dialOptions
events trace . EventLog
mu sync . Mutex
state ConnectivityState
stateCV * sync . Cond
down func ( error ) // the handler called when a connection is down.
// ready is closed and becomes nil when a new transport is up or failed
// due to timeout.
ready chan struct { }
transport transport . ClientTransport
// The reason this addrConn is torn down.
tearDownErr error
}
// printf records an event in ac's event log, unless ac has been closed.
// REQUIRES ac.mu is held.
func ( ac * addrConn ) printf ( format string , a ... interface { } ) {
if ac . events != nil {
ac . events . Printf ( format , a ... )
2016-07-11 19:16:16 +00:00
}
}
2016-08-30 22:27:31 +00:00
// errorf records an error in ac's event log, unless ac has been closed.
// REQUIRES ac.mu is held.
func ( ac * addrConn ) errorf ( format string , a ... interface { } ) {
if ac . events != nil {
ac . events . Errorf ( format , a ... )
}
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
// getState returns the connectivity state of the Conn
func ( ac * addrConn ) getState ( ) ConnectivityState {
ac . mu . Lock ( )
defer ac . mu . Unlock ( )
return ac . state
}
// waitForStateChange blocks until the state changes to something other than the sourceState.
func ( ac * addrConn ) waitForStateChange ( ctx context . Context , sourceState ConnectivityState ) ( ConnectivityState , error ) {
ac . mu . Lock ( )
defer ac . mu . Unlock ( )
if sourceState != ac . state {
return ac . state , nil
2016-07-11 19:16:16 +00:00
}
done := make ( chan struct { } )
var err error
go func ( ) {
select {
case <- ctx . Done ( ) :
2016-08-30 22:27:31 +00:00
ac . mu . Lock ( )
2016-07-11 19:16:16 +00:00
err = ctx . Err ( )
2016-08-30 22:27:31 +00:00
ac . stateCV . Broadcast ( )
ac . mu . Unlock ( )
2016-07-11 19:16:16 +00:00
case <- done :
}
} ( )
defer close ( done )
2016-08-30 22:27:31 +00:00
for sourceState == ac . state {
ac . stateCV . Wait ( )
2016-07-11 19:16:16 +00:00
if err != nil {
2016-08-30 22:27:31 +00:00
return ac . state , err
2016-07-11 19:16:16 +00:00
}
}
2016-08-30 22:27:31 +00:00
return ac . state , nil
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
func ( ac * addrConn ) resetTransport ( closeTransport bool ) error {
for retries := 0 ; ; retries ++ {
ac . mu . Lock ( )
ac . printf ( "connecting" )
if ac . state == Shutdown {
// ac.tearDown(...) has been invoked.
ac . mu . Unlock ( )
return errConnClosing
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
if ac . down != nil {
ac . down ( downErrorf ( false , true , "%v" , errNetworkIO ) )
ac . down = nil
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ac . state = Connecting
ac . stateCV . Broadcast ( )
t := ac . transport
ac . mu . Unlock ( )
if closeTransport && t != nil {
t . Close ( )
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
sleepTime := ac . dopts . bs . backoff ( retries )
timeout := minConnectTimeout
if timeout < sleepTime {
timeout = sleepTime
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ctx , cancel := context . WithTimeout ( ac . ctx , timeout )
2016-07-11 19:16:16 +00:00
connectTime := time . Now ( )
2016-08-30 22:27:31 +00:00
newTransport , err := transport . NewClientTransport ( ctx , ac . addr . Addr , ac . dopts . copts )
2016-07-11 19:16:16 +00:00
if err != nil {
2016-08-30 22:27:31 +00:00
cancel ( )
if e , ok := err . ( transport . ConnectionError ) ; ok && ! e . Temporary ( ) {
return err
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
grpclog . Printf ( "grpc: addrConn.resetTransport failed to create client transport: %v; Reconnecting to %q" , err , ac . addr )
ac . mu . Lock ( )
if ac . state == Shutdown {
// ac.tearDown(...) has been invoked.
ac . mu . Unlock ( )
return errConnClosing
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ac . errorf ( "transient failure: %v" , err )
ac . state = TransientFailure
ac . stateCV . Broadcast ( )
if ac . ready != nil {
close ( ac . ready )
ac . ready = nil
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ac . mu . Unlock ( )
2016-07-11 19:16:16 +00:00
closeTransport = false
2016-08-30 22:27:31 +00:00
select {
case <- time . After ( sleepTime - time . Since ( connectTime ) ) :
case <- ac . ctx . Done ( ) :
return ac . ctx . Err ( )
}
2016-07-11 19:16:16 +00:00
continue
}
2016-08-30 22:27:31 +00:00
ac . mu . Lock ( )
ac . printf ( "ready" )
if ac . state == Shutdown {
// ac.tearDown(...) has been invoked.
ac . mu . Unlock ( )
2016-07-11 19:16:16 +00:00
newTransport . Close ( )
2016-08-30 22:27:31 +00:00
return errConnClosing
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ac . state = Ready
ac . stateCV . Broadcast ( )
ac . transport = newTransport
if ac . ready != nil {
close ( ac . ready )
ac . ready = nil
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
if ac . cc . dopts . balancer != nil {
ac . down = ac . cc . dopts . balancer . Up ( ac . addr )
}
ac . mu . Unlock ( )
2016-07-11 19:16:16 +00:00
return nil
}
}
// Run in a goroutine to track the error in transport and create the
// new transport if an error happens. It returns when the channel is closing.
2016-08-30 22:27:31 +00:00
func ( ac * addrConn ) transportMonitor ( ) {
2016-07-11 19:16:16 +00:00
for {
2016-08-30 22:27:31 +00:00
ac . mu . Lock ( )
t := ac . transport
ac . mu . Unlock ( )
2016-07-11 19:16:16 +00:00
select {
2016-08-30 22:27:31 +00:00
// This is needed to detect the teardown when
// the addrConn is idle (i.e., no RPC in flight).
case <- ac . ctx . Done ( ) :
select {
case <- t . Error ( ) :
t . Close ( )
default :
}
2016-07-11 19:16:16 +00:00
return
2016-08-30 22:27:31 +00:00
case <- t . GoAway ( ) :
// If GoAway happens without any network I/O error, ac is closed without shutting down the
// underlying transport (the transport will be closed when all the pending RPCs finished or
// failed.).
// If GoAway and some network I/O error happen concurrently, ac and its underlying transport
// are closed.
// In both cases, a new ac is created.
select {
case <- t . Error ( ) :
ac . cc . resetAddrConn ( ac . addr , true , errNetworkIO )
default :
ac . cc . resetAddrConn ( ac . addr , true , errConnDrain )
}
return
case <- t . Error ( ) :
select {
case <- ac . ctx . Done ( ) :
t . Close ( )
return
case <- t . GoAway ( ) :
ac . cc . resetAddrConn ( ac . addr , true , errNetworkIO )
2016-07-11 19:16:16 +00:00
return
2016-08-30 22:27:31 +00:00
default :
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ac . mu . Lock ( )
if ac . state == Shutdown {
// ac has been shutdown.
ac . mu . Unlock ( )
2016-07-11 19:16:16 +00:00
return
}
2016-08-30 22:27:31 +00:00
ac . state = TransientFailure
ac . stateCV . Broadcast ( )
ac . mu . Unlock ( )
if err := ac . resetTransport ( true ) ; err != nil {
ac . mu . Lock ( )
ac . printf ( "transport exiting: %v" , err )
ac . mu . Unlock ( )
grpclog . Printf ( "grpc: addrConn.transportMonitor exits due to: %v" , err )
if err != errConnClosing {
// Keep this ac in cc.conns, to get the reason it's torn down.
ac . tearDown ( err )
}
return
2016-07-11 19:16:16 +00:00
}
}
}
}
2016-08-30 22:27:31 +00:00
// wait blocks until i) the new transport is up or ii) ctx is done or iii) ac is closed or
// iv) transport is in TransientFailure and there's no balancer/failfast is true.
func ( ac * addrConn ) wait ( ctx context . Context , hasBalancer , failfast bool ) ( transport . ClientTransport , error ) {
2016-07-11 19:16:16 +00:00
for {
2016-08-30 22:27:31 +00:00
ac . mu . Lock ( )
2016-07-11 19:16:16 +00:00
switch {
2016-08-30 22:27:31 +00:00
case ac . state == Shutdown :
if failfast || ! hasBalancer {
// RPC is failfast or balancer is nil. This RPC should fail with ac.tearDownErr.
err := ac . tearDownErr
ac . mu . Unlock ( )
return nil , err
2016-07-11 19:16:16 +00:00
}
2016-08-30 22:27:31 +00:00
ac . mu . Unlock ( )
return nil , errConnClosing
case ac . state == Ready :
ct := ac . transport
ac . mu . Unlock ( )
return ct , nil
case ac . state == TransientFailure :
if failfast || hasBalancer {
ac . mu . Unlock ( )
return nil , errConnUnavailable
2016-07-11 19:16:16 +00:00
}
}
2016-08-30 22:27:31 +00:00
ready := ac . ready
if ready == nil {
ready = make ( chan struct { } )
ac . ready = ready
}
ac . mu . Unlock ( )
select {
case <- ctx . Done ( ) :
return nil , toRPCErr ( ctx . Err ( ) )
// Wait until the new transport is ready or failed.
case <- ready :
}
2016-07-11 19:16:16 +00:00
}
}
2016-08-30 22:27:31 +00:00
// tearDown starts to tear down the addrConn.
2016-07-11 19:16:16 +00:00
// TODO(zhaoq): Make this synchronous to avoid unbounded memory consumption in
2016-08-30 22:27:31 +00:00
// some edge cases (e.g., the caller opens and closes many addrConn's in a
2016-07-11 19:16:16 +00:00
// tight loop.
2016-08-30 22:27:31 +00:00
// tearDown doesn't remove ac from ac.cc.conns.
func ( ac * addrConn ) tearDown ( err error ) {
ac . cancel ( )
ac . mu . Lock ( )
defer ac . mu . Unlock ( )
if ac . down != nil {
ac . down ( downErrorf ( false , false , "%v" , err ) )
ac . down = nil
}
if err == errConnDrain && ac . transport != nil {
// GracefulClose(...) may be executed multiple times when
// i) receiving multiple GoAway frames from the server; or
// ii) there are concurrent name resolver/Balancer triggered
// address removal and GoAway.
ac . transport . GracefulClose ( )
}
if ac . state == Shutdown {
return
}
ac . state = Shutdown
ac . tearDownErr = err
ac . stateCV . Broadcast ( )
if ac . events != nil {
ac . events . Finish ( )
ac . events = nil
}
if ac . ready != nil {
close ( ac . ready )
ac . ready = nil
}
if ac . transport != nil && err != errConnDrain {
ac . transport . Close ( )
}
return
2016-07-11 19:16:16 +00:00
}