feat(compose): emit and handle container healthy events

Signed-off-by: Mark Jagyo <mark.jagyo@gmail.com>

Author: mjagyo

pkg/api/api.go

@@ -746,6 +746,8 @@ const (
 	ContainerEventRecreated
 	// ContainerEventExited is a ContainerEvent of type exit. ExitCode is set
 	ContainerEventExited
+	// ContainerEventHealthy let consumer know container is healthy
+	ContainerEventHealthy
 	// UserCancel user canceled compose up, we are stopping containers
 	HookEventLog
 )

pkg/compose/monitor.go

@@ -164,6 +164,11 @@ func (c *monitor) Start(ctx context.Context) error {
 					}
 					containers.Remove(ctr.ID)
 				}
+			case events.ActionHealthStatusHealthy:
+				logrus.Debugf("container %s healthy", ctr.Name)
+				for _, listener := range c.listeners {
+					listener(newContainerEvent(event.TimeNano, ctr, api.ContainerEventHealthy))
+				}
 			}
 		}
 	}

pkg/compose/start.go

@@ -95,16 +95,6 @@ func (s *composeService) start(ctx context.Context, projectName string, options
 			}
 			return err
 		}
-
-		if options.Log {
-			s.Logs(ctx, projectName, options.Attach, api.LogOptions{
-				Project:    options.Project,
-				Services:   options.Services,
-				Follow:     false,
-				Tail:       "all",
-				Timestamps: false,
-			})
-		}
 	}
 
 	return nil

pkg/compose/up.go

@@ -198,6 +198,11 @@ func (s *composeService) Up(ctx context.Context, project *types.Project, options
 	}
 	monitor.withListener(printer.HandleEvent)
 
+	monitor.withListener(onHealthy(func(e api.ContainerEvent) {
+		fmt.Println("Healthy:", e.Service, e.Container.Name)
+		cancel()
+	}))
+
 	var exitCode int
 	if options.Start.OnExit != api.CascadeIgnore {
 		once := true
@@ -301,3 +306,11 @@ func (s *composeService) Up(ctx context.Context, project *types.Project, options
 	}
 	return err
 }
+
+func onHealthy(fn func(api.ContainerEvent)) api.ContainerEventListener {
+	return func(e api.ContainerEvent) {
+		if e.Type == api.ContainerEventHealthy {
+			fn(e)
+		}
+	}
+}