bug: prevent go routine leakage due to existing DeferCheck

1 year ago · 06d4c72f68
1 changed files with 6 additions and 0 deletions
--- a/agent/local/state.go
+++ b/agent/local/state.go
@ -838,6 +838,12 @@ func (l *State) setCheckStateLocked(c *CheckState) {
 	existing := l.checks[id]
 	if existing != nil {
 		c.InSync = c.Check.IsSame(existing.Check)
+		// If the existing check has a Defercheck, it needs to be
+		// assigned to the new check
+		if existing.DeferCheck != nil && c.DeferCheck == nil {
+			c.DeferCheck = existing.DeferCheck
+			c.InSync = false
+		}
 	}

 	l.checks[id] = c