123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720 |
- /*
- 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 run
- import (
- "context"
- "fmt"
- "time"
- "github.com/docker/distribution/reference"
- "github.com/spf13/cobra"
- "k8s.io/klog"
- corev1 "k8s.io/api/core/v1"
- "k8s.io/apimachinery/pkg/api/errors"
- "k8s.io/apimachinery/pkg/api/meta"
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
- "k8s.io/apimachinery/pkg/fields"
- "k8s.io/apimachinery/pkg/runtime"
- utilerrors "k8s.io/apimachinery/pkg/util/errors"
- "k8s.io/apimachinery/pkg/watch"
- "k8s.io/cli-runtime/pkg/genericclioptions"
- "k8s.io/cli-runtime/pkg/resource"
- "k8s.io/client-go/dynamic"
- "k8s.io/client-go/kubernetes"
- corev1client "k8s.io/client-go/kubernetes/typed/core/v1"
- "k8s.io/client-go/tools/cache"
- watchtools "k8s.io/client-go/tools/watch"
- "k8s.io/kubernetes/pkg/kubectl"
- "k8s.io/kubernetes/pkg/kubectl/cmd/attach"
- "k8s.io/kubernetes/pkg/kubectl/cmd/delete"
- "k8s.io/kubernetes/pkg/kubectl/cmd/exec"
- "k8s.io/kubernetes/pkg/kubectl/cmd/logs"
- 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/polymorphichelpers"
- "k8s.io/kubernetes/pkg/kubectl/scheme"
- "k8s.io/kubernetes/pkg/kubectl/util/i18n"
- "k8s.io/kubernetes/pkg/kubectl/util/interrupt"
- "k8s.io/kubernetes/pkg/kubectl/util/templates"
- uexec "k8s.io/utils/exec"
- )
- var (
- runLong = templates.LongDesc(i18n.T(`
- Create and run a particular image, possibly replicated.
- Creates a deployment or job to manage the created container(s).`))
- runExample = templates.Examples(i18n.T(`
- # Start a single instance of nginx.
- kubectl run nginx --image=nginx
- # Start a single instance of hazelcast and let the container expose port 5701 .
- kubectl run hazelcast --image=hazelcast --port=5701
- # Start a single instance of hazelcast and set environment variables "DNS_DOMAIN=cluster" and "POD_NAMESPACE=default" in the container.
- kubectl run hazelcast --image=hazelcast --env="DNS_DOMAIN=cluster" --env="POD_NAMESPACE=default"
- # Start a single instance of hazelcast and set labels "app=hazelcast" and "env=prod" in the container.
- kubectl run hazelcast --image=hazelcast --labels="app=hazelcast,env=prod"
- # Start a replicated instance of nginx.
- kubectl run nginx --image=nginx --replicas=5
- # Dry run. Print the corresponding API objects without creating them.
- kubectl run nginx --image=nginx --dry-run
- # Start a single instance of nginx, but overload the spec of the deployment with a partial set of values parsed from JSON.
- kubectl run nginx --image=nginx --overrides='{ "apiVersion": "v1", "spec": { ... } }'
- # Start a pod of busybox and keep it in the foreground, don't restart it if it exits.
- kubectl run -i -t busybox --image=busybox --restart=Never
- # Start the nginx container using the default command, but use custom arguments (arg1 .. argN) for that command.
- kubectl run nginx --image=nginx -- <arg1> <arg2> ... <argN>
- # Start the nginx container using a different command and custom arguments.
- kubectl run nginx --image=nginx --command -- <cmd> <arg1> ... <argN>
- # Start the perl container to compute π to 2000 places and print it out.
- kubectl run pi --image=perl --restart=OnFailure -- perl -Mbignum=bpi -wle 'print bpi(2000)'
- # Start the cron job to compute π to 2000 places and print it out every 5 minutes.
- kubectl run pi --schedule="0/5 * * * ?" --image=perl --restart=OnFailure -- perl -Mbignum=bpi -wle 'print bpi(2000)'`))
- )
- const (
- defaultPodAttachTimeout = 60 * time.Second
- )
- var metadataAccessor = meta.NewAccessor()
- type RunObject struct {
- Object runtime.Object
- Mapping *meta.RESTMapping
- }
- type RunOptions struct {
- PrintFlags *genericclioptions.PrintFlags
- RecordFlags *genericclioptions.RecordFlags
- DeleteFlags *delete.DeleteFlags
- DeleteOptions *delete.DeleteOptions
- DryRun bool
- PrintObj func(runtime.Object) error
- Recorder genericclioptions.Recorder
- DynamicClient dynamic.Interface
- ArgsLenAtDash int
- Attach bool
- Expose bool
- Generator string
- Image string
- Interactive bool
- LeaveStdinOpen bool
- Port string
- Quiet bool
- Schedule string
- TTY bool
- genericclioptions.IOStreams
- }
- func NewRunOptions(streams genericclioptions.IOStreams) *RunOptions {
- return &RunOptions{
- PrintFlags: genericclioptions.NewPrintFlags("created").WithTypeSetter(scheme.Scheme),
- DeleteFlags: delete.NewDeleteFlags("to use to replace the resource."),
- RecordFlags: genericclioptions.NewRecordFlags(),
- Recorder: genericclioptions.NoopRecorder{},
- IOStreams: streams,
- }
- }
- func NewCmdRun(f cmdutil.Factory, streams genericclioptions.IOStreams) *cobra.Command {
- o := NewRunOptions(streams)
- cmd := &cobra.Command{
- Use: "run NAME --image=image [--env=\"key=value\"] [--port=port] [--replicas=replicas] [--dry-run=bool] [--overrides=inline-json] [--command] -- [COMMAND] [args...]",
- DisableFlagsInUseLine: true,
- Short: i18n.T("Run a particular image on the cluster"),
- Long: runLong,
- Example: runExample,
- Run: func(cmd *cobra.Command, args []string) {
- cmdutil.CheckErr(o.Complete(f, cmd))
- cmdutil.CheckErr(o.Run(f, cmd, args))
- },
- }
- o.DeleteFlags.AddFlags(cmd)
- o.PrintFlags.AddFlags(cmd)
- o.RecordFlags.AddFlags(cmd)
- addRunFlags(cmd, o)
- cmdutil.AddApplyAnnotationFlags(cmd)
- cmdutil.AddPodRunningTimeoutFlag(cmd, defaultPodAttachTimeout)
- return cmd
- }
- func addRunFlags(cmd *cobra.Command, opt *RunOptions) {
- cmdutil.AddDryRunFlag(cmd)
- cmd.Flags().StringVar(&opt.Generator, "generator", opt.Generator, i18n.T("The name of the API generator to use, see http://kubernetes.io/docs/user-guide/kubectl-conventions/#generators for a list."))
- cmd.Flags().StringVar(&opt.Image, "image", opt.Image, i18n.T("The image for the container to run."))
- cmd.MarkFlagRequired("image")
- cmd.Flags().String("image-pull-policy", "", i18n.T("The image pull policy for the container. If left empty, this value will not be specified by the client and defaulted by the server"))
- cmd.Flags().IntP("replicas", "r", 1, "Number of replicas to create for this container. Default is 1.")
- cmd.Flags().Bool("rm", false, "If true, delete resources created in this command for attached containers.")
- cmd.Flags().String("overrides", "", i18n.T("An inline JSON override for the generated object. If this is non-empty, it is used to override the generated object. Requires that the object supply a valid apiVersion field."))
- cmd.Flags().StringArray("env", []string{}, "Environment variables to set in the container")
- cmd.Flags().String("serviceaccount", "", "Service account to set in the pod spec")
- cmd.Flags().StringVar(&opt.Port, "port", opt.Port, i18n.T("The port that this container exposes. If --expose is true, this is also the port used by the service that is created."))
- cmd.Flags().Int("hostport", -1, "The host port mapping for the container port. To demonstrate a single-machine container.")
- cmd.Flags().StringP("labels", "l", "", "Comma separated labels to apply to the pod(s). Will override previous values.")
- cmd.Flags().BoolVarP(&opt.Interactive, "stdin", "i", opt.Interactive, "Keep stdin open on the container(s) in the pod, even if nothing is attached.")
- cmd.Flags().BoolVarP(&opt.TTY, "tty", "t", opt.TTY, "Allocated a TTY for each container in the pod.")
- cmd.Flags().BoolVar(&opt.Attach, "attach", opt.Attach, "If true, wait for the Pod to start running, and then attach to the Pod as if 'kubectl attach ...' were called. Default false, unless '-i/--stdin' is set, in which case the default is true. With '--restart=Never' the exit code of the container process is returned.")
- cmd.Flags().BoolVar(&opt.LeaveStdinOpen, "leave-stdin-open", opt.LeaveStdinOpen, "If the pod is started in interactive mode or with stdin, leave stdin open after the first attach completes. By default, stdin will be closed after the first attach completes.")
- cmd.Flags().String("restart", "Always", i18n.T("The restart policy for this Pod. Legal values [Always, OnFailure, Never]. If set to 'Always' a deployment is created, if set to 'OnFailure' a job is created, if set to 'Never', a regular pod is created. For the latter two --replicas must be 1. Default 'Always', for CronJobs `Never`."))
- cmd.Flags().Bool("command", false, "If true and extra arguments are present, use them as the 'command' field in the container, rather than the 'args' field which is the default.")
- cmd.Flags().String("requests", "", i18n.T("The resource requirement requests for this container. For example, 'cpu=100m,memory=256Mi'. Note that server side components may assign requests depending on the server configuration, such as limit ranges."))
- cmd.Flags().String("limits", "", i18n.T("The resource requirement limits for this container. For example, 'cpu=200m,memory=512Mi'. Note that server side components may assign limits depending on the server configuration, such as limit ranges."))
- cmd.Flags().BoolVar(&opt.Expose, "expose", opt.Expose, "If true, a public, external service is created for the container(s) which are run")
- cmd.Flags().String("service-generator", "service/v2", i18n.T("The name of the generator to use for creating a service. Only used if --expose is true"))
- cmd.Flags().String("service-overrides", "", i18n.T("An inline JSON override for the generated service object. If this is non-empty, it is used to override the generated object. Requires that the object supply a valid apiVersion field. Only used if --expose is true."))
- cmd.Flags().BoolVar(&opt.Quiet, "quiet", opt.Quiet, "If true, suppress prompt messages.")
- cmd.Flags().StringVar(&opt.Schedule, "schedule", opt.Schedule, i18n.T("A schedule in the Cron format the job should be run with."))
- }
- func (o *RunOptions) Complete(f cmdutil.Factory, cmd *cobra.Command) error {
- var err error
- o.RecordFlags.Complete(cmd)
- o.Recorder, err = o.RecordFlags.ToRecorder()
- if err != nil {
- return err
- }
- o.DynamicClient, err = f.DynamicClient()
- if err != nil {
- return err
- }
- o.ArgsLenAtDash = cmd.ArgsLenAtDash()
- o.DryRun = cmdutil.GetFlagBool(cmd, "dry-run")
- attachFlag := cmd.Flags().Lookup("attach")
- if !attachFlag.Changed && o.Interactive {
- o.Attach = true
- }
- if o.DryRun {
- o.PrintFlags.Complete("%s (dry run)")
- }
- printer, err := o.PrintFlags.ToPrinter()
- if err != nil {
- return err
- }
- o.PrintObj = func(obj runtime.Object) error {
- return printer.PrintObj(obj, o.Out)
- }
- deleteOpts := o.DeleteFlags.ToOptions(o.DynamicClient, o.IOStreams)
- deleteOpts.IgnoreNotFound = true
- deleteOpts.WaitForDeletion = false
- deleteOpts.GracePeriod = -1
- deleteOpts.Quiet = o.Quiet
- o.DeleteOptions = deleteOpts
- return nil
- }
- func (o *RunOptions) Run(f cmdutil.Factory, cmd *cobra.Command, args []string) error {
- // Let kubectl run follow rules for `--`, see #13004 issue
- if len(args) == 0 || o.ArgsLenAtDash == 0 {
- return cmdutil.UsageErrorf(cmd, "NAME is required for run")
- }
- timeout, err := cmdutil.GetPodRunningTimeoutFlag(cmd)
- if err != nil {
- return cmdutil.UsageErrorf(cmd, "%v", err)
- }
- // validate image name
- imageName := o.Image
- if imageName == "" {
- return fmt.Errorf("--image is required")
- }
- validImageRef := reference.ReferenceRegexp.MatchString(imageName)
- if !validImageRef {
- return fmt.Errorf("Invalid image name %q: %v", imageName, reference.ErrReferenceInvalidFormat)
- }
- if o.TTY && !o.Interactive {
- return cmdutil.UsageErrorf(cmd, "-i/--stdin is required for containers with -t/--tty=true")
- }
- replicas := cmdutil.GetFlagInt(cmd, "replicas")
- if o.Interactive && replicas != 1 {
- return cmdutil.UsageErrorf(cmd, "-i/--stdin requires that replicas is 1, found %d", replicas)
- }
- if o.Expose && len(o.Port) == 0 {
- return cmdutil.UsageErrorf(cmd, "--port must be set when exposing a service")
- }
- namespace, _, err := f.ToRawKubeConfigLoader().Namespace()
- if err != nil {
- return err
- }
- restartPolicy, err := getRestartPolicy(cmd, o.Interactive)
- if err != nil {
- return err
- }
- if restartPolicy != corev1.RestartPolicyAlways && replicas != 1 {
- return cmdutil.UsageErrorf(cmd, "--restart=%s requires that --replicas=1, found %d", restartPolicy, replicas)
- }
- remove := cmdutil.GetFlagBool(cmd, "rm")
- if !o.Attach && remove {
- return cmdutil.UsageErrorf(cmd, "--rm should only be used for attached containers")
- }
- if o.Attach && o.DryRun {
- return cmdutil.UsageErrorf(cmd, "--dry-run can't be used with attached containers options (--attach, --stdin, or --tty)")
- }
- if err := verifyImagePullPolicy(cmd); err != nil {
- return err
- }
- clientset, err := f.KubernetesClientSet()
- if err != nil {
- return err
- }
- generatorName := o.Generator
- if len(o.Schedule) != 0 && len(generatorName) == 0 {
- generatorName = generateversioned.CronJobV1Beta1GeneratorName
- }
- if len(generatorName) == 0 {
- switch restartPolicy {
- case corev1.RestartPolicyAlways:
- generatorName = generateversioned.DeploymentAppsV1GeneratorName
- case corev1.RestartPolicyOnFailure:
- generatorName = generateversioned.JobV1GeneratorName
- case corev1.RestartPolicyNever:
- generatorName = generateversioned.RunPodV1GeneratorName
- }
- // Falling back because the generator was not provided and the default one could be unavailable.
- generatorNameTemp, err := generateversioned.FallbackGeneratorNameIfNecessary(generatorName, clientset.Discovery(), o.ErrOut)
- if err != nil {
- return err
- }
- if generatorNameTemp != generatorName {
- cmdutil.Warning(o.ErrOut, generatorName, generatorNameTemp)
- } else {
- generatorName = generatorNameTemp
- }
- }
- // start deprecating all generators except for 'run-pod/v1' which will be
- // the only supported on a route to simple kubectl run which should mimic
- // docker run
- if generatorName != generateversioned.RunPodV1GeneratorName {
- fmt.Fprintf(o.ErrOut, "kubectl run --generator=%s is DEPRECATED and will be removed in a future version. Use kubectl run --generator=%s or kubectl create instead.\n", generatorName, generateversioned.RunPodV1GeneratorName)
- }
- generators := generateversioned.GeneratorFn("run")
- generator, found := generators[generatorName]
- if !found {
- return cmdutil.UsageErrorf(cmd, "generator %q not found", generatorName)
- }
- names := generator.ParamNames()
- params := generate.MakeParams(cmd, names)
- params["name"] = args[0]
- if len(args) > 1 {
- params["args"] = args[1:]
- }
- params["env"] = cmdutil.GetFlagStringArray(cmd, "env")
- var createdObjects = []*RunObject{}
- runObject, err := o.createGeneratedObject(f, cmd, generator, names, params, cmdutil.GetFlagString(cmd, "overrides"), namespace)
- if err != nil {
- return err
- }
- createdObjects = append(createdObjects, runObject)
- allErrs := []error{}
- if o.Expose {
- serviceGenerator := cmdutil.GetFlagString(cmd, "service-generator")
- if len(serviceGenerator) == 0 {
- return cmdutil.UsageErrorf(cmd, "No service generator specified")
- }
- serviceRunObject, err := o.generateService(f, cmd, serviceGenerator, params, namespace)
- if err != nil {
- allErrs = append(allErrs, err)
- } else {
- createdObjects = append(createdObjects, serviceRunObject)
- }
- }
- if o.Attach {
- if remove {
- defer o.removeCreatedObjects(f, createdObjects)
- }
- opts := &attach.AttachOptions{
- StreamOptions: exec.StreamOptions{
- IOStreams: o.IOStreams,
- Stdin: o.Interactive,
- TTY: o.TTY,
- Quiet: o.Quiet,
- },
- GetPodTimeout: timeout,
- CommandName: cmd.Parent().CommandPath() + " attach",
- Attach: &attach.DefaultRemoteAttach{},
- }
- config, err := f.ToRESTConfig()
- if err != nil {
- return err
- }
- opts.Config = config
- opts.AttachFunc = attach.DefaultAttachFunc
- clientset, err := kubernetes.NewForConfig(config)
- if err != nil {
- return err
- }
- attachablePod, err := polymorphichelpers.AttachablePodForObjectFn(f, runObject.Object, opts.GetPodTimeout)
- if err != nil {
- return err
- }
- err = handleAttachPod(f, clientset.CoreV1(), attachablePod.Namespace, attachablePod.Name, opts)
- if err != nil {
- return err
- }
- var pod *corev1.Pod
- leaveStdinOpen := o.LeaveStdinOpen
- waitForExitCode := !leaveStdinOpen && restartPolicy == corev1.RestartPolicyNever
- if waitForExitCode {
- pod, err = waitForPod(clientset.CoreV1(), attachablePod.Namespace, attachablePod.Name, kubectl.PodCompleted)
- if err != nil {
- return err
- }
- }
- // after removal is done, return successfully if we are not interested in the exit code
- if !waitForExitCode {
- return nil
- }
- switch pod.Status.Phase {
- case corev1.PodSucceeded:
- return nil
- case corev1.PodFailed:
- unknownRcErr := fmt.Errorf("pod %s/%s failed with unknown exit code", pod.Namespace, pod.Name)
- if len(pod.Status.ContainerStatuses) == 0 || pod.Status.ContainerStatuses[0].State.Terminated == nil {
- return unknownRcErr
- }
- // assume here that we have at most one status because kubectl-run only creates one container per pod
- rc := pod.Status.ContainerStatuses[0].State.Terminated.ExitCode
- if rc == 0 {
- return unknownRcErr
- }
- return uexec.CodeExitError{
- Err: fmt.Errorf("pod %s/%s terminated (%s)\n%s", pod.Namespace, pod.Name, pod.Status.ContainerStatuses[0].State.Terminated.Reason, pod.Status.ContainerStatuses[0].State.Terminated.Message),
- Code: int(rc),
- }
- default:
- return fmt.Errorf("pod %s/%s left in phase %s", pod.Namespace, pod.Name, pod.Status.Phase)
- }
- }
- if runObject != nil {
- if err := o.PrintObj(runObject.Object); err != nil {
- return err
- }
- }
- return utilerrors.NewAggregate(allErrs)
- }
- func (o *RunOptions) removeCreatedObjects(f cmdutil.Factory, createdObjects []*RunObject) error {
- for _, obj := range createdObjects {
- namespace, err := metadataAccessor.Namespace(obj.Object)
- if err != nil {
- return err
- }
- var name string
- name, err = metadataAccessor.Name(obj.Object)
- if err != nil {
- return err
- }
- r := f.NewBuilder().
- WithScheme(scheme.Scheme, scheme.Scheme.PrioritizedVersionsAllGroups()...).
- ContinueOnError().
- NamespaceParam(namespace).DefaultNamespace().
- ResourceNames(obj.Mapping.Resource.Resource+"."+obj.Mapping.Resource.Group, name).
- Flatten().
- Do()
- if err := o.DeleteOptions.DeleteResult(r); err != nil {
- return err
- }
- }
- return nil
- }
- // waitForPod watches the given pod until the exitCondition is true
- func waitForPod(podClient corev1client.PodsGetter, ns, name string, exitCondition watchtools.ConditionFunc) (*corev1.Pod, error) {
- // TODO: expose the timeout
- ctx, cancel := watchtools.ContextWithOptionalTimeout(context.Background(), 0*time.Second)
- defer cancel()
- preconditionFunc := func(store cache.Store) (bool, error) {
- _, exists, err := store.Get(&metav1.ObjectMeta{Namespace: ns, Name: name})
- if err != nil {
- return true, err
- }
- if !exists {
- // We need to make sure we see the object in the cache before we start waiting for events
- // or we would be waiting for the timeout if such object didn't exist.
- // (e.g. it was deleted before we started informers so they wouldn't even see the delete event)
- return true, errors.NewNotFound(corev1.Resource("pods"), name)
- }
- return false, nil
- }
- fieldSelector := fields.OneTermEqualSelector("metadata.name", name).String()
- lw := &cache.ListWatch{
- ListFunc: func(options metav1.ListOptions) (runtime.Object, error) {
- options.FieldSelector = fieldSelector
- return podClient.Pods(ns).List(options)
- },
- WatchFunc: func(options metav1.ListOptions) (watch.Interface, error) {
- options.FieldSelector = fieldSelector
- return podClient.Pods(ns).Watch(options)
- },
- }
- intr := interrupt.New(nil, cancel)
- var result *corev1.Pod
- err := intr.Run(func() error {
- ev, err := watchtools.UntilWithSync(ctx, lw, &corev1.Pod{}, preconditionFunc, func(ev watch.Event) (bool, error) {
- return exitCondition(ev)
- })
- if ev != nil {
- result = ev.Object.(*corev1.Pod)
- }
- return err
- })
- return result, err
- }
- func handleAttachPod(f cmdutil.Factory, podClient corev1client.PodsGetter, ns, name string, opts *attach.AttachOptions) error {
- pod, err := waitForPod(podClient, ns, name, kubectl.PodRunningAndReady)
- if err != nil && err != kubectl.ErrPodCompleted {
- return err
- }
- if pod.Status.Phase == corev1.PodSucceeded || pod.Status.Phase == corev1.PodFailed {
- return logOpts(f, pod, opts)
- }
- opts.Pod = pod
- opts.PodName = name
- opts.Namespace = ns
- if opts.AttachFunc == nil {
- opts.AttachFunc = attach.DefaultAttachFunc
- }
- if err := opts.Run(); err != nil {
- fmt.Fprintf(opts.ErrOut, "Error attaching, falling back to logs: %v\n", err)
- return logOpts(f, pod, opts)
- }
- return nil
- }
- // logOpts logs output from opts to the pods log.
- func logOpts(restClientGetter genericclioptions.RESTClientGetter, pod *corev1.Pod, opts *attach.AttachOptions) error {
- ctrName, err := opts.GetContainerName(pod)
- if err != nil {
- return err
- }
- requests, err := polymorphichelpers.LogsForObjectFn(restClientGetter, pod, &corev1.PodLogOptions{Container: ctrName}, opts.GetPodTimeout, false)
- if err != nil {
- return err
- }
- for _, request := range requests {
- if err := logs.DefaultConsumeRequest(request, opts.Out); err != nil {
- return err
- }
- }
- return nil
- }
- func getRestartPolicy(cmd *cobra.Command, interactive bool) (corev1.RestartPolicy, error) {
- restart := cmdutil.GetFlagString(cmd, "restart")
- if len(restart) == 0 {
- if interactive {
- return corev1.RestartPolicyOnFailure, nil
- }
- return corev1.RestartPolicyAlways, nil
- }
- switch corev1.RestartPolicy(restart) {
- case corev1.RestartPolicyAlways:
- return corev1.RestartPolicyAlways, nil
- case corev1.RestartPolicyOnFailure:
- return corev1.RestartPolicyOnFailure, nil
- case corev1.RestartPolicyNever:
- return corev1.RestartPolicyNever, nil
- }
- return "", cmdutil.UsageErrorf(cmd, "invalid restart policy: %s", restart)
- }
- func verifyImagePullPolicy(cmd *cobra.Command) error {
- pullPolicy := cmdutil.GetFlagString(cmd, "image-pull-policy")
- switch corev1.PullPolicy(pullPolicy) {
- case corev1.PullAlways, corev1.PullIfNotPresent, corev1.PullNever:
- return nil
- case "":
- return nil
- }
- return cmdutil.UsageErrorf(cmd, "invalid image pull policy: %s", pullPolicy)
- }
- func (o *RunOptions) generateService(f cmdutil.Factory, cmd *cobra.Command, serviceGenerator string, paramsIn map[string]interface{}, namespace string) (*RunObject, error) {
- generators := generateversioned.GeneratorFn("expose")
- generator, found := generators[serviceGenerator]
- if !found {
- return nil, fmt.Errorf("missing service generator: %s", serviceGenerator)
- }
- names := generator.ParamNames()
- params := map[string]interface{}{}
- for key, value := range paramsIn {
- _, isString := value.(string)
- if isString {
- params[key] = value
- }
- }
- name, found := params["name"]
- if !found || len(name.(string)) == 0 {
- return nil, fmt.Errorf("name is a required parameter")
- }
- selector, found := params["labels"]
- if !found || len(selector.(string)) == 0 {
- selector = fmt.Sprintf("run=%s", name.(string))
- }
- params["selector"] = selector
- if defaultName, found := params["default-name"]; !found || len(defaultName.(string)) == 0 {
- params["default-name"] = name
- }
- runObject, err := o.createGeneratedObject(f, cmd, generator, names, params, cmdutil.GetFlagString(cmd, "service-overrides"), namespace)
- if err != nil {
- return nil, err
- }
- if err := o.PrintObj(runObject.Object); err != nil {
- return nil, err
- }
- // separate yaml objects
- if o.PrintFlags.OutputFormat != nil && *o.PrintFlags.OutputFormat == "yaml" {
- fmt.Fprintln(o.Out, "---")
- }
- return runObject, nil
- }
- func (o *RunOptions) createGeneratedObject(f cmdutil.Factory, cmd *cobra.Command, generator generate.Generator, names []generate.GeneratorParam, params map[string]interface{}, overrides, namespace string) (*RunObject, error) {
- err := generate.ValidateParams(names, params)
- if err != nil {
- return nil, err
- }
- // TODO: Validate flag usage against selected generator. More tricky since --expose was added.
- obj, err := generator.Generate(params)
- if err != nil {
- return nil, err
- }
- mapper, err := f.ToRESTMapper()
- if err != nil {
- return nil, err
- }
- // run has compiled knowledge of the thing is creating
- gvks, _, err := scheme.Scheme.ObjectKinds(obj)
- if err != nil {
- return nil, err
- }
- mapping, err := mapper.RESTMapping(gvks[0].GroupKind(), gvks[0].Version)
- if err != nil {
- return nil, err
- }
- if len(overrides) > 0 {
- codec := runtime.NewCodec(scheme.DefaultJSONEncoder(), scheme.Codecs.UniversalDecoder(scheme.Scheme.PrioritizedVersionsAllGroups()...))
- obj, err = cmdutil.Merge(codec, obj, overrides)
- if err != nil {
- return nil, err
- }
- }
- if err := o.Recorder.Record(obj); err != nil {
- klog.V(4).Infof("error recording current command: %v", err)
- }
- actualObj := obj
- if !o.DryRun {
- if err := kubectl.CreateOrUpdateAnnotation(cmdutil.GetFlagBool(cmd, cmdutil.ApplyAnnotationsFlag), obj, scheme.DefaultJSONEncoder()); err != nil {
- return nil, err
- }
- client, err := f.ClientForMapping(mapping)
- if err != nil {
- return nil, err
- }
- actualObj, err = resource.NewHelper(client, mapping).Create(namespace, false, obj, nil)
- if err != nil {
- return nil, err
- }
- }
- return &RunObject{
- Object: actualObj,
- Mapping: mapping,
- }, nil
- }
|