From 52ea0deee64c7466635452c62354604bdc7b48b4 Mon Sep 17 00:00:00 2001 From: baude Date: Fri, 24 Nov 2017 09:08:59 -0600 Subject: kpod_wait Convert to libpod container backend Signed-off-by: baude Closes: #70 Approved by: rhatdan --- libpod/container.go | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) (limited to 'libpod') diff --git a/libpod/container.go b/libpod/container.go index 8bd1a0abf..9ec0fb121 100644 --- a/libpod/container.go +++ b/libpod/container.go @@ -19,6 +19,7 @@ import ( crioAnnotations "github.com/projectatomic/libpod/pkg/annotations" "github.com/sirupsen/logrus" "github.com/ulule/deepcopier" + "k8s.io/apimachinery/pkg/util/wait" "k8s.io/client-go/tools/remotecommand" ) @@ -639,3 +640,35 @@ func (c *Container) Export(path string) error { func (c *Container) Commit() (*storage.Image, error) { return nil, ErrNotImplemented } + +// Wait blocks on a container to exit and returns its exit code +func (c *Container) Wait() (int32, error) { + err := wait.PollImmediateInfinite(1, + func() (bool, error) { + stopped, err := c.isStopped() + if err != nil { + return false, err + } + if !stopped { + return false, nil + } else { // nolint + return true, nil // nolint + } // nolint + }, + ) + if err != nil { + return 0, err + } + exitCode := c.state.ExitCode + return exitCode, nil +} + +func (c *Container) isStopped() (bool, error) { + c.lock.Lock() + defer c.lock.Unlock() + err := c.syncContainer() + if err != nil { + return true, err + } + return c.state.State == ContainerStateStopped, nil +} -- cgit v1.2.3-54-g00ecf