task: WithExit() doesn't have to overwrite existing options

avagin · avagin · commit fc2fcf6b2a9d · 2018-07-20T13:19:05.000-07:00
Signed-off-by: Andrei Vagin &lt;avagin@openvz.org&gt;
diff --git a/cmd/ctr/commands/tasks/checkpoint.go b/cmd/ctr/commands/tasks/checkpoint.go
@@ -78,12 +78,22 @@ func withExit(rt string) containerd.CheckpointTaskOpts {
 	return func(r *containerd.CheckpointTaskInfo) error {
 		switch rt {
 		case "io.containerd.runc.v1":
-			r.Options = &options.CheckpointOptions{
-				Exit: true,
+			if r.Options == nil {
+				r.Options = &options.CheckpointOptions{
+					Exit: true,
+				}
+			} else {
+				opts, _ := r.Options.(*options.CheckpointOptions)
+				opts.Exit = true
 			}
 		default:
-			r.Options = &runctypes.CheckpointOptions{
-				Exit: true,
+			if r.Options == nil {
+				r.Options = &runctypes.CheckpointOptions{
+					Exit: true,
+				}
+			} else {
+				opts, _ := r.Options.(*runctypes.CheckpointOptions)
+				opts.Exit = true
 			}
 		}
 		return nil
