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

在监控系统运行时上报API连接状况

Browse Source
This commit is contained in:
GoEdgeLab
2022-11-22 11:23:39 +08:00
parent 2c4c64d0f5
commit 33dd1e9305
4 changed files with 102 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/rpc/rpc_client.go
View File

@@ -287,14 +287,15 @@ func (this *RPCClient) pickConn() *grpc.ClientConn {
func (this *RPCClient) Invoke(ctx context.Context, method string, args interface{}, reply interface{}, opts ...grpc.CallOption) error {
var conn = this.pickConn()
if conn == nil {
return errors.New("can not get available grpc connection")
return errors.New("could not get available grpc connection")
}
return conn.Invoke(ctx, method, args, reply, opts...)
}
func (this *RPCClient) NewStream(ctx context.Context, desc *grpc.StreamDesc, method string, opts ...grpc.CallOption) (grpc.ClientStream, error) {
var conn = this.pickConn()
if conn == nil {
return nil, errors.New("can not get available grpc connection")
return nil, errors.New("could not get available grpc connection")
}
return conn.NewStream(ctx, desc, method, opts...)
}