mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-31 13:50:01 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			91 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			91 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| /*
 | |
| 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 cmd
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"io"
 | |
| 
 | |
| 	"github.com/spf13/cobra"
 | |
| 
 | |
| 	"k8s.io/kubernetes/pkg/kubectl"
 | |
| 	"k8s.io/kubernetes/pkg/kubectl/cmd/templates"
 | |
| 	cmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util"
 | |
| )
 | |
| 
 | |
| var (
 | |
| 	pdbLong = templates.LongDesc(`
 | |
| 		Create a pod disruption budget with the specified name, selector, and desired minimum available pods`)
 | |
| 
 | |
| 	pdbExample = templates.Examples(`
 | |
| 		# 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%`)
 | |
| )
 | |
| 
 | |
| // NewCmdCreatePodDisruptionBudget is a macro command to create a new pod disruption budget.
 | |
| func NewCmdCreatePodDisruptionBudget(f cmdutil.Factory, cmdOut io.Writer) *cobra.Command {
 | |
| 	cmd := &cobra.Command{
 | |
| 		Use:     "poddisruptionbudget NAME --selector=SELECTOR --min-available=N [--dry-run]",
 | |
| 		Aliases: []string{"pdb"},
 | |
| 		Short:   "Create a pod disruption budget with the specified name.",
 | |
| 		Long:    pdbLong,
 | |
| 		Example: pdbExample,
 | |
| 		Run: func(cmd *cobra.Command, args []string) {
 | |
| 			err := CreatePodDisruptionBudget(f, cmdOut, cmd, args)
 | |
| 			cmdutil.CheckErr(err)
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	cmdutil.AddApplyAnnotationFlags(cmd)
 | |
| 	cmdutil.AddValidateFlags(cmd)
 | |
| 	cmdutil.AddPrinterFlags(cmd)
 | |
| 	cmdutil.AddGeneratorFlags(cmd, cmdutil.PodDisruptionBudgetV1GeneratorName)
 | |
| 	cmd.Flags().String("min-available", "1", "The minimum number or percentage of available pods this budget requires.")
 | |
| 	cmd.Flags().String("selector", "", "A label selector to use for this budget. Only equality-based selector requirements are supported.")
 | |
| 	return cmd
 | |
| }
 | |
| 
 | |
| // CreatePodDisruptionBudget implements the behavior to run the create pdb command.
 | |
| func CreatePodDisruptionBudget(f cmdutil.Factory, cmdOut io.Writer, cmd *cobra.Command, args []string) error {
 | |
| 	name, err := NameFromCommandArgs(cmd, args)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	var generator kubectl.StructuredGenerator
 | |
| 	switch generatorName := cmdutil.GetFlagString(cmd, "generator"); generatorName {
 | |
| 	case cmdutil.PodDisruptionBudgetV1GeneratorName:
 | |
| 		generator = &kubectl.PodDisruptionBudgetV1Generator{
 | |
| 			Name:         name,
 | |
| 			MinAvailable: cmdutil.GetFlagString(cmd, "min-available"),
 | |
| 			Selector:     cmdutil.GetFlagString(cmd, "selector"),
 | |
| 		}
 | |
| 	default:
 | |
| 		return cmdutil.UsageError(cmd, fmt.Sprintf("Generator: %s not supported.", generatorName))
 | |
| 	}
 | |
| 	return RunCreateSubcommand(f, cmd, cmdOut, &CreateSubcommandOptions{
 | |
| 		Name:                name,
 | |
| 		StructuredGenerator: generator,
 | |
| 		DryRun:              cmdutil.GetFlagBool(cmd, "dry-run"),
 | |
| 		OutputFormat:        cmdutil.GetFlagString(cmd, "output"),
 | |
| 	})
 | |
| }
 |