103 lines
2.4 KiB
Go
103 lines
2.4 KiB
Go
package supervisor
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/docker/containerd/runtime"
|
|
)
|
|
|
|
// Worker interface
|
|
type Worker interface {
|
|
Start()
|
|
}
|
|
|
|
type startTask struct {
|
|
Container runtime.Container
|
|
CheckpointPath string
|
|
Stdin string
|
|
Stdout string
|
|
Stderr string
|
|
Err chan error
|
|
StartResponse chan StartResponse
|
|
}
|
|
|
|
// NewWorker return a new initialized worker
|
|
func NewWorker(s *Supervisor, wg *sync.WaitGroup) Worker {
|
|
return &worker{
|
|
s: s,
|
|
wg: wg,
|
|
}
|
|
}
|
|
|
|
type worker struct {
|
|
wg *sync.WaitGroup
|
|
s *Supervisor
|
|
}
|
|
|
|
// Start runs a loop in charge of starting new containers
|
|
func (w *worker) Start() {
|
|
defer w.wg.Done()
|
|
for t := range w.s.startTasks {
|
|
process, err := t.Container.Start(t.CheckpointPath, runtime.NewStdio(t.Stdin, t.Stdout, t.Stderr))
|
|
if err != nil {
|
|
logrus.WithFields(logrus.Fields{
|
|
"error": err,
|
|
"id": t.Container.ID(),
|
|
}).Error("containerd: start container")
|
|
t.Err <- err
|
|
evt := &DeleteTask{
|
|
ID: t.Container.ID(),
|
|
NoEvent: true,
|
|
Process: process,
|
|
}
|
|
w.s.SendTask(evt)
|
|
continue
|
|
}
|
|
oom, err := t.Container.OOM()
|
|
if err != nil {
|
|
logrus.WithField("error", err).Error("containerd: get oom FD")
|
|
}
|
|
if err := w.s.monitor.Add(oom); err != nil && err != runtime.ErrContainerExited {
|
|
if process.State() != runtime.Stopped {
|
|
logrus.WithField("error", err).Error("containerd: notify OOM events")
|
|
}
|
|
}
|
|
if err := w.s.monitor.Add(process); err != nil {
|
|
logrus.WithField("error", err).Error("containerd: add process to monitor")
|
|
t.Err <- err
|
|
evt := &DeleteTask{
|
|
ID: t.Container.ID(),
|
|
NoEvent: true,
|
|
Process: process,
|
|
}
|
|
w.s.SendTask(evt)
|
|
continue
|
|
}
|
|
// only call process start if we aren't restoring from a checkpoint
|
|
// if we have restored from a checkpoint then the process is already started
|
|
if t.CheckpointPath == "" {
|
|
if err := process.Start(); err != nil {
|
|
logrus.WithField("error", err).Error("containerd: start init process")
|
|
t.Err <- err
|
|
evt := &DeleteTask{
|
|
ID: t.Container.ID(),
|
|
NoEvent: true,
|
|
Process: process,
|
|
}
|
|
w.s.SendTask(evt)
|
|
continue
|
|
}
|
|
}
|
|
t.Err <- nil
|
|
t.StartResponse <- StartResponse{
|
|
Container: t.Container,
|
|
}
|
|
w.s.notifySubscribers(Event{
|
|
Timestamp: time.Now(),
|
|
ID: t.Container.ID(),
|
|
Type: StateStart,
|
|
})
|
|
}
|
|
}
|