fix: update container state according to containerd events

Signed-off-by: zhuangqh <zhuangqhc@gmail.com>

daemon/mgr/container.go

@@ -221,7 +221,7 @@ func NewContainerManager(ctx context.Context, store *meta.Store, cli ctrd.APICli
 
  mgr.Client.SetExitHooks(mgr.exitedAndRelease)
  mgr.Client.SetExecExitHooks(mgr.execExitedAndRelease)
- mgr.Client.SetEventsHooks(mgr.publishContainerdEvent)
+ mgr.Client.SetEventsHooks(mgr.publishContainerdEvent, mgr.updateContainerState)
 
  go mgr.execProcessGC()
 

daemon/mgr/container_state.go

@@ -140,6 +140,18 @@ func (c *Container) SetStatusUnpaused() {
  c.setStatusFlags(types.StatusRunning)
 }
 
+// SetStatusOOM sets a container to be status exit because of OOM.
+func (c *Container) SetStatusOOM() {
+ c.Lock()
+ defer c.Unlock()
+ c.State.OOMKilled = true
+ c.State.Status = types.StatusExited
+ c.State.Pid = 0
+ c.State.ExitCode = 137
+ c.State.Error = "OOMKilled"
+ c.setStatusFlags(types.StatusExited)
+}
+
 // Notes(ziren): i still feel uncomfortable for a function hasing no return
 // setStatusFlags set the specified status flag to true, and unset others
 func (c *Container) setStatusFlags(status types.Status) {

daemon/mgr/events.go

@@ -2,10 +2,13 @@ package mgr
 
 import (
  "context"
+ "strconv"
  "strings"
 
  "github.com/alibaba/pouch/apis/types"
+
  "github.com/docker/libnetwork"
+ "github.com/sirupsen/logrus"
 )
 
 // LogContainerEvent generates an event related to a container with only the default attributes.
@@ -109,3 +112,33 @@ func (mgr *ContainerManager) publishContainerdEvent(ctx context.Context, id, act
 
  return nil
 }
+
+// updateContainerState update container's state according to the containerd events.
+func (mgr *ContainerManager) updateContainerState(ctx context.Context, id, action string, attributes map[string]string) error {
+ c, err := mgr.container(id)
+ if err != nil {
+ return err
+ }
+
+ dirty := true
+ switch action {
+ case "die":
+ exitCode, err := strconv.ParseInt(attributes["exitCode"], 10, 64)
+ if err != nil {
+ logrus.Warnf("failed to parse exitCode: %v", err)
+ }
+ c.SetStatusExited(exitCode, "")
+ case "oom":
+ c.SetStatusOOM()
+ default:
+ dirty = false
+ }
+
+ if dirty {
+ if err := mgr.Store.Put(c); err != nil {
+ return err
+ }
+ }
+
+ return nil
+}