From a031b83a09a8628435317a03f199cdc18b78262f Mon Sep 17 00:00:00 2001
From: Matthew Heon <matthew.heon@gmail.com>
Date: Wed, 1 Nov 2017 11:24:59 -0400
Subject: Initial checkin from CRI-O repo

Signed-off-by: Matthew Heon <matthew.heon@gmail.com>
---
 libkpod/wait.go | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 libkpod/wait.go

(limited to 'libkpod/wait.go')

diff --git a/libkpod/wait.go b/libkpod/wait.go
new file mode 100644
index 000000000..c7ba57328
--- /dev/null
+++ b/libkpod/wait.go
@@ -0,0 +1,42 @@
+package libkpod
+
+import (
+	"github.com/kubernetes-incubator/cri-o/oci"
+	"github.com/pkg/errors"
+	"k8s.io/apimachinery/pkg/util/wait"
+)
+
+func isStopped(c *ContainerServer, ctr *oci.Container) bool {
+	c.runtime.UpdateStatus(ctr)
+	cStatus := c.runtime.ContainerStatus(ctr)
+	if cStatus.Status == oci.ContainerStateStopped {
+		return true
+	}
+	return false
+}
+
+// ContainerWait stops a running container with a grace period (i.e., timeout).
+func (c *ContainerServer) ContainerWait(container string) (int32, error) {
+	ctr, err := c.LookupContainer(container)
+	if err != nil {
+		return 0, errors.Wrapf(err, "failed to find container %s", container)
+	}
+
+	err = wait.PollImmediateInfinite(1,
+		func() (bool, error) {
+			if !isStopped(c, ctr) {
+				return false, nil
+			} else { // nolint
+				return true, nil // nolint
+			} // nolint
+
+		},
+	)
+
+	if err != nil {
+		return 0, err
+	}
+	exitCode := ctr.State().ExitCode
+	c.ContainerStateToDisk(ctr)
+	return exitCode, nil
+}
-- 
cgit v1.2.3-54-g00ecf