81 lines
1.4 KiB
Go
81 lines
1.4 KiB
Go
package supervisor
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/docker/containerd/runtime"
|
|
)
|
|
|
|
type UpdateTask struct {
|
|
s *Supervisor
|
|
}
|
|
|
|
func (h *UpdateTask) Handle(e *Task) error {
|
|
i, ok := h.s.containers[e.ID]
|
|
if !ok {
|
|
return ErrContainerNotFound
|
|
}
|
|
container := i.container
|
|
if e.State != "" {
|
|
switch e.State {
|
|
case runtime.Running:
|
|
if err := container.Resume(); err != nil {
|
|
return ErrUnknownContainerStatus
|
|
}
|
|
h.s.notifySubscribers(Event{
|
|
ID: e.ID,
|
|
Type: "resume",
|
|
Timestamp: time.Now(),
|
|
})
|
|
case runtime.Paused:
|
|
if err := container.Pause(); err != nil {
|
|
return ErrUnknownContainerStatus
|
|
}
|
|
h.s.notifySubscribers(Event{
|
|
ID: e.ID,
|
|
Type: "pause",
|
|
Timestamp: time.Now(),
|
|
})
|
|
default:
|
|
return ErrUnknownContainerStatus
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type UpdateProcessTask struct {
|
|
s *Supervisor
|
|
}
|
|
|
|
func (h *UpdateProcessTask) Handle(e *Task) error {
|
|
i, ok := h.s.containers[e.ID]
|
|
if !ok {
|
|
return ErrContainerNotFound
|
|
}
|
|
processes, err := i.container.Processes()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var process runtime.Process
|
|
for _, p := range processes {
|
|
if p.ID() == e.Pid {
|
|
process = p
|
|
break
|
|
}
|
|
}
|
|
if process == nil {
|
|
return ErrProcessNotFound
|
|
}
|
|
if e.CloseStdin {
|
|
if err := process.CloseStdin(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if e.Width > 0 || e.Height > 0 {
|
|
if err := process.Resize(e.Width, e.Height); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|