diff --git a/controller/cmd/controller/main.go b/controller/cmd/controller/main.go index efe032f..97d2ade 100644 --- a/controller/cmd/controller/main.go +++ b/controller/cmd/controller/main.go @@ -222,7 +222,7 @@ func main() { sched.Every("stack-scan", 2*time.Minute, func(ctx context.Context) error { return stackMgr.ScanStacks() }) - sched.Every("health-probes", 1*time.Minute, func(ctx context.Context) error { + sched.Every("health-probes", 10*time.Second, func(ctx context.Context) error { return stackMgr.RunHealthProbes() }) diff --git a/controller/internal/stacks/healthprobe.go b/controller/internal/stacks/healthprobe.go index d6cf6dd..97a8218 100644 --- a/controller/internal/stacks/healthprobe.go +++ b/controller/internal/stacks/healthprobe.go @@ -33,10 +33,18 @@ func (m *Manager) RunHealthProbes() error { continue } - // Check if interval has elapsed since last probe + // Check if interval has elapsed since last probe. + // Fast 10s probes until healthy, then normal interval (default 5m). + // When HealthProbe is nil (just started/restarted), probe immediately. interval := parseInterval(hc.Interval) - if stack.HealthProbe != nil && time.Since(stack.HealthProbe.LastCheck) < interval { - continue + if stack.HealthProbe != nil { + effectiveInterval := interval + if !stack.HealthProbe.Healthy { + effectiveInterval = 10 * time.Second + } + if time.Since(stack.HealthProbe.LastCheck) < effectiveInterval { + continue + } } // Find the main container to probe (matching stack name) diff --git a/controller/internal/stacks/manager.go b/controller/internal/stacks/manager.go index ba6cbf6..c52b54b 100644 --- a/controller/internal/stacks/manager.go +++ b/controller/internal/stacks/manager.go @@ -576,6 +576,15 @@ func (m *Manager) StartStack(name string) error { m.logger.Printf("[INFO] Stack %s started successfully (took %.1fs)", name, time.Since(start).Seconds()) m.logPostStartStatus(name, dir, env) + + // Clear stale health probe so refreshStatus won't re-apply an old unhealthy override. + // The next health-probes tick (≤10s) will run a fresh probe. + m.mu.Lock() + if s, ok := m.stacks[name]; ok { + s.HealthProbe = nil + } + m.mu.Unlock() + return m.RefreshStatus() } @@ -624,6 +633,14 @@ func (m *Manager) RestartStack(name string) error { m.logger.Printf("[INFO] Stack %s restarted successfully (took %.1fs)", name, time.Since(start).Seconds()) m.logPostStartStatus(name, dir, env) + + // Clear stale health probe so refreshStatus won't re-apply an old unhealthy override. + m.mu.Lock() + if s, ok := m.stacks[name]; ok { + s.HealthProbe = nil + } + m.mu.Unlock() + return m.RefreshStatus() }