| /* |
| Copyright 2016 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 create |
| |
| import ( |
| "github.com/spf13/cobra" |
| |
| "k8s.io/cli-runtime/pkg/genericclioptions" |
| cmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util" |
| "k8s.io/kubernetes/pkg/kubectl/generate" |
| generateversioned "k8s.io/kubernetes/pkg/kubectl/generate/versioned" |
| "k8s.io/kubernetes/pkg/kubectl/util/i18n" |
| "k8s.io/kubernetes/pkg/kubectl/util/templates" |
| ) |
| |
| var ( |
| pdbLong = templates.LongDesc(i18n.T(` |
| Create a pod disruption budget with the specified name, selector, and desired minimum available pods`)) |
| |
| pdbExample = templates.Examples(i18n.T(` |
| # Create a pod disruption budget named my-pdb that will select all pods with the app=rails label |
| # and require at least one of them being available at any point in time. |
| kubectl create poddisruptionbudget my-pdb --selector=app=rails --min-available=1 |
| |
| # Create a pod disruption budget named my-pdb that will select all pods with the app=nginx label |
| # and require at least half of the pods selected to be available at any point in time. |
| kubectl create pdb my-pdb --selector=app=nginx --min-available=50%`)) |
| ) |
| |
| type PodDisruptionBudgetOpts struct { |
| CreateSubcommandOptions *CreateSubcommandOptions |
| } |
| |
| // NewCmdCreatePodDisruptionBudget is a macro command to create a new pod disruption budget. |
| func NewCmdCreatePodDisruptionBudget(f cmdutil.Factory, ioStreams genericclioptions.IOStreams) *cobra.Command { |
| options := &PodDisruptionBudgetOpts{ |
| CreateSubcommandOptions: NewCreateSubcommandOptions(ioStreams), |
| } |
| |
| cmd := &cobra.Command{ |
| Use: "poddisruptionbudget NAME --selector=SELECTOR --min-available=N [--dry-run]", |
| DisableFlagsInUseLine: true, |
| Aliases: []string{"pdb"}, |
| Short: i18n.T("Create a pod disruption budget with the specified name."), |
| Long: pdbLong, |
| Example: pdbExample, |
| Run: func(cmd *cobra.Command, args []string) { |
| cmdutil.CheckErr(options.Complete(f, cmd, args)) |
| cmdutil.CheckErr(options.Run()) |
| }, |
| } |
| |
| options.CreateSubcommandOptions.PrintFlags.AddFlags(cmd) |
| |
| cmdutil.AddApplyAnnotationFlags(cmd) |
| cmdutil.AddValidateFlags(cmd) |
| cmdutil.AddGeneratorFlags(cmd, generateversioned.PodDisruptionBudgetV2GeneratorName) |
| |
| cmd.Flags().String("min-available", "", i18n.T("The minimum number or percentage of available pods this budget requires.")) |
| cmd.Flags().String("max-unavailable", "", i18n.T("The maximum number or percentage of unavailable pods this budget requires.")) |
| cmd.Flags().String("selector", "", i18n.T("A label selector to use for this budget. Only equality-based selector requirements are supported.")) |
| return cmd |
| } |
| |
| func (o *PodDisruptionBudgetOpts) Complete(f cmdutil.Factory, cmd *cobra.Command, args []string) error { |
| name, err := NameFromCommandArgs(cmd, args) |
| if err != nil { |
| return err |
| } |
| |
| var generator generate.StructuredGenerator |
| switch generatorName := cmdutil.GetFlagString(cmd, "generator"); generatorName { |
| case generateversioned.PodDisruptionBudgetV1GeneratorName: |
| generator = &generateversioned.PodDisruptionBudgetV1Generator{ |
| Name: name, |
| MinAvailable: cmdutil.GetFlagString(cmd, "min-available"), |
| Selector: cmdutil.GetFlagString(cmd, "selector"), |
| } |
| case generateversioned.PodDisruptionBudgetV2GeneratorName: |
| generator = &generateversioned.PodDisruptionBudgetV2Generator{ |
| Name: name, |
| MinAvailable: cmdutil.GetFlagString(cmd, "min-available"), |
| MaxUnavailable: cmdutil.GetFlagString(cmd, "max-unavailable"), |
| Selector: cmdutil.GetFlagString(cmd, "selector"), |
| } |
| default: |
| return errUnsupportedGenerator(cmd, generatorName) |
| } |
| |
| return o.CreateSubcommandOptions.Complete(f, cmd, args, generator) |
| } |
| |
| // CreatePodDisruptionBudget implements the behavior to run the create pdb command. |
| func (o *PodDisruptionBudgetOpts) Run() error { |
| return o.CreateSubcommandOptions.Run() |
| } |