add restart cluster command.

Signed-off-by: Jintao Zhang <zhangjintao9020@gmail.com>

Author: tao12345666333

cmd/kind/kind.go

@@ -31,6 +31,7 @@ import (
 	"sigs.k8s.io/kind/cmd/kind/export"
 	"sigs.k8s.io/kind/cmd/kind/get"
 	"sigs.k8s.io/kind/cmd/kind/load"
+	"sigs.k8s.io/kind/cmd/kind/restart"
 	"sigs.k8s.io/kind/cmd/kind/version"
 	logutil "sigs.k8s.io/kind/pkg/log"
 )
@@ -70,6 +71,7 @@ func NewCommand() *cobra.Command {
 	cmd.AddCommand(get.NewCommand())
 	cmd.AddCommand(version.NewCommand())
 	cmd.AddCommand(load.NewCommand())
+	cmd.AddCommand(restart.NewCommand())
 	return cmd
 }
 

cmd/kind/restart/cluster/restartcluster.go

@@ -0,0 +1,65 @@
+/*
+Copyright 2019 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+// Package cluster implements the `restart` command
+package cluster
+
+import (
+	"fmt"
+
+	"github.com/pkg/errors"
+	"github.com/spf13/cobra"
+
+	"sigs.k8s.io/kind/pkg/cluster"
+)
+
+type flagpole struct {
+	Name string
+}
+
+// NewCommand returns a new cobra.Command for cluster creation
+func NewCommand() *cobra.Command {
+	flags := &flagpole{}
+	cmd := &cobra.Command{
+		Args:  cobra.NoArgs,
+		Use:   "cluster",
+		Short: "Restart a cluster",
+		Long:  "Restart a cluster",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runE(flags, cmd, args)
+		},
+	}
+	cmd.Flags().StringVar(&flags.Name, "name", cluster.DefaultName, "the cluster name")
+	return cmd
+}
+
+func runE(flags *flagpole, cmd *cobra.Command, args []string) error {
+	// Check if the cluster name exists
+	known, err := cluster.IsKnown(flags.Name)
+	if err != nil {
+		return err
+	}
+	if !known {
+		return errors.Errorf("unknown cluster %q", flags.Name)
+	}
+	// Restart the cluster
+	fmt.Printf("Restarting cluster %q ...\n", flags.Name)
+	ctx := cluster.NewContext(flags.Name)
+	if err := ctx.Restart(); err != nil {
+		return errors.Wrap(err, "failed to restart cluster")
+	}
+	return nil
+}

cmd/kind/restart/restart.go

@@ -0,0 +1,36 @@
+/*
+Copyright 2019 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+// Package restart implements the `restart` command
+package restart
+
+import (
+	"github.com/spf13/cobra"
+
+	restartcluster "sigs.k8s.io/kind/cmd/kind/restart/cluster"
+)
+
+// NewCommand returns a new cobra.Command for cluster creation
+func NewCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Args:  cobra.NoArgs,
+		Use:   "restart",
+		Short: "restarts one of [cluster]",
+		Long:  "restarts one of local Kubernetes cluster (cluster)",
+	}
+	cmd.AddCommand(restartcluster.NewCommand())
+	return cmd
+}

pkg/cluster/context.go

@@ -23,6 +23,7 @@ import (
 	internalcontext "sigs.k8s.io/kind/pkg/cluster/internal/context"
 	internalcreate "sigs.k8s.io/kind/pkg/cluster/internal/create"
 	internaldelete "sigs.k8s.io/kind/pkg/cluster/internal/delete"
+	internalrestart "sigs.k8s.io/kind/pkg/cluster/internal/restart"
 	"sigs.k8s.io/kind/pkg/cluster/logs"
 	"sigs.k8s.io/kind/pkg/cluster/nodes"
 )
@@ -99,3 +100,8 @@ func (c *Context) CollectLogs(dir string) error {
 	}
 	return logs.Collect(nodes, dir)
 }
+
+// Restart a kubernetes-in-docker cluster
+func (c *Context) Restart() error {
+	return internalrestart.Cluster(c.ic)
+}

pkg/cluster/internal/restart/restart.go

@@ -0,0 +1,34 @@
+/*
+Copyright 2019 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package restart
+
+import (
+	"github.com/pkg/errors"
+
+	"sigs.k8s.io/kind/pkg/cluster/internal/context"
+	"sigs.k8s.io/kind/pkg/cluster/nodes"
+)
+
+// Cluster restarts the cluster identified by ctx
+func Cluster(c *context.Context) error {
+	n, err := c.ListNodes()
+	if err != nil {
+		return errors.Wrap(err, "error listing nodes")
+	}
+
+	return nodes.Restart(n)
+}

pkg/cluster/nodes/nodes.go

@@ -134,3 +134,49 @@ func WaitForReady(node *Node, until time.Time) bool {
 		return true
 	})
 }
+
+// Restart restarts nodes by name / ID (see Node.String())
+func Restart(nodes []Node) error {
+	if len(nodes) == 0 {
+		return nil
+	}
+	ids := []string{}
+	for _, node := range nodes {
+		ids = append(ids, node.name)
+	}
+	cmd := exec.Command(
+		"docker",
+		append(
+			[]string{
+				"restart",
+				// the default wait time is 10s
+				"--time=1",
+			},
+			ids...,
+		)...,
+	)
+
+	if err := cmd.Run(); err != nil {
+		// TODO: logging here
+		return err
+	}
+
+	for _, node := range nodes {
+
+		if err := node.FixMounts(); err != nil {
+			// TODO: logging here
+			return err
+		}
+
+		if err := node.SignalStart(); err != nil {
+			// TODO: logging here
+			return err
+		}
+
+		if !node.WaitForDocker(time.Now().Add(time.Second * 30)) {
+			return errors.Errorf("timed out waiting for docker to be ready on node %s", node.Name())
+		}
+	}
+
+	return nil
+}