Update checkpoint/restore to match changes in latest docker.

Docker-DCO-1.1-Signed-off-by: Ross Boucher <rboucher@gmail.com> (github: boucher)
boucher · Aug 11, 2015 · 59c375a · 59c375a
1 parent 39ac46b
commit 59c375a
Show file tree

Hide file tree

Showing 10 changed files with 30 additions and 46 deletions.
diff --git a/api/client/checkpoint.go b/api/client/checkpoint.go
@@ -5,12 +5,13 @@ package client
 import (
 	"fmt"
 
+	Cli "github.com/docker/docker/cli"
 	flag "github.com/docker/docker/pkg/mflag"
 	"github.com/docker/docker/runconfig"
 )
 
 func (cli *DockerCli) CmdCheckpoint(args ...string) error {
-	cmd := cli.Subcmd("checkpoint", []string{"CONTAINER [CONTAINER...]"}, "Checkpoint one or more running containers", true)
+	cmd := Cli.Subcmd("checkpoint", []string{"CONTAINER [CONTAINER...]"}, "Checkpoint one or more running containers", true)
 	cmd.Require(flag.Min, 1)
 
 	var (

diff --git a/api/client/restore.go b/api/client/restore.go
@@ -5,12 +5,13 @@ package client
 import (
 	"fmt"
 
+	Cli "github.com/docker/docker/cli"
 	flag "github.com/docker/docker/pkg/mflag"
 	"github.com/docker/docker/runconfig"
 )
 
 func (cli *DockerCli) CmdRestore(args ...string) error {
-	cmd := cli.Subcmd("restore", []string{"CONTAINER [CONTAINER...]"}, "Restore one or more checkpointed containers", true)
+	cmd := Cli.Subcmd("restore", []string{"CONTAINER [CONTAINER...]"}, "Restore one or more checkpointed containers", true)
 	cmd.Require(flag.Min, 1)
 
 	var (

diff --git a/api/server/server_experimental_unix.go b/api/server/server_experimental_unix.go
@@ -5,12 +5,12 @@ package server
 import (
 	"encoding/json"
 	"fmt"
-	"net/http"
 	"github.com/docker/docker/pkg/version"
 	"github.com/docker/docker/runconfig"
-	)
+	"net/http"
+)
 
-func addExperimentalRoutes(s *Server, m map[string]map[string]HttpApiFunc) {
+func addExperimentalRoutes(s *Server, m map[string]map[string]HTTPAPIFunc) {
 	m["POST"]["/containers/{name:.*}/checkpoint"] = s.postContainersCheckpoint
 	m["POST"]["/containers/{name:.*}/restore"] = s.postContainersRestore
 }

diff --git a/api/server/server_stub.go b/api/server/server_stub.go
@@ -2,7 +2,7 @@
 
 package server
 
-func addExperimentalRoutes(s *Server, m map[string]map[string]HttpApiFunc) {
+func addExperimentalRoutes(s *Server, m map[string]map[string]HTTPAPIFunc) {
 
 }
 

diff --git a/api/types/types.go b/api/types/types.go
@@ -228,16 +228,16 @@ type ExecStartCheck struct {
 // it's part of ContainerJSONBase and will return by "inspect" command
 type ContainerState struct {
 	Running        bool
-	Paused    	   bool
+	Paused         bool
 	Checkpointed   bool
-	Restarting 	   bool
-	OOMKilled  	   bool
-	Dead       	   bool
-	Pid        	   int
-	ExitCode   	   int
-	Error      	   string
-	StartedAt  	   string
-	FinishedAt 	   string
+	Restarting     bool
+	OOMKilled      bool
+	Dead           bool
+	Pid            int
+	ExitCode       int
+	Error          string
+	StartedAt      string
+	FinishedAt     string
 	CheckpointedAt string `json:"-"`
 }
 

diff --git a/daemon/execdriver/lxc/driver.go b/daemon/execdriver/lxc/driver.go
@@ -561,11 +561,11 @@ func (d *Driver) Unpause(c *execdriver.Command) error {
 	return err
 }
 
-func (d *driver) Checkpoint(c *execdriver.Command, opts *runconfig.CriuConfig) error {
+func (d *Driver) Checkpoint(c *execdriver.Command, opts *runconfig.CriuConfig) error {
 	return fmt.Errorf("Checkpointing lxc containers not supported yet\n")
 }
 
-func (d *driver) Restore(c *execdriver.Command, pipes *execdriver.Pipes, restoreCallback execdriver.RestoreCallback, opts *runconfig.CriuConfig, forceRestore bool) (execdriver.ExitStatus, error) {
+func (d *Driver) Restore(c *execdriver.Command, pipes *execdriver.Pipes, restoreCallback execdriver.RestoreCallback, opts *runconfig.CriuConfig, forceRestore bool) (execdriver.ExitStatus, error) {
 	return execdriver.ExitStatus{ExitCode: 0}, fmt.Errorf("Restoring lxc containers not supported yet\n")
 }
 

diff --git a/daemon/execdriver/native/create.go b/daemon/execdriver/native/create.go
@@ -113,24 +113,6 @@ func generateIfaceName() (string, error) {
 	return "", errors.New("Failed to find name for new interface")
 }
 
-// Re-create the container type from the image that was saved during checkpoint.
-func (d *driver) createRestoreContainer(c *execdriver.Command, imageDir string) (*libcontainer.Config, error) {
-	// Read the container.json.
-	f1, err := os.Open(filepath.Join(imageDir, "container.json"))
-	if err != nil {
-		return nil, err
-	}
-	defer f1.Close()
-
-	var container *libcontainer.Config
-	err = json.NewDecoder(f1).Decode(&container)
-	if err != nil {
-		return nil, err
-	}
-
-	return container, nil
-}
-
 func (d *Driver) createNetwork(container *configs.Config, c *execdriver.Command) error {
 	if c.Network == nil {
 		return nil

diff --git a/daemon/execdriver/native/driver.go b/daemon/execdriver/native/driver.go
@@ -313,7 +313,7 @@ func libcontainerCriuOpts(runconfigOpts *runconfig.CriuConfig) *libcontainer.Cri
 	}
 }
 
-func (d *driver) Checkpoint(c *execdriver.Command, opts *runconfig.CriuConfig) error {
+func (d *Driver) Checkpoint(c *execdriver.Command, opts *runconfig.CriuConfig) error {
 	active := d.activeContainers[c.ID]
 	if active == nil {
 		return fmt.Errorf("active container for %s does not exist", c.ID)
@@ -329,7 +329,7 @@ func (d *driver) Checkpoint(c *execdriver.Command, opts *runconfig.CriuConfig) e
 	return nil
 }
 
-func (d *driver) Restore(c *execdriver.Command, pipes *execdriver.Pipes, restoreCallback execdriver.RestoreCallback, opts *runconfig.CriuConfig, forceRestore bool) (execdriver.ExitStatus, error) {
+func (d *Driver) Restore(c *execdriver.Command, pipes *execdriver.Pipes, restoreCallback execdriver.RestoreCallback, opts *runconfig.CriuConfig, forceRestore bool) (execdriver.ExitStatus, error) {
 	var (
 		cont libcontainer.Container
 		err  error

diff --git a/daemon/inspect.go b/daemon/inspect.go
@@ -45,14 +45,14 @@ func (daemon *Daemon) getInspectData(container *Container) (*types.ContainerJSON
 		Running:        container.State.Running,
 		Paused:         container.State.Paused,
 		Checkpointed:   container.State.Checkpointed,
-		Restarting: 	container.State.Restarting,
-		OOMKilled:  	container.State.OOMKilled,
-		Dead:       	container.State.Dead,
-		Pid:        	container.State.Pid,
-		ExitCode:   	container.State.ExitCode,
-		Error:      	container.State.Error,
-		StartedAt:  	container.State.StartedAt.Format(time.RFC3339Nano),
-		FinishedAt: 	container.State.FinishedAt.Format(time.RFC3339Nano),
+		Restarting:     container.State.Restarting,
+		OOMKilled:      container.State.OOMKilled,
+		Dead:           container.State.Dead,
+		Pid:            container.State.Pid,
+		ExitCode:       container.State.ExitCode,
+		Error:          container.State.Error,
+		StartedAt:      container.State.StartedAt.Format(time.RFC3339Nano),
+		FinishedAt:     container.State.FinishedAt.Format(time.RFC3339Nano),
 		CheckpointedAt: container.State.CheckpointedAt.Format(time.RFC3339Nano),
 	}
 

diff --git a/docker/docker.go b/docker/docker.go
@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 	"os"
+	"sort"
 
 	"github.com/Sirupsen/logrus"
 	"github.com/docker/docker/api/client"
@@ -35,7 +36,6 @@ func main() {
 
 		help := "\nCommands:\n"
 
-		// TODO(tiborvass): no need to sort if we ensure dockerCommands is sorted
 		allCommands := append(dockerCommands, experimentalCommands...)
 		sort.Sort(byName(allCommands))