kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use time.AfterFunc for session inactivity timeout to reduce goroutine count

This commit is contained in:
Andy Wang 2020-12-05 21:38:16 +00:00
parent a3520c1018
commit c0040f20c3
No known key found for this signature in database
GPG Key ID: 181B49F9F38F3374
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/multiplex/session.go
View File

@ -111,7 +111,7 @@ func MakeSession(id uint32, config SessionConfig) *Session {
sesh.maxStreamUnitWrite = sesh.MsgOnWireSizeLimit - frameHeaderLength - sesh.Obfuscator.maxOverhead
sesh.sb = makeSwitchboard(sesh)
go sesh.timeoutAfter(sesh.InactivityTimeout)
time.AfterFunc(sesh.InactivityTimeout, sesh.checkTimeout)
return sesh
}
@ -204,7 +204,7 @@ func (sesh *Session) closeStream(s *Stream, active bool) error {
return sesh.Close()
} else {
log.Debugf("session %v has no active stream left", sesh.id)
go sesh.timeoutAfter(sesh.InactivityTimeout)
time.AfterFunc(sesh.InactivityTimeout, sesh.checkTimeout)
}
}
return nil
@ -334,9 +334,7 @@ func (sesh *Session) IsClosed() bool {
return atomic.LoadUint32(&sesh.closed) == 1
}
func (sesh *Session) timeoutAfter(to time.Duration) {
time.Sleep(to)
func (sesh *Session) checkTimeout() {
if sesh.streamCount() == 0 && !sesh.IsClosed() {
sesh.SetTerminalMsg("timeout")
sesh.Close()