CyberCoveLLC/threshold
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove connection from worker if in terminated state and all conns are closed, not just in canceling state

Browse Source
This commit is contained in:
Jeff Mitchell committed 2020-10-01 20:49:13 -04:00
1 parent d1d94c19c8
commit 8615d21a1f
1 file changed
+4 -2
Download Patch File Download Diff File Expand all files Collapse all files
internal/servers/worker/status.go
+4 -2
View File
@@ -137,8 +137,10 @@ func (w *Worker) startStatusTicking(cancelCtx context.Context) {
w.sessionInfoMap.Range(func(key, value interface{}) bool { w.sessionInfoMap.Range(func(key, value interface{}) bool {
si := value.(*sessionInfo) si := value.(*sessionInfo)
si.Lock() si.Lock()
if time.Until(si.lookupSessionResponse.Expiration.AsTime()) < 0 || switch {
si.status == pbs.SESSIONSTATUS_SESSIONSTATUS_CANCELING { case si.status == pbs.SESSIONSTATUS_SESSIONSTATUS_CANCELING,
si.status == pbs.SESSIONSTATUS_SESSIONSTATUS_TERMINATED,
time.Until(si.lookupSessionResponse.Expiration.AsTime()) < 0:
var toClose int var toClose int
for k, v := range si.connInfoMap { for k, v := range si.connInfoMap {
if v.closeTime.IsZero() { if v.closeTime.IsZero() {