Function RunTask

internal/integtest/containerd.go:33–71

RunTask is a utility function for running a task and returning the result.

(ctx context.Context, c containerd.Container)

// RunTask is a utility function for running a task and returning the result.
func RunTask(ctx context.Context, c containerd.Container) (*CommandResult, error) {
	var stdout bytes.Buffer
	var stderr bytes.Buffer

	task, err := c.NewTask(ctx, cio.NewCreator(cio.WithStreams(nil, &stdout, &stderr)))
	if err != nil {
		return nil, err
	}

	exitCh, err := task.Wait(ctx)
	if err != nil {
		return nil, err
	}

	err = task.Start(ctx)
	if err != nil {
		return nil, err
	}

	select {
	case exitStatus := <-exitCh:
		if err := exitStatus.Error(); err != nil {
			return nil, err
		}

		_, err := task.Delete(ctx)
		if err != nil {
			return nil, err
		}

		return &CommandResult{
			Stdout:   stdout.String(),
			Stderr:   stderr.String(),
			ExitCode: exitStatus.ExitCode(),
		}, nil
	case <-ctx.Done():
		return nil, ctx.Err()
	}
}