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

集群服务设置中增加性能设置

Browse Source
This commit is contained in:
刘祥超
2023-01-01 19:27:38 +08:00
parent c42ff1e1e9
commit b4a4b2e9b1
1 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
internal/nodes/client_conn.go
View File

@@ -67,13 +67,17 @@ func NewClientConn(rawConn net.Conn, isTLS bool, quickClose bool, isInAllowList
}
func (this *ClientConn) Read(b []byte) (n int, err error) {
this.lastReadAt = time.Now().Unix()
var globalServerConfig = sharedNodeConfig.GlobalServerConfig
defer func() {
if err != nil {
this.lastErr = errors.New("read error: " + err.Error())
}
}()
if globalServerConfig != nil && globalServerConfig.Performance.Debug {
this.lastReadAt = time.Now().Unix()
defer func() {
if err != nil {
this.lastErr = errors.New("read error: " + err.Error())
}
}()
}
// 环路直接读取
if this.isLO {
@@ -122,21 +126,27 @@ func (this *ClientConn) Read(b []byte) (n int, err error) {
}
func (this *ClientConn) Write(b []byte) (n int, err error) {
this.lastWriteAt = time.Now().Unix()
var globalServerConfig = sharedNodeConfig.GlobalServerConfig
defer func() {
if err != nil {
this.lastErr = errors.New("write error: " + err.Error())
}
}()
if globalServerConfig != nil && globalServerConfig.Performance.Debug {
this.lastWriteAt = time.Now().Unix()
defer func() {
if err != nil {
this.lastErr = errors.New("write error: " + err.Error())
}
}()
}
// 设置超时时间
// TODO L2 -> L1 写入时不限制时间
var timeoutSeconds = len(b) / 4096
if timeoutSeconds < 3 {
timeoutSeconds = 3
if globalServerConfig != nil && globalServerConfig.Performance.AutoWriteTimeout {
// TODO L2 -> L1 写入时不限制时间
var timeoutSeconds = len(b) / 4096
if timeoutSeconds < 3 {
timeoutSeconds = 3
}
_ = this.rawConn.SetWriteDeadline(time.Now().Add(time.Duration(timeoutSeconds) * time.Second)) // TODO 时间可以设置
}
_ = this.rawConn.SetWriteDeadline(time.Now().Add(time.Duration(timeoutSeconds) * time.Second)) // TODO 时间可以设置
n, err = this.rawConn.Write(b)
if n > 0 {