2019-01-12 04:58:27 +00:00
/ *
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"
2020-08-10 17:43:49 +00:00
"k8s.io/klog/v2"
2019-01-12 04:58:27 +00:00
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"
2019-04-07 17:07:55 +00:00
"k8s.io/apimachinery/pkg/fields"
2019-01-12 04:58:27 +00:00
"k8s.io/apimachinery/pkg/runtime"
2019-09-27 21:51:53 +00:00
"k8s.io/apimachinery/pkg/runtime/schema"
2019-01-12 04:58:27 +00:00
utilerrors "k8s.io/apimachinery/pkg/util/errors"
"k8s.io/apimachinery/pkg/watch"
"k8s.io/cli-runtime/pkg/genericclioptions"
2019-04-07 17:07:55 +00:00
"k8s.io/cli-runtime/pkg/resource"
2019-01-12 04:58:27 +00:00
"k8s.io/client-go/kubernetes"
corev1client "k8s.io/client-go/kubernetes/typed/core/v1"
2019-04-07 17:07:55 +00:00
"k8s.io/client-go/tools/cache"
2019-01-12 04:58:27 +00:00
watchtools "k8s.io/client-go/tools/watch"
2019-09-27 21:51:53 +00:00
"k8s.io/kubectl/pkg/cmd/attach"
"k8s.io/kubectl/pkg/cmd/delete"
"k8s.io/kubectl/pkg/cmd/exec"
"k8s.io/kubectl/pkg/cmd/logs"
cmdutil "k8s.io/kubectl/pkg/cmd/util"
"k8s.io/kubectl/pkg/generate"
generateversioned "k8s.io/kubectl/pkg/generate/versioned"
"k8s.io/kubectl/pkg/polymorphichelpers"
"k8s.io/kubectl/pkg/scheme"
"k8s.io/kubectl/pkg/util"
"k8s.io/kubectl/pkg/util/i18n"
"k8s.io/kubectl/pkg/util/interrupt"
"k8s.io/kubectl/pkg/util/templates"
2019-01-12 04:58:27 +00:00
uexec "k8s.io/utils/exec"
)
var (
2020-03-26 21:07:15 +00:00
runLong = templates . LongDesc ( i18n . T ( ` Create and run a particular image in a pod. ` ) )
2019-01-12 04:58:27 +00:00
runExample = templates . Examples ( i18n . T ( `
2020-03-26 21:07:15 +00:00
# Start a nginx pod .
2019-01-12 04:58:27 +00:00
kubectl run nginx -- image = nginx
2020-03-26 21:07:15 +00:00
# Start a hazelcast pod and let the container expose port 5701.
kubectl run hazelcast -- image = hazelcast / hazelcast -- port = 5701
2019-01-12 04:58:27 +00:00
2020-03-26 21:07:15 +00:00
# Start a hazelcast pod and set environment variables "DNS_DOMAIN=cluster" and "POD_NAMESPACE=default" in the container .
kubectl run hazelcast -- image = hazelcast / hazelcast -- env = "DNS_DOMAIN=cluster" -- env = "POD_NAMESPACE=default"
2019-01-12 04:58:27 +00:00
2020-03-26 21:07:15 +00:00
# Start a hazelcast pod and set labels "app=hazelcast" and "env=prod" in the container .
kubectl run hazelcast -- image = hazelcast / hazelcast -- labels = "app=hazelcast,env=prod"
2019-01-12 04:58:27 +00:00
# Dry run . Print the corresponding API objects without creating them .
2020-03-26 21:07:15 +00:00
kubectl run nginx -- image = nginx -- dry - run = client
2019-01-12 04:58:27 +00:00
2020-03-26 21:07:15 +00:00
# Start a nginx pod , but overload the spec with a partial set of values parsed from JSON .
2019-01-12 04:58:27 +00:00
kubectl run nginx -- image = nginx -- overrides = ' { "apiVersion" : "v1" , "spec" : { ... } } '
2020-03-26 21:07:15 +00:00
# Start a busybox pod and keep it in the foreground , don ' t restart it if it exits .
2019-01-12 04:58:27 +00:00
kubectl run - i - t busybox -- image = busybox -- restart = Never
2020-03-26 21:07:15 +00:00
# Start the nginx pod using the default command , but use custom arguments ( arg1 . . argN ) for that command .
2019-01-12 04:58:27 +00:00
kubectl run nginx -- image = nginx -- < arg1 > < arg2 > ... < argN >
2020-03-26 21:07:15 +00:00
# Start the nginx pod using a different command and custom arguments .
kubectl run nginx -- image = nginx -- command -- < cmd > < arg1 > ... < argN > ` ) )
2019-01-12 04:58:27 +00:00
)
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
2020-03-26 21:07:15 +00:00
DryRunStrategy cmdutil . DryRunStrategy
DryRunVerifier * resource . DryRunVerifier
2019-01-12 04:58:27 +00:00
PrintObj func ( runtime . Object ) error
Recorder genericclioptions . Recorder
ArgsLenAtDash int
Attach bool
Expose bool
Generator string
Image string
Interactive bool
LeaveStdinOpen bool
Port string
2020-08-10 17:43:49 +00:00
Privileged bool
2019-01-12 04:58:27 +00:00
Quiet bool
Schedule string
TTY bool
2020-08-10 17:43:49 +00:00
fieldManager string
Namespace string
EnforceNamespace bool
2019-01-12 04:58:27 +00:00
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 ( ) ,
2020-03-26 21:07:15 +00:00
Generator : generateversioned . RunPodV1GeneratorName ,
2019-01-12 04:58:27 +00:00
Recorder : genericclioptions . NoopRecorder { } ,
IOStreams : streams ,
}
}
func NewCmdRun ( f cmdutil . Factory , streams genericclioptions . IOStreams ) * cobra . Command {
o := NewRunOptions ( streams )
cmd := & cobra . Command {
2020-03-26 21:07:15 +00:00
Use : "run NAME --image=image [--env=\"key=value\"] [--port=port] [--dry-run=server|client] [--overrides=inline-json] [--command] -- [COMMAND] [args...]" ,
2019-01-12 04:58:27 +00:00
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 )
2020-12-01 01:06:26 +00:00
cmd . Flags ( ) . StringArray ( "annotations" , [ ] string { } , i18n . T ( "Annotations to apply to the pod." ) )
2019-01-12 04:58:27 +00:00
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." ) )
2020-03-26 21:07:15 +00:00
cmd . Flags ( ) . MarkDeprecated ( "generator" , "has no effect and will be removed in the future." )
2019-01-12 04:58:27 +00:00
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." )
2020-03-26 21:07:15 +00:00
cmd . Flags ( ) . MarkDeprecated ( "replicas" , "has no effect and will be removed in the future." )
2019-01-12 04:58:27 +00:00
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." ) )
2020-03-26 21:07:15 +00:00
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." ) )
2019-01-12 04:58:27 +00:00
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." )
2020-08-10 17:43:49 +00:00
cmd . Flags ( ) . String ( "restart" , "Always" , i18n . T ( "The restart policy for this Pod. Legal values [Always, OnFailure, Never]." ) )
2019-01-12 04:58:27 +00:00
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." ) )
2020-03-26 21:07:15 +00:00
cmd . Flags ( ) . BoolVar ( & opt . Expose , "expose" , opt . Expose , "If true, service is created for the container(s) which are run" )
2019-01-12 04:58:27 +00:00
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" ) )
2020-03-26 21:07:15 +00:00
cmd . Flags ( ) . MarkDeprecated ( "service-generator" , "and will be removed in the future." )
2019-01-12 04:58:27 +00:00
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." ) )
2020-03-26 21:07:15 +00:00
cmd . Flags ( ) . MarkDeprecated ( "service-overrides" , "and will be removed in the future." )
2019-01-12 04:58:27 +00:00
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." ) )
2020-03-26 21:07:15 +00:00
cmd . Flags ( ) . MarkDeprecated ( "schedule" , "has no effect and will be removed in the future." )
2020-08-10 17:43:49 +00:00
cmd . Flags ( ) . BoolVar ( & opt . Privileged , "privileged" , opt . Privileged , i18n . T ( "If true, run the container in privileged mode." ) )
cmdutil . AddFieldManagerFlagVar ( cmd , & opt . fieldManager , "kubectl-run" )
2019-01-12 04:58:27 +00:00
}
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
}
2020-03-26 21:07:15 +00:00
o . ArgsLenAtDash = cmd . ArgsLenAtDash ( )
o . DryRunStrategy , err = cmdutil . GetDryRunStrategy ( cmd )
2019-01-12 04:58:27 +00:00
if err != nil {
return err
}
2020-03-26 21:07:15 +00:00
dynamicClient , err := f . DynamicClient ( )
if err != nil {
return err
}
discoveryClient , err := f . ToDiscoveryClient ( )
if err != nil {
return err
}
o . DryRunVerifier = resource . NewDryRunVerifier ( dynamicClient , discoveryClient )
2019-01-12 04:58:27 +00:00
attachFlag := cmd . Flags ( ) . Lookup ( "attach" )
if ! attachFlag . Changed && o . Interactive {
o . Attach = true
}
2020-03-26 21:07:15 +00:00
cmdutil . PrintFlagsWithDryRunStrategy ( o . PrintFlags , o . DryRunStrategy )
2019-01-12 04:58:27 +00:00
printer , err := o . PrintFlags . ToPrinter ( )
if err != nil {
return err
}
o . PrintObj = func ( obj runtime . Object ) error {
return printer . PrintObj ( obj , o . Out )
}
2020-12-01 01:06:26 +00:00
deleteOpts , err := o . DeleteFlags . ToOptions ( dynamicClient , o . IOStreams )
if err != nil {
return err
}
2019-01-12 04:58:27 +00:00
deleteOpts . IgnoreNotFound = true
deleteOpts . WaitForDeletion = false
deleteOpts . GracePeriod = - 1
2019-04-07 17:07:55 +00:00
deleteOpts . Quiet = o . Quiet
2019-01-12 04:58:27 +00:00
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" )
}
if o . Expose && len ( o . Port ) == 0 {
return cmdutil . UsageErrorf ( cmd , "--port must be set when exposing a service" )
}
2020-08-10 17:43:49 +00:00
o . Namespace , o . EnforceNamespace , err = f . ToRawKubeConfigLoader ( ) . Namespace ( )
2019-01-12 04:58:27 +00:00
if err != nil {
return err
}
restartPolicy , err := getRestartPolicy ( cmd , o . Interactive )
if err != nil {
return err
}
remove := cmdutil . GetFlagBool ( cmd , "rm" )
if ! o . Attach && remove {
return cmdutil . UsageErrorf ( cmd , "--rm should only be used for attached containers" )
}
2020-03-26 21:07:15 +00:00
if o . Attach && o . DryRunStrategy != cmdutil . DryRunNone {
return cmdutil . UsageErrorf ( cmd , "--dry-run=[server|client] can't be used with attached containers options (--attach, --stdin, or --tty)" )
2019-01-12 04:58:27 +00:00
}
if err := verifyImagePullPolicy ( cmd ) ; err != nil {
return err
}
2019-12-12 01:27:03 +00:00
generators := generateversioned . GeneratorFn ( "run" )
2020-03-26 21:07:15 +00:00
generator , found := generators [ generateversioned . RunPodV1GeneratorName ]
2019-12-12 01:27:03 +00:00
if ! found {
2020-03-26 21:07:15 +00:00
return cmdutil . UsageErrorf ( cmd , "generator %q not found" , o . Generator )
2019-01-12 04:58:27 +00:00
}
names := generator . ParamNames ( )
params := generate . MakeParams ( cmd , names )
params [ "name" ] = args [ 0 ]
if len ( args ) > 1 {
params [ "args" ] = args [ 1 : ]
}
2020-12-01 01:06:26 +00:00
params [ "annotations" ] = cmdutil . GetFlagStringArray ( cmd , "annotations" )
2019-01-12 04:58:27 +00:00
params [ "env" ] = cmdutil . GetFlagStringArray ( cmd , "env" )
var createdObjects = [ ] * RunObject { }
2020-08-10 17:43:49 +00:00
runObject , err := o . createGeneratedObject ( f , cmd , generator , names , params , cmdutil . GetFlagString ( cmd , "overrides" ) )
2019-01-12 04:58:27 +00:00
if err != nil {
return err
}
2019-04-07 17:07:55 +00:00
createdObjects = append ( createdObjects , runObject )
2019-01-12 04:58:27 +00:00
allErrs := [ ] error { }
if o . Expose {
serviceGenerator := cmdutil . GetFlagString ( cmd , "service-generator" )
if len ( serviceGenerator ) == 0 {
return cmdutil . UsageErrorf ( cmd , "No service generator specified" )
}
2020-08-10 17:43:49 +00:00
serviceRunObject , err := o . generateService ( f , cmd , serviceGenerator , params )
2019-01-12 04:58:27 +00:00
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
2020-08-10 17:43:49 +00:00
waitForExitCode := ! o . LeaveStdinOpen && ( restartPolicy == corev1 . RestartPolicyNever || restartPolicy == corev1 . RestartPolicyOnFailure )
2019-01-12 04:58:27 +00:00
if waitForExitCode {
2020-08-10 17:43:49 +00:00
// we need different exit condition depending on restart policy
// for Never it can either fail or succeed, for OnFailure only
// success matters
exitCondition := podCompleted
if restartPolicy == corev1 . RestartPolicyOnFailure {
exitCondition = podSucceeded
}
pod , err = waitForPod ( clientset . CoreV1 ( ) , attachablePod . Namespace , attachablePod . Name , opts . GetPodTimeout , exitCondition )
2019-01-12 04:58:27 +00:00
if err != nil {
return err
}
2020-08-10 17:43:49 +00:00
} else {
// after removal is done, return successfully if we are not interested in the exit code
2019-01-12 04:58:27 +00:00
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
2020-08-10 17:43:49 +00:00
func waitForPod ( podClient corev1client . PodsGetter , ns , name string , timeout time . Duration , exitCondition watchtools . ConditionFunc ) ( * corev1 . Pod , error ) {
ctx , cancel := watchtools . ContextWithOptionalTimeout ( context . Background ( ) , timeout )
2019-01-12 04:58:27 +00:00
defer cancel ( )
2019-04-07 17:07:55 +00:00
fieldSelector := fields . OneTermEqualSelector ( "metadata.name" , name ) . String ( )
lw := & cache . ListWatch {
ListFunc : func ( options metav1 . ListOptions ) ( runtime . Object , error ) {
options . FieldSelector = fieldSelector
2020-03-26 21:07:15 +00:00
return podClient . Pods ( ns ) . List ( context . TODO ( ) , options )
2019-04-07 17:07:55 +00:00
} ,
WatchFunc : func ( options metav1 . ListOptions ) ( watch . Interface , error ) {
options . FieldSelector = fieldSelector
2020-03-26 21:07:15 +00:00
return podClient . Pods ( ns ) . Watch ( context . TODO ( ) , options )
2019-04-07 17:07:55 +00:00
} ,
}
2019-01-12 04:58:27 +00:00
intr := interrupt . New ( nil , cancel )
var result * corev1 . Pod
2019-04-07 17:07:55 +00:00
err := intr . Run ( func ( ) error {
2020-08-10 17:43:49 +00:00
ev , err := watchtools . UntilWithSync ( ctx , lw , & corev1 . Pod { } , nil , exitCondition )
2019-01-12 04:58:27 +00:00
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 {
2020-08-10 17:43:49 +00:00
pod , err := waitForPod ( podClient , ns , name , opts . GetPodTimeout , podRunningAndReady )
2019-09-27 21:51:53 +00:00
if err != nil && err != ErrPodCompleted {
2019-01-12 04:58:27 +00:00
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
}
2019-04-07 17:07:55 +00:00
return corev1 . RestartPolicyAlways , nil
2019-01-12 04:58:27 +00:00
}
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 )
}
2020-08-10 17:43:49 +00:00
func ( o * RunOptions ) generateService ( f cmdutil . Factory , cmd * cobra . Command , serviceGenerator string , paramsIn map [ string ] interface { } ) ( * RunObject , error ) {
2019-01-12 04:58:27 +00:00
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
}
2020-08-10 17:43:49 +00:00
runObject , err := o . createGeneratedObject ( f , cmd , generator , names , params , cmdutil . GetFlagString ( cmd , "service-overrides" ) )
2019-01-12 04:58:27 +00:00
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
}
2020-08-10 17:43:49 +00:00
func ( o * RunOptions ) createGeneratedObject ( f cmdutil . Factory , cmd * cobra . Command , generator generate . Generator , names [ ] generate . GeneratorParam , params map [ string ] interface { } , overrides string ) ( * RunObject , error ) {
2019-01-12 04:58:27 +00:00
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
}
2019-04-07 17:07:55 +00:00
// run has compiled knowledge of the thing is creating
2019-01-12 04:58:27 +00:00
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
2020-03-26 21:07:15 +00:00
if o . DryRunStrategy != cmdutil . DryRunClient {
2019-09-27 21:51:53 +00:00
if err := util . CreateOrUpdateAnnotation ( cmdutil . GetFlagBool ( cmd , cmdutil . ApplyAnnotationsFlag ) , obj , scheme . DefaultJSONEncoder ( ) ) ; err != nil {
2019-01-12 04:58:27 +00:00
return nil , err
}
client , err := f . ClientForMapping ( mapping )
if err != nil {
return nil , err
}
2020-03-26 21:07:15 +00:00
if o . DryRunStrategy == cmdutil . DryRunServer {
if err := o . DryRunVerifier . HasSupport ( mapping . GroupVersionKind ) ; err != nil {
return nil , err
}
}
actualObj , err = resource .
NewHelper ( client , mapping ) .
DryRun ( o . DryRunStrategy == cmdutil . DryRunServer ) .
2020-08-10 17:43:49 +00:00
WithFieldManager ( o . fieldManager ) .
Create ( o . Namespace , false , obj )
2019-01-12 04:58:27 +00:00
if err != nil {
return nil , err
}
2020-08-10 17:43:49 +00:00
} else {
if meta , err := meta . Accessor ( actualObj ) ; err == nil && o . EnforceNamespace {
meta . SetNamespace ( o . Namespace )
}
2019-01-12 04:58:27 +00:00
}
return & RunObject {
Object : actualObj ,
Mapping : mapping ,
} , nil
}
2019-09-27 21:51:53 +00:00
// ErrPodCompleted is returned by PodRunning or PodContainerRunning to indicate that
// the pod has already reached completed state.
var ErrPodCompleted = fmt . Errorf ( "pod ran to completion" )
// podCompleted returns true if the pod has run to completion, false if the pod has not yet
// reached running state, or an error in any other case.
func podCompleted ( event watch . Event ) ( bool , error ) {
switch event . Type {
case watch . Deleted :
return false , errors . NewNotFound ( schema . GroupResource { Resource : "pods" } , "" )
}
switch t := event . Object . ( type ) {
case * corev1 . Pod :
switch t . Status . Phase {
case corev1 . PodFailed , corev1 . PodSucceeded :
return true , nil
}
}
return false , nil
}
2020-08-10 17:43:49 +00:00
// podSucceeded returns true if the pod has run to completion, false if the pod has not yet
// reached running state, or an error in any other case.
func podSucceeded ( event watch . Event ) ( bool , error ) {
switch event . Type {
case watch . Deleted :
return false , errors . NewNotFound ( schema . GroupResource { Resource : "pods" } , "" )
}
switch t := event . Object . ( type ) {
case * corev1 . Pod :
return t . Status . Phase == corev1 . PodSucceeded , nil
}
return false , nil
}
2019-09-27 21:51:53 +00:00
// podRunningAndReady returns true if the pod is running and ready, false if the pod has not
// yet reached those states, returns ErrPodCompleted if the pod has run to completion, or
// an error in any other case.
func podRunningAndReady ( event watch . Event ) ( bool , error ) {
switch event . Type {
case watch . Deleted :
return false , errors . NewNotFound ( schema . GroupResource { Resource : "pods" } , "" )
}
switch t := event . Object . ( type ) {
case * corev1 . Pod :
switch t . Status . Phase {
case corev1 . PodFailed , corev1 . PodSucceeded :
return false , ErrPodCompleted
case corev1 . PodRunning :
conditions := t . Status . Conditions
if conditions == nil {
return false , nil
}
for i := range conditions {
if conditions [ i ] . Type == corev1 . PodReady &&
conditions [ i ] . Status == corev1 . ConditionTrue {
return true , nil
}
}
}
}
return false , nil
}