Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-03-06 22:27:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

新增判断连接状态接口

Browse Source
This commit is contained in:
duanhf2012
2020-04-08 14:43:30 +08:00
parent 89fd5b8525
commit a0b02c7ec2
3 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cluster/cluster.go
View File

@@ -145,3 +145,8 @@ func GetRpcClient(nodeId int,serviceMethod string) ([]*rpc.Client,error) {
func GetRpcServer() *rpc.Server{ func GetRpcServer() *rpc.Server{
return &cluster.rpcServer return &cluster.rpcServer
} }
func (slf *Cluster) IsNodeConnected (nodeId int) bool {
pClient := slf.GetRpcClient(nodeId)
return pClient!=nil && pClient.IsConnected()
}

4
network/tcp_conn.go
View File

@@ -111,3 +111,7 @@ func (tcpConn *TCPConn) ReadMsg() ([]byte, error) {
func (tcpConn *TCPConn) WriteMsg(args ...[]byte) error { func (tcpConn *TCPConn) WriteMsg(args ...[]byte) error {
return tcpConn.msgParser.Write(tcpConn, args...) return tcpConn.msgParser.Write(tcpConn, args...)
} }
func (tcpConn *TCPConn) IsConnected() bool {
return tcpConn.closeFlag == false
}

4
rpc/client.go
View File

@@ -219,3 +219,7 @@ func (slf *Client) Run(){
func (slf *Client) OnClose(){ func (slf *Client) OnClose(){
} }
func (slf *Client) IsConnected() bool {
return slf.conn!=nil && slf.conn.IsConnected()==true
}