grpc 版本1.50
client.go 代码：
func main() {
flag.Parse()
// Set up a connection to the server.
conn, err := grpc.Dial(*addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
log.Fatalf("did not connect: %v", err)
}
defer conn.Close()
c := pb.NewGreeterClient(conn)
// Contact the server and print out its response.
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
r, err := c.SayHello(ctx, &pb.HelloRequest{Name: *name})
if err != nil {
log.Fatalf("could not greet: %v", err)
}
log.Printf("Greeting: %s", r.GetMessage())
}Dial 源码：
func Dial(target string, opts ...DialOption) (*ClientConn, error) {
return DialContext(context.Background(), target, opts...)
}DialContext 源码：
省略次部分代码
// 首先会创建 ClientConn，初始化相关字段
func DialContext(ctx context.Context, target string, opts ...DialOption) (conn *ClientConn, err error) {
cc := &ClientConn{
target: target,
csMgr: &connectivityStateManager{},
conns: make(map[*addrConn]struct{}),
dopts: defaultDialOptions(),
blockingpicker: newPickerWrapper(),
czData: new(channelzData),
firstResolveEvent: grpcsync.NewEvent(),
}
// 将用户设置的连接参数更新到客户端连接器 ClientConn
for _, opt := range opts {
opt.apply(&cc.dopts)
}
return cc, nil
}connect() 方法：
func (ac *addrConn) connect() error {
ac.mu.Lock()
// if 校验状态
if ac.state == connectivity.Shutdown {
ac.mu.Unlock()
return errConnClosing
}
if ac.state != connectivity.Idle {
ac.mu.Unlock()
return nil
}
ac.updateConnectivityState(connectivity.Connecting, nil)
ac.mu.Unlock()
// 主要看这个方法，重试连接
ac.resetTransport()
return nil
}进入 resetTransport() 源码
func (ac *addrConn) resetTransport() {
ac.mu.Lock()
// 判断状态若为 shutdown，则不再连接直接推出
if ac.state == connectivity.Shutdown {
ac.mu.Unlock()
return
}
addrs := ac.addrs
// 连接失败，需要进行重试的
// Backoff 是需要等待的时间，ac.backoffIdx表示第几次重试
backoffFor := ac.dopts.bs.Backoff(ac.backoffIdx)
// 计算出本次向gRPC服务尝试建立TCP连接的最长时间
// 若超过这个时间还是连接不上，则主动断开，等待尝试下次连接
// This will be the duration that dial gets to finish.
dialDuration := minConnectTimeout
if ac.dopts.minConnectTimeout != nil {
dialDuration = ac.dopts.minConnectTimeout()
}
if dialDuration < backoffFor {
// Give dial more time as we keep failing to connect.
dialDuration = backoffFor
}
connectDeadline := time.Now().Add(dialDuration)
// 更新结构addrConn状态为connecting
ac.updateConnectivityState(connectivity.Connecting, nil)
ac.mu.Unlock()
// 向服务器连接失败后需要做的逻辑
if err := ac.tryAllAddrs(addrs, connectDeadline); err != nil {
ac.cc.resolveNow(resolver.ResolveNowOptions{})
// After exhausting all addresses, the addrConn enters
// TRANSIENT_FAILURE.
ac.mu.Lock()
if ac.state == connectivity.Shutdown {
ac.mu.Unlock()
return
}
ac.updateConnectivityState(connectivity.TransientFailure, err)
// Backoff.
b := ac.resetBackoff
ac.mu.Unlock()
// 定时的超时间
timer := time.NewTimer(backoffFor)
// 1.timer.C如果连接超时，重试的次数+1，继续下一次重试连接，但需要等待一段时间
// 2. b,直接关闭，将继续进行重新连接
// 2. ac.ctx.Done，走这里的话，上下文结束，这里不会再次重试了
select {
case <-timer.C:
ac.mu.Lock()
ac.backoffIdx++
ac.mu.Unlock()
case <-b:
timer.Stop()
case <-ac.ctx.Done():
timer.Stop()
return
}
ac.mu.Lock()
// 状态 != shutdown就更新为空闲状态
if ac.state != connectivity.Shutdown {
ac.updateConnectivityState(connectivity.Idle, err)
}
ac.mu.Unlock()
return
}
// 连接成功，重新设置backoff为原始值0
ac.mu.Lock()
ac.backoffIdx = 0
ac.mu.Unlock()
}如何计算重试连接等待时间？
进入Backoff方法
func (bc Exponential) Backoff(retries int) time.Duration {
if retries == 0 {
return bc.Config.BaseDelay
}
backoff, max := float64(bc.Config.BaseDelay), float64(bc.Config.MaxDelay)
for backoff < max && retries > 0 {
// 幂次方
backoff *= bc.Config.Multiplier
retries--
}
// 不能超过最大延时时间
if backoff > max {
backoff = max
}
// Randomize backoff delays so that if a cluster of requests start at
// the same time, they won't operate in lockstep.
backoff *= 1 + bc.Config.Jitter*(grpcrand.Float64()*2-1)
if backoff < 0 {
return 0
}
return time.Duration(backoff)
}总结：
- 连接失败后，客户端会进行重试连接。
- 重试次数越多，等待下一次连接时间也会变长，但不能超过MaxDelay值。
更多Go云原生学习资料，收录于Github：https://github.com/metashops/GoFamily
本文由mdnice多平台发布