bug: prevent go routine leakage due to existing DeferCheck

hashicorp · Aug 22, 2023 · 06d4c72 · 06d4c72
1 parent 5b88aae
commit 06d4c72
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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