mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-11-18 12:20:24 +00:00
289 lines
9.0 KiB
Go
289 lines
9.0 KiB
Go
/*
|
|
Copyright 2014 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 cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/golang/glog"
|
|
"k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/apimachinery/pkg/util/wait"
|
|
"k8s.io/kubernetes/pkg/kubectl"
|
|
"k8s.io/kubernetes/pkg/kubectl/cmd/templates"
|
|
cmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util"
|
|
"k8s.io/kubernetes/pkg/kubectl/resource"
|
|
"k8s.io/kubernetes/pkg/kubectl/util/i18n"
|
|
)
|
|
|
|
var (
|
|
replaceLong = templates.LongDesc(i18n.T(`
|
|
Replace a resource by filename or stdin.
|
|
|
|
JSON and YAML formats are accepted. If replacing an existing resource, the
|
|
complete resource spec must be provided. This can be obtained by
|
|
|
|
$ kubectl get TYPE NAME -o yaml
|
|
|
|
Please refer to the models in https://htmlpreview.github.io/?https://github.com/kubernetes/kubernetes/blob/HEAD/docs/api-reference/v1/definitions.html to find if a field is mutable.`))
|
|
|
|
replaceExample = templates.Examples(i18n.T(`
|
|
# Replace a pod using the data in pod.json.
|
|
kubectl replace -f ./pod.json
|
|
|
|
# Replace a pod based on the JSON passed into stdin.
|
|
cat pod.json | kubectl replace -f -
|
|
|
|
# Update a single-container pod's image version (tag) to v4
|
|
kubectl get pod mypod -o yaml | sed 's/\(image: myimage\):.*$/\1:v4/' | kubectl replace -f -
|
|
|
|
# Force replace, delete and then re-create the resource
|
|
kubectl replace --force -f ./pod.json`))
|
|
)
|
|
|
|
func NewCmdReplace(f cmdutil.Factory, out io.Writer) *cobra.Command {
|
|
options := &resource.FilenameOptions{}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "replace -f FILENAME",
|
|
DisableFlagsInUseLine: true,
|
|
Short: i18n.T("Replace a resource by filename or stdin"),
|
|
Long: replaceLong,
|
|
Example: replaceExample,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cmdutil.CheckErr(cmdutil.ValidateOutputArgs(cmd))
|
|
err := RunReplace(f, out, cmd, args, options)
|
|
cmdutil.CheckErr(err)
|
|
},
|
|
}
|
|
usage := "to use to replace the resource."
|
|
cmdutil.AddFilenameOptionFlags(cmd, options, usage)
|
|
cmd.MarkFlagRequired("filename")
|
|
cmd.Flags().Bool("force", false, "Delete and re-create the specified resource")
|
|
cmd.Flags().Bool("cascade", false, "Only relevant during a force replace. If true, cascade the deletion of the resources managed by this resource (e.g. Pods created by a ReplicationController).")
|
|
cmd.Flags().Int("grace-period", -1, "Only relevant during a force replace. Period of time in seconds given to the old resource to terminate gracefully. Ignored if negative.")
|
|
cmd.Flags().Duration("timeout", 0, "Only relevant during a force replace. The length of time to wait before giving up on a delete of the old resource, zero means determine a timeout from the size of the object. Any other values should contain a corresponding time unit (e.g. 1s, 2m, 3h).")
|
|
cmdutil.AddValidateFlags(cmd)
|
|
cmdutil.AddOutputFlagsForMutation(cmd)
|
|
cmdutil.AddApplyAnnotationFlags(cmd)
|
|
cmdutil.AddRecordFlag(cmd)
|
|
cmdutil.AddInclude3rdPartyFlags(cmd)
|
|
|
|
return cmd
|
|
}
|
|
|
|
func RunReplace(f cmdutil.Factory, out io.Writer, cmd *cobra.Command, args []string, options *resource.FilenameOptions) error {
|
|
schema, err := f.Validator(cmdutil.GetFlagBool(cmd, "validate"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cmdNamespace, enforceNamespace, err := f.DefaultNamespace()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
force := cmdutil.GetFlagBool(cmd, "force")
|
|
if cmdutil.IsFilenameSliceEmpty(options.Filenames) {
|
|
return cmdutil.UsageErrorf(cmd, "Must specify --filename to replace")
|
|
}
|
|
|
|
shortOutput := cmdutil.GetFlagString(cmd, "output") == "name"
|
|
if force {
|
|
return forceReplace(f, out, cmd, args, shortOutput, options)
|
|
}
|
|
|
|
if cmdutil.GetFlagInt(cmd, "grace-period") >= 0 {
|
|
return fmt.Errorf("--grace-period must have --force specified")
|
|
}
|
|
|
|
if cmdutil.GetFlagDuration(cmd, "timeout") != 0 {
|
|
return fmt.Errorf("--timeout must have --force specified")
|
|
}
|
|
|
|
r := f.NewBuilder().
|
|
Unstructured().
|
|
Schema(schema).
|
|
ContinueOnError().
|
|
NamespaceParam(cmdNamespace).DefaultNamespace().
|
|
FilenameParam(enforceNamespace, options).
|
|
Flatten().
|
|
Do()
|
|
if err := r.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.Visit(func(info *resource.Info, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := kubectl.CreateOrUpdateAnnotation(cmdutil.GetFlagBool(cmd, cmdutil.ApplyAnnotationsFlag), info, cmdutil.InternalVersionJSONEncoder()); err != nil {
|
|
return cmdutil.AddSourceToErr("replacing", info.Source, err)
|
|
}
|
|
|
|
if cmdutil.ShouldRecord(cmd, info) {
|
|
if err := cmdutil.RecordChangeCause(info.Object, f.Command(cmd, false)); err != nil {
|
|
return cmdutil.AddSourceToErr("replacing", info.Source, err)
|
|
}
|
|
}
|
|
|
|
// Serialize the object with the annotation applied.
|
|
obj, err := resource.NewHelper(info.Client, info.Mapping).Replace(info.Namespace, info.Name, true, info.Object)
|
|
if err != nil {
|
|
return cmdutil.AddSourceToErr("replacing", info.Source, err)
|
|
}
|
|
|
|
info.Refresh(obj, true)
|
|
cmdutil.PrintSuccess(shortOutput, out, info.Object, false, "replaced")
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func forceReplace(f cmdutil.Factory, out io.Writer, cmd *cobra.Command, args []string, shortOutput bool, options *resource.FilenameOptions) error {
|
|
schema, err := f.Validator(cmdutil.GetFlagBool(cmd, "validate"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cmdNamespace, enforceNamespace, err := f.DefaultNamespace()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i, filename := range options.Filenames {
|
|
if filename == "-" {
|
|
tempDir, err := ioutil.TempDir("", "kubectl_replace_")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer os.RemoveAll(tempDir)
|
|
tempFilename := filepath.Join(tempDir, "resource.stdin")
|
|
err = cmdutil.DumpReaderToFile(os.Stdin, tempFilename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
options.Filenames[i] = tempFilename
|
|
}
|
|
}
|
|
|
|
r := f.NewBuilder().
|
|
Unstructured().
|
|
ContinueOnError().
|
|
NamespaceParam(cmdNamespace).DefaultNamespace().
|
|
FilenameParam(enforceNamespace, options).
|
|
ResourceTypeOrNameArgs(false, args...).RequireObject(false).
|
|
Flatten().
|
|
Do()
|
|
if err := r.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
//Replace will create a resource if it doesn't exist already, so ignore not found error
|
|
ignoreNotFound := true
|
|
timeout := cmdutil.GetFlagDuration(cmd, "timeout")
|
|
gracePeriod := cmdutil.GetFlagInt(cmd, "grace-period")
|
|
waitForDeletion := false
|
|
if gracePeriod == 0 {
|
|
// To preserve backwards compatibility, but prevent accidental data loss, we convert --grace-period=0
|
|
// into --grace-period=1 and wait until the object is successfully deleted.
|
|
gracePeriod = 1
|
|
waitForDeletion = true
|
|
}
|
|
// By default use a reaper to delete all related resources.
|
|
if cmdutil.GetFlagBool(cmd, "cascade") {
|
|
glog.Warningf("\"cascade\" is set, kubectl will delete and re-create all resources managed by this resource (e.g. Pods created by a ReplicationController). Consider using \"kubectl rolling-update\" if you want to update a ReplicationController together with its Pods.")
|
|
err = ReapResult(r, f, out, cmdutil.GetFlagBool(cmd, "cascade"), ignoreNotFound, timeout, gracePeriod, waitForDeletion, shortOutput, false)
|
|
} else {
|
|
err = DeleteResult(r, out, ignoreNotFound, gracePeriod, shortOutput)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if timeout == 0 {
|
|
timeout = kubectl.Timeout
|
|
}
|
|
err = r.Visit(func(info *resource.Info, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return wait.PollImmediate(kubectl.Interval, timeout, func() (bool, error) {
|
|
if err := info.Get(); !errors.IsNotFound(err) {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
})
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r = f.NewBuilder().
|
|
Unstructured().
|
|
Schema(schema).
|
|
ContinueOnError().
|
|
NamespaceParam(cmdNamespace).DefaultNamespace().
|
|
FilenameParam(enforceNamespace, options).
|
|
Flatten().
|
|
Do()
|
|
err = r.Err()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
count := 0
|
|
err = r.Visit(func(info *resource.Info, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := kubectl.CreateOrUpdateAnnotation(cmdutil.GetFlagBool(cmd, cmdutil.ApplyAnnotationsFlag), info, cmdutil.InternalVersionJSONEncoder()); err != nil {
|
|
return err
|
|
}
|
|
|
|
if cmdutil.ShouldRecord(cmd, info) {
|
|
if err := cmdutil.RecordChangeCause(info.Object, f.Command(cmd, false)); err != nil {
|
|
return cmdutil.AddSourceToErr("replacing", info.Source, err)
|
|
}
|
|
}
|
|
|
|
obj, err := resource.NewHelper(info.Client, info.Mapping).Create(info.Namespace, true, info.Object)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
count++
|
|
info.Refresh(obj, true)
|
|
cmdutil.PrintSuccess(shortOutput, out, info.Object, false, "replaced")
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if count == 0 {
|
|
return fmt.Errorf("no objects passed to replace")
|
|
}
|
|
return nil
|
|
}
|