2018-01-09 18:57:14 +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 cmd
import (
"fmt"
"io"
"github.com/docker/distribution/reference"
"github.com/spf13/cobra"
2018-07-18 14:47:22 +00:00
"k8s.io/client-go/dynamic"
"github.com/golang/glog"
2018-01-09 18:57:14 +00:00
"k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/api/meta"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
2018-03-06 22:33:18 +00:00
utilerrors "k8s.io/apimachinery/pkg/util/errors"
2018-01-09 18:57:14 +00:00
"k8s.io/apimachinery/pkg/watch"
2018-07-18 14:47:22 +00:00
"k8s.io/kubernetes/pkg/api/legacyscheme"
2018-01-09 18:57:14 +00:00
api "k8s.io/kubernetes/pkg/apis/core"
coreclient "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/core/internalversion"
"k8s.io/kubernetes/pkg/kubectl"
"k8s.io/kubernetes/pkg/kubectl/cmd/templates"
cmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util"
2018-07-18 14:47:22 +00:00
"k8s.io/kubernetes/pkg/kubectl/genericclioptions"
"k8s.io/kubernetes/pkg/kubectl/genericclioptions/resource"
"k8s.io/kubernetes/pkg/kubectl/polymorphichelpers"
"k8s.io/kubernetes/pkg/kubectl/scheme"
2018-01-09 18:57:14 +00:00
"k8s.io/kubernetes/pkg/kubectl/util/i18n"
"k8s.io/kubernetes/pkg/util/interrupt"
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 = nginx -- 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 ) ' ` ) )
)
type RunObject struct {
Object runtime . Object
Mapping * meta . RESTMapping
}
2018-07-18 14:47:22 +00:00
type RunOptions struct {
PrintFlags * genericclioptions . PrintFlags
DeleteFlags * DeleteFlags
DeleteOptions * DeleteOptions
RecordFlags * genericclioptions . RecordFlags
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 : 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 )
2018-01-09 18:57:14 +00:00
cmd := & cobra . Command {
2018-03-06 22:33:18 +00:00
Use : "run NAME --image=image [--env=\"key=value\"] [--port=port] [--replicas=replicas] [--dry-run=bool] [--overrides=inline-json] [--command] -- [COMMAND] [args...]" ,
DisableFlagsInUseLine : true ,
2018-01-09 18:57:14 +00:00
Short : i18n . T ( "Run a particular image on the cluster" ) ,
Long : runLong ,
Example : runExample ,
Run : func ( cmd * cobra . Command , args [ ] string ) {
2018-07-18 14:47:22 +00:00
cmdutil . CheckErr ( o . Complete ( f , cmd ) )
cmdutil . CheckErr ( o . Run ( f , cmd , args ) )
2018-01-09 18:57:14 +00:00
} ,
}
2018-07-18 14:47:22 +00:00
o . DeleteFlags . AddFlags ( cmd )
o . PrintFlags . AddFlags ( cmd )
o . RecordFlags . AddFlags ( cmd )
addRunFlags ( cmd , o )
2018-01-09 18:57:14 +00:00
cmdutil . AddApplyAnnotationFlags ( cmd )
cmdutil . AddPodRunningTimeoutFlag ( cmd , defaultPodAttachTimeout )
return cmd
}
2018-07-18 14:47:22 +00:00
func addRunFlags ( cmd * cobra . Command , opt * RunOptions ) {
2018-01-09 18:57:14 +00:00
cmdutil . AddDryRunFlag ( cmd )
2018-07-18 14:47:22 +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." ) )
cmd . Flags ( ) . StringVar ( & opt . Image , "image" , opt . Image , i18n . T ( "The image for the container to run." ) )
2018-01-09 18:57:14 +00:00
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" )
2018-07-18 14:47:22 +00:00
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." ) )
2018-01-09 18:57:14 +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." )
2018-07-18 14:47:22 +00:00
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." )
2018-01-09 18:57:14 +00:00
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." ) )
2018-07-18 14:47:22 +00:00
cmd . Flags ( ) . BoolVar ( & opt . Expose , "expose" , opt . Expose , "If true, a public, external service is created for the container(s) which are run" )
2018-01-09 18:57:14 +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" ) )
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." ) )
2018-07-18 14:47:22 +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." ) )
}
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
o . DeleteOptions = deleteOpts
return nil
2018-01-09 18:57:14 +00:00
}
2018-07-18 14:47:22 +00:00
func ( o * RunOptions ) Run ( f cmdutil . Factory , cmd * cobra . Command , args [ ] string ) error {
2018-01-09 18:57:14 +00:00
// Let kubectl run follow rules for `--`, see #13004 issue
2018-07-18 14:47:22 +00:00
if len ( args ) == 0 || o . ArgsLenAtDash == 0 {
2018-01-09 18:57:14 +00:00
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
2018-07-18 14:47:22 +00:00
imageName := o . Image
2018-01-09 18:57:14 +00:00
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 )
}
2018-07-18 14:47:22 +00:00
if o . TTY && ! o . Interactive {
2018-01-09 18:57:14 +00:00
return cmdutil . UsageErrorf ( cmd , "-i/--stdin is required for containers with -t/--tty=true" )
}
replicas := cmdutil . GetFlagInt ( cmd , "replicas" )
2018-07-18 14:47:22 +00:00
if o . Interactive && replicas != 1 {
2018-01-09 18:57:14 +00:00
return cmdutil . UsageErrorf ( cmd , "-i/--stdin requires that replicas is 1, found %d" , replicas )
}
2018-07-18 14:47:22 +00:00
if o . Expose && len ( o . Port ) == 0 {
2018-01-09 18:57:14 +00:00
return cmdutil . UsageErrorf ( cmd , "--port must be set when exposing a service" )
}
2018-07-18 14:47:22 +00:00
namespace , _ , err := f . ToRawKubeConfigLoader ( ) . Namespace ( )
2018-01-09 18:57:14 +00:00
if err != nil {
return err
}
2018-07-18 14:47:22 +00:00
restartPolicy , err := getRestartPolicy ( cmd , o . Interactive )
2018-01-09 18:57:14 +00:00
if err != nil {
return err
}
if restartPolicy != api . RestartPolicyAlways && replicas != 1 {
return cmdutil . UsageErrorf ( cmd , "--restart=%s requires that --replicas=1, found %d" , restartPolicy , replicas )
}
remove := cmdutil . GetFlagBool ( cmd , "rm" )
2018-07-18 14:47:22 +00:00
if ! o . Attach && remove {
2018-01-09 18:57:14 +00:00
return cmdutil . UsageErrorf ( cmd , "--rm should only be used for attached containers" )
}
2018-07-18 14:47:22 +00:00
if o . Attach && o . DryRun {
2018-01-09 18:57:14 +00:00
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 . ClientSet ( )
if err != nil {
return err
}
2018-07-18 14:47:22 +00:00
generatorName := o . Generator
if len ( o . Schedule ) != 0 && len ( generatorName ) == 0 {
generatorName = cmdutil . CronJobV1Beta1GeneratorName
2018-01-09 18:57:14 +00:00
}
if len ( generatorName ) == 0 {
switch restartPolicy {
case api . RestartPolicyAlways :
2018-07-18 14:47:22 +00:00
generatorName = cmdutil . DeploymentAppsV1Beta1GeneratorName
2018-01-09 18:57:14 +00:00
case api . RestartPolicyOnFailure :
2018-07-18 14:47:22 +00:00
generatorName = cmdutil . JobV1GeneratorName
2018-01-09 18:57:14 +00:00
case api . RestartPolicyNever :
generatorName = cmdutil . RunPodV1GeneratorName
}
2018-07-18 14:47:22 +00:00
// Falling back because the generator was not provided and the default one could be unavailable.
generatorNameTemp , err := cmdutil . FallbackGeneratorNameIfNecessary ( generatorName , clientset . Discovery ( ) , o . ErrOut )
if err != nil {
return err
}
if generatorNameTemp != generatorName {
cmdutil . Warning ( o . ErrOut , generatorName , generatorNameTemp )
} else {
generatorName = generatorNameTemp
}
2018-01-09 18:57:14 +00:00
}
2018-07-18 14:47:22 +00:00
generators := cmdutil . GeneratorFn ( "run" )
2018-01-09 18:57:14 +00:00
generator , found := generators [ generatorName ]
if ! found {
return cmdutil . UsageErrorf ( cmd , "generator %q not found" , generatorName )
}
names := generator . ParamNames ( )
params := kubectl . MakeParams ( cmd , names )
params [ "name" ] = args [ 0 ]
if len ( args ) > 1 {
params [ "args" ] = args [ 1 : ]
}
params [ "env" ] = cmdutil . GetFlagStringArray ( cmd , "env" )
2018-07-18 14:47:22 +00:00
var createdObjects = [ ] * RunObject { }
runObject , err := o . createGeneratedObject ( f , cmd , generator , names , params , cmdutil . GetFlagString ( cmd , "overrides" ) , namespace )
2018-01-09 18:57:14 +00:00
if err != nil {
2018-07-18 14:47:22 +00:00
return err
2018-03-06 22:33:18 +00:00
} else {
2018-07-18 14:47:22 +00:00
createdObjects = append ( createdObjects , runObject )
2018-01-09 18:57:14 +00:00
}
2018-07-18 14:47:22 +00:00
allErrs := [ ] error { }
if o . Expose {
2018-01-09 18:57:14 +00:00
serviceGenerator := cmdutil . GetFlagString ( cmd , "service-generator" )
if len ( serviceGenerator ) == 0 {
return cmdutil . UsageErrorf ( cmd , "No service generator specified" )
}
2018-07-18 14:47:22 +00:00
serviceRunObject , err := o . generateService ( f , cmd , serviceGenerator , params , namespace )
2018-01-09 18:57:14 +00:00
if err != nil {
2018-03-06 22:33:18 +00:00
allErrs = append ( allErrs , err )
} else {
2018-07-18 14:47:22 +00:00
createdObjects = append ( createdObjects , serviceRunObject )
2018-01-09 18:57:14 +00:00
}
}
2018-07-18 14:47:22 +00:00
if o . Attach {
if remove {
defer o . removeCreatedObjects ( f , createdObjects )
}
2018-01-09 18:57:14 +00:00
opts := & AttachOptions {
StreamOptions : StreamOptions {
2018-07-18 14:47:22 +00:00
IOStreams : o . IOStreams ,
Stdin : o . Interactive ,
TTY : o . TTY ,
Quiet : o . Quiet ,
2018-01-09 18:57:14 +00:00
} ,
GetPodTimeout : timeout ,
CommandName : cmd . Parent ( ) . CommandPath ( ) + " attach" ,
Attach : & DefaultRemoteAttach { } ,
}
2018-07-18 14:47:22 +00:00
config , err := f . ToRESTConfig ( )
2018-01-09 18:57:14 +00:00
if err != nil {
return err
}
opts . Config = config
clientset , err := f . ClientSet ( )
if err != nil {
return err
}
opts . PodClient = clientset . Core ( )
2018-07-18 14:47:22 +00:00
attachablePod , err := polymorphichelpers . AttachablePodForObjectFn ( f , runObject . Object , opts . GetPodTimeout )
2018-01-09 18:57:14 +00:00
if err != nil {
return err
}
err = handleAttachPod ( f , clientset . Core ( ) , attachablePod . Namespace , attachablePod . Name , opts )
if err != nil {
return err
}
var pod * api . Pod
2018-07-18 14:47:22 +00:00
leaveStdinOpen := o . LeaveStdinOpen
2018-01-09 18:57:14 +00:00
waitForExitCode := ! leaveStdinOpen && restartPolicy == api . RestartPolicyNever
if waitForExitCode {
2018-03-06 22:33:18 +00:00
pod , err = waitForPod ( clientset . Core ( ) , attachablePod . Namespace , attachablePod . Name , kubectl . PodCompleted )
2018-01-09 18:57:14 +00:00
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 api . PodSucceeded :
return nil
case api . 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 )
}
}
2018-03-06 22:33:18 +00:00
if runObject != nil {
2018-07-18 14:47:22 +00:00
if err := o . PrintObj ( runObject . Object ) ; err != nil {
return err
2018-03-06 22:33:18 +00:00
}
2018-01-09 18:57:14 +00:00
}
2018-03-06 22:33:18 +00:00
return utilerrors . NewAggregate ( allErrs )
2018-01-09 18:57:14 +00:00
}
2018-07-18 14:47:22 +00:00
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 ( legacyscheme . Scheme ) .
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
}
2018-01-09 18:57:14 +00:00
// waitForPod watches the given pod until the exitCondition is true
func waitForPod ( podClient coreclient . PodsGetter , ns , name string , exitCondition watch . ConditionFunc ) ( * api . Pod , error ) {
w , err := podClient . Pods ( ns ) . Watch ( metav1 . SingleObject ( metav1 . ObjectMeta { Name : name } ) )
if err != nil {
return nil , err
}
intr := interrupt . New ( nil , w . Stop )
var result * api . Pod
err = intr . Run ( func ( ) error {
ev , err := watch . Until ( 0 , w , func ( ev watch . Event ) ( bool , error ) {
return exitCondition ( ev )
} )
if ev != nil {
result = ev . Object . ( * api . Pod )
}
return err
} )
// Fix generic not found error.
if err != nil && errors . IsNotFound ( err ) {
err = errors . NewNotFound ( api . Resource ( "pods" ) , name )
}
return result , err
}
func handleAttachPod ( f cmdutil . Factory , podClient coreclient . PodsGetter , ns , name string , opts * AttachOptions ) error {
2018-03-06 22:33:18 +00:00
pod , err := waitForPod ( podClient , ns , name , kubectl . PodRunningAndReady )
if err != nil && err != kubectl . ErrPodCompleted {
2018-01-09 18:57:14 +00:00
return err
}
if pod . Status . Phase == api . PodSucceeded || pod . Status . Phase == api . PodFailed {
return logOpts ( f , pod , opts )
}
opts . PodClient = podClient
opts . PodName = name
opts . Namespace = ns
// TODO: opts.Run sets opts.Err to nil, we need to find a better way
2018-07-18 14:47:22 +00:00
stderr := opts . ErrOut
2018-01-09 18:57:14 +00:00
if err := opts . Run ( ) ; err != nil {
fmt . Fprintf ( stderr , "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.
2018-07-18 14:47:22 +00:00
func logOpts ( restClientGetter genericclioptions . RESTClientGetter , pod * api . Pod , opts * AttachOptions ) error {
2018-01-09 18:57:14 +00:00
ctrName , err := opts . GetContainerName ( pod )
if err != nil {
return err
}
2018-07-18 14:47:22 +00:00
req , err := polymorphichelpers . LogsForObjectFn ( restClientGetter , pod , & api . PodLogOptions { Container : ctrName } , opts . GetPodTimeout )
2018-01-09 18:57:14 +00:00
if err != nil {
return err
}
readCloser , err := req . Stream ( )
if err != nil {
return err
}
defer readCloser . Close ( )
_ , err = io . Copy ( opts . Out , readCloser )
if err != nil {
return err
}
return nil
}
func getRestartPolicy ( cmd * cobra . Command , interactive bool ) ( api . RestartPolicy , error ) {
restart := cmdutil . GetFlagString ( cmd , "restart" )
if len ( restart ) == 0 {
if interactive {
return api . RestartPolicyOnFailure , nil
} else {
return api . RestartPolicyAlways , nil
}
}
switch api . RestartPolicy ( restart ) {
case api . RestartPolicyAlways :
return api . RestartPolicyAlways , nil
case api . RestartPolicyOnFailure :
return api . RestartPolicyOnFailure , nil
case api . RestartPolicyNever :
return api . RestartPolicyNever , nil
}
2018-03-06 22:33:18 +00:00
return "" , cmdutil . UsageErrorf ( cmd , "invalid restart policy: %s" , restart )
2018-01-09 18:57:14 +00:00
}
func verifyImagePullPolicy ( cmd * cobra . Command ) error {
pullPolicy := cmdutil . GetFlagString ( cmd , "image-pull-policy" )
switch api . PullPolicy ( pullPolicy ) {
case api . PullAlways , api . PullIfNotPresent , api . PullNever :
return nil
case "" :
return nil
}
return cmdutil . UsageErrorf ( cmd , "invalid image pull policy: %s" , pullPolicy )
}
2018-07-18 14:47:22 +00:00
func ( o * RunOptions ) generateService ( f cmdutil . Factory , cmd * cobra . Command , serviceGenerator string , paramsIn map [ string ] interface { } , namespace string ) ( * RunObject , error ) {
generators := cmdutil . GeneratorFn ( "expose" )
2018-01-09 18:57:14 +00:00
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
}
2018-07-18 14:47:22 +00:00
runObject , err := o . createGeneratedObject ( f , cmd , generator , names , params , cmdutil . GetFlagString ( cmd , "service-overrides" ) , namespace )
2018-01-09 18:57:14 +00:00
if err != nil {
return nil , err
}
2018-07-18 14:47:22 +00:00
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 , "---" )
2018-01-09 18:57:14 +00:00
}
return runObject , nil
}
2018-07-18 14:47:22 +00:00
func ( o * RunOptions ) createGeneratedObject ( f cmdutil . Factory , cmd * cobra . Command , generator kubectl . Generator , names [ ] kubectl . GeneratorParam , params map [ string ] interface { } , overrides , namespace string ) ( * RunObject , error ) {
2018-01-09 18:57:14 +00:00
err := kubectl . 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
}
2018-07-18 14:47:22 +00:00
mapper , err := f . ToRESTMapper ( )
if err != nil {
return nil , err
}
// run has compiled knowledge of the thing is is creating
gvks , _ , err := scheme . Scheme . ObjectKinds ( obj )
if err != nil {
return nil , err
}
mapping , err := mapper . RESTMapping ( gvks [ 0 ] . GroupKind ( ) , gvks [ 0 ] . Version )
2018-01-09 18:57:14 +00:00
if err != nil {
return nil , err
}
if len ( overrides ) > 0 {
2018-07-18 14:47:22 +00:00
codec := runtime . NewCodec ( scheme . DefaultJSONEncoder ( ) , scheme . Codecs . UniversalDecoder ( scheme . Scheme . PrioritizedVersionsAllGroups ( ) ... ) )
2018-01-09 18:57:14 +00:00
obj , err = cmdutil . Merge ( codec , obj , overrides )
if err != nil {
return nil , err
}
}
2018-07-18 14:47:22 +00:00
if err := o . Recorder . Record ( obj ) ; err != nil {
glog . V ( 4 ) . Infof ( "error recording current command: %v" , err )
2018-01-09 18:57:14 +00:00
}
2018-07-18 14:47:22 +00:00
actualObj := obj
if ! o . DryRun {
if err := kubectl . CreateOrUpdateAnnotation ( cmdutil . GetFlagBool ( cmd , cmdutil . ApplyAnnotationsFlag ) , obj , scheme . DefaultJSONEncoder ( ) ) ; err != nil {
2018-01-09 18:57:14 +00:00
return nil , err
}
2018-07-18 14:47:22 +00:00
client , err := f . ClientForMapping ( mapping )
2018-01-09 18:57:14 +00:00
if err != nil {
return nil , err
}
2018-07-18 14:47:22 +00:00
actualObj , err = resource . NewHelper ( client , mapping ) . Create ( namespace , false , obj )
2018-01-09 18:57:14 +00:00
if err != nil {
return nil , err
}
}
2018-07-18 14:47:22 +00:00
actualObj = cmdutil . AsDefaultVersionedOrOriginal ( actualObj , mapping )
2018-01-09 18:57:14 +00:00
return & RunObject {
2018-07-18 14:47:22 +00:00
Object : actualObj ,
2018-01-09 18:57:14 +00:00
Mapping : mapping ,
} , nil
}