TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-10 04:20:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
刘祥超
2022-11-17 10:38:20 +08:00
parent c209ab912f
commit 06ec4d3fba
1 changed files with 1 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/rpc/rpc_client.go
View File

@@ -218,12 +218,12 @@ func (this *RPCClient) pickConn() *grpc.ClientConn {
// 检查连接状态 // 检查连接状态
if len(this.conns) > 0 { if len(this.conns) > 0 {
var availableConns = []*grpc.ClientConn{}
for _, stateArray := range [][2]connectivity.State{ for _, stateArray := range [][2]connectivity.State{
{connectivity.Ready, connectivity.Idle}, // 优先Ready和Idle {connectivity.Ready, connectivity.Idle}, // 优先Ready和Idle
{connectivity.Connecting, connectivity.Connecting}, {connectivity.Connecting, connectivity.Connecting},
{connectivity.TransientFailure, connectivity.TransientFailure}, {connectivity.TransientFailure, connectivity.TransientFailure},
} { } {
var availableConns = []*grpc.ClientConn{}
for _, conn := range this.conns { for _, conn := range this.conns {
var state = conn.GetState() var state = conn.GetState()
if state == stateArray[0] || state == stateArray[1] { if state == stateArray[0] || state == stateArray[1] {
@@ -234,21 +234,6 @@ func (this *RPCClient) pickConn() *grpc.ClientConn {
return this.randConn(availableConns) return this.randConn(availableConns)
} }
} }
if len(availableConns) > 0 {
return this.randConn(availableConns)
}
}
// 重新初始化
err := this.init()
if err != nil {
// 错误提示已经在构造对象时打印过，所以这里不再重复打印
return nil
}
if len(this.conns) == 0 {
return nil
} }
return this.randConn(this.conns) return this.randConn(this.conns)