-
Notifications
You must be signed in to change notification settings - Fork 0
/
container_checkpoint.go
35 lines (30 loc) · 1.05 KB
/
container_checkpoint.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package server
import (
"syscall"
"github.com/containerd/containerd"
"github.com/pkg/errors"
"golang.org/x/net/context"
runtime "k8s.io/cri-api/pkg/apis/runtime/v1alpha2"
)
func (c *criService) CheckpointContainer(ctx context.Context, r *runtime.CheckpointContainerRequest) (retRes *runtime.CheckpointContainerResponse, retErr error) {
cntr, err := c.containerStore.Get(r.GetContainerId())
if err != nil {
return nil, errors.Wrap(err, "failed to find container")
}
task, err := cntr.Container.Task(ctx, nil)
if err != nil {
return nil, errors.Wrap(err, "failed to checkpoint container task")
}
opts := []containerd.CheckpointTaskOpts{containerd.WithCheckpointImagePath(r.GetOptions().GetCheckpointPath())}
if !r.GetOptions().LeaveRunning {
opts = append(opts, containerd.WithCheckpointExit())
}
_, err = task.Checkpoint(ctx, opts...)
if err != nil {
return nil, errors.Wrap(err, "failed to checkpoint container")
}
if !r.GetOptions().GetLeaveRunning() {
task.Kill(ctx, syscall.SIGKILL)
}
return &runtime.CheckpointContainerResponse{}, nil
}