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

优化RPC

Browse Source
This commit is contained in:
orgin
2022-10-27 15:38:28 +08:00
parent 1eab31209c
commit fd364cf579
5 changed files with 34 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
network/tcp_client.go
View File

@@ -56,11 +56,11 @@ func (client *TCPClient) init() {
}
if client.ReadDeadline == 0 {
client.ReadDeadline = 15*time.Second
log.SRelease("invalid ReadDeadline, reset to ", client.ReadDeadline,"s")
log.SRelease("invalid ReadDeadline, reset to ", int64(client.ReadDeadline.Seconds()),"s")
}
if client.WriteDeadline == 0 {
client.WriteDeadline = 15*time.Second
log.SRelease("invalid WriteDeadline, reset to ", client.WriteDeadline,"s")
log.SRelease("invalid WriteDeadline, reset to ", int64(client.WriteDeadline.Seconds()),"s")
}
if client.NewAgent == nil {
log.SFatal("NewAgent must not be nil")
@@ -79,6 +79,13 @@ func (client *TCPClient) init() {
client.msgParser = msgParser
}
func (client *TCPClient) GetCloseFlag() bool{
client.Lock()
defer client.Unlock()
return client.closeFlag
}
func (client *TCPClient) dial() net.Conn {
for {
conn, err := net.Dial("tcp", client.Addr)