diff --git a/cli/command/container/logs.go b/cli/command/container/logs.go index 3d536f721b73..4ffe61dec309 100644 --- a/cli/command/container/logs.go +++ b/cli/command/container/logs.go @@ -3,11 +3,14 @@ package container import ( "context" "io" + "strconv" + "time" "github.com/docker/cli/cli" "github.com/docker/cli/cli/command" "github.com/docker/cli/cli/command/completion" "github.com/docker/docker/api/types/container" + "github.com/docker/docker/api/types/events" "github.com/docker/docker/pkg/stdcopy" "github.com/spf13/cobra" ) @@ -20,6 +23,8 @@ type logsOptions struct { details bool tail string + detachDelay int + container string } @@ -49,6 +54,7 @@ func NewLogsCommand(dockerCli command.Cli) *cobra.Command { flags.BoolVarP(&opts.timestamps, "timestamps", "t", false, "Show timestamps") flags.BoolVar(&opts.details, "details", false, "Show extra details provided to logs") flags.StringVarP(&opts.tail, "tail", "n", "all", "Number of lines to show from the end of the logs") + flags.IntVarP(&opts.detachDelay, "delay", "d", 0, "Number of seconds to wait for container restart before exiting") return cmd } @@ -58,6 +64,45 @@ func runLogs(ctx context.Context, dockerCli command.Cli, opts *logsOptions) erro return err } + since := opts.since + for { + restarting := make(chan events.Message) + if opts.detachDelay != 0 { + go func() { + eventsCtx, eventsCtxCancel := context.WithCancel(ctx) + eventC, eventErrs := dockerCli.Client().Events(eventsCtx, events.ListOptions{}) + defer eventsCtxCancel() + for { + select { + case event := <-eventC: + if event.Action == events.ActionRestart && event.Actor.ID == c.ID { + restarting <- event + return + } + case <-eventErrs: + return + } + } + }() + } + + opts.since = since + err = streamLogs(ctx, dockerCli, opts, c) + + if opts.detachDelay == 0 { + return err + } + + select { + case restartEvent := <-restarting: + since = strconv.FormatInt(restartEvent.Time, 10) + case <-time.After(time.Duration(opts.detachDelay) * time.Second): + return err + } + } +} + +func streamLogs(ctx context.Context, dockerCli command.Cli, opts *logsOptions, c container.InspectResponse) error { responseBody, err := dockerCli.Client().ContainerLogs(ctx, c.ID, container.LogsOptions{ ShowStdout: true, ShowStderr: true, @@ -78,5 +123,6 @@ func runLogs(ctx context.Context, dockerCli command.Cli, opts *logsOptions) erro } else { _, err = stdcopy.StdCopy(dockerCli.Out(), dockerCli.Err(), responseBody) } + return err }