mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2025-09-20 04:29:23 +08:00
recreate grpc connections if too many errors
address https://github.com/chrislusf/seaweedfs/issues/2098
This commit is contained in:
@@ -31,7 +31,8 @@ var (
|
|||||||
|
|
||||||
type versionedGrpcClient struct {
|
type versionedGrpcClient struct {
|
||||||
*grpc.ClientConn
|
*grpc.ClientConn
|
||||||
version int
|
version int
|
||||||
|
errCount int
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
@@ -103,6 +104,7 @@ func getOrCreateConnection(address string, opts ...grpc.DialOption) (*versionedG
|
|||||||
vgc := &versionedGrpcClient{
|
vgc := &versionedGrpcClient{
|
||||||
grpcConnection,
|
grpcConnection,
|
||||||
rand.Int(),
|
rand.Int(),
|
||||||
|
0,
|
||||||
}
|
}
|
||||||
grpcClients[address] = vgc
|
grpcClients[address] = vgc
|
||||||
|
|
||||||
@@ -116,15 +118,20 @@ func WithCachedGrpcClient(fn func(*grpc.ClientConn) error, address string, opts
|
|||||||
return fmt.Errorf("getOrCreateConnection %s: %v", address, err)
|
return fmt.Errorf("getOrCreateConnection %s: %v", address, err)
|
||||||
}
|
}
|
||||||
executionErr := fn(vgc.ClientConn)
|
executionErr := fn(vgc.ClientConn)
|
||||||
if executionErr != nil && strings.Contains(executionErr.Error(), "transport") {
|
if executionErr != nil {
|
||||||
grpcClientsLock.Lock()
|
vgc.errCount++
|
||||||
if t, ok := grpcClients[address]; ok {
|
if vgc.errCount > 3 ||
|
||||||
if t.version == vgc.version {
|
strings.Contains(executionErr.Error(), "transport") ||
|
||||||
vgc.Close()
|
strings.Contains(executionErr.Error(), "connection closed") {
|
||||||
delete(grpcClients, address)
|
grpcClientsLock.Lock()
|
||||||
|
if t, ok := grpcClients[address]; ok {
|
||||||
|
if t.version == vgc.version {
|
||||||
|
vgc.Close()
|
||||||
|
delete(grpcClients, address)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
grpcClientsLock.Unlock()
|
||||||
}
|
}
|
||||||
grpcClientsLock.Unlock()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return executionErr
|
return executionErr
|
||||||
|
Reference in New Issue
Block a user