grpc pool should check state (#1435)
Co-authored-by: huangshaojie <huangshaojie@corp.netease.com>
This commit is contained in:
parent
e4acc63d5f
commit
8282e781e4
@ -5,6 +5,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
|
"google.golang.org/grpc/connectivity"
|
||||||
)
|
)
|
||||||
|
|
||||||
type pool struct {
|
type pool struct {
|
||||||
@ -77,6 +78,32 @@ func (p *pool) getConn(addr string, opts ...grpc.DialOption) (*poolConn, error)
|
|||||||
// otherwise we'll create a new conn
|
// otherwise we'll create a new conn
|
||||||
conn := sp.head.next
|
conn := sp.head.next
|
||||||
for conn != nil {
|
for conn != nil {
|
||||||
|
// check conn state
|
||||||
|
// https://github.com/grpc/grpc/blob/master/doc/connectivity-semantics-and-api.md
|
||||||
|
switch conn.GetState() {
|
||||||
|
case connectivity.Connecting:
|
||||||
|
conn = conn.next
|
||||||
|
continue
|
||||||
|
case connectivity.Shutdown:
|
||||||
|
next := conn.next
|
||||||
|
if conn.streams == 0 {
|
||||||
|
removeConn(conn)
|
||||||
|
sp.idle--
|
||||||
|
}
|
||||||
|
conn = next
|
||||||
|
continue
|
||||||
|
case connectivity.TransientFailure:
|
||||||
|
next := conn.next
|
||||||
|
if conn.streams == 0 {
|
||||||
|
removeConn(conn)
|
||||||
|
conn.ClientConn.Close()
|
||||||
|
sp.idle--
|
||||||
|
}
|
||||||
|
conn = next
|
||||||
|
continue
|
||||||
|
case connectivity.Ready:
|
||||||
|
case connectivity.Idle:
|
||||||
|
}
|
||||||
// a old conn
|
// a old conn
|
||||||
if now-conn.created > p.ttl {
|
if now-conn.created > p.ttl {
|
||||||
next := conn.next
|
next := conn.next
|
||||||
|
Loading…
Reference in New Issue
Block a user