Improve Docker Compose container execution

internal/compose/compose.go

@@ -23,6 +23,8 @@ import (
 	"github.com/elastic/elastic-package/internal/signal"
 )
 
+const waitForHealthyTimeout = 10 * time.Minute
+
 // Project represents a Docker Compose project.
 type Project struct {
 	name             string
@@ -275,20 +277,23 @@ func (p *Project) WaitForHealthy(opts CommandOptions) error {
 		return err
 	}
 
+	timeout := time.Duration(waitForHealthyTimeout)
+	startTime := time.Now()
+	timeoutTime := startTime.Add(timeout)
+	interval := time.Duration(1 * time.Second)
+	healthy := true
+
 	containerIDs := strings.Split(strings.TrimSpace(b.String()), "\n")
-	for {
+	for time.Now().Before(timeoutTime) {
 		if signal.SIGINT() {
 			return errors.New("SIGINT: cancel waiting for policy assigned")
 		}
 
-		healthy := true
-
 		logger.Debugf("Wait for healthy containers: %s", strings.Join(containerIDs, ","))
 		descriptions, err := docker.InspectContainers(containerIDs...)
 		if err != nil {
 			return err
 		}
-
 		for _, containerDescription := range descriptions {
 			logger.Debugf("Container status: %s", containerDescription.String())
 
@@ -316,12 +321,19 @@ func (p *Project) WaitForHealthy(opts CommandOptions) error {
 			healthy = false
 		}
 
+		// end loop before timeout if healthy
 		if healthy {
 			break
 		}
 
-		time.Sleep(time.Second)
+		// NOTE: using sleep does not guarantee interval but it's ok for this use case
+		time.Sleep(interval)
 	}
+
+	if !healthy {
+		return errors.New("timeout waiting for healthy container")
+	}
+
 	return nil
 }