github.com/jenspinney/cli@v6.42.1-0.20190207184520-7450c600020e+incompatible/command/v6/restart_command.go (about) 1 package v6 2 3 import ( 4 "code.cloudfoundry.org/cli/actor/sharedaction" 5 "code.cloudfoundry.org/cli/actor/v2action" 6 "code.cloudfoundry.org/cli/actor/v2v3action" 7 "code.cloudfoundry.org/cli/actor/v3action" 8 "code.cloudfoundry.org/cli/command" 9 "code.cloudfoundry.org/cli/command/flag" 10 "code.cloudfoundry.org/cli/command/v6/shared" 11 sharedV3 "code.cloudfoundry.org/cli/command/v6/shared" 12 "github.com/cloudfoundry/noaa/consumer" 13 log "github.com/sirupsen/logrus" 14 ) 15 16 //go:generate counterfeiter . RestartActor 17 18 type RestartActor interface { 19 GetApplicationByNameAndSpace(name string, spaceGUID string) (v2action.Application, v2action.Warnings, error) 20 GetApplicationSummaryByNameAndSpace(name string, spaceGUID string) (v2action.ApplicationSummary, v2action.Warnings, error) 21 RestartApplication(app v2action.Application, client v2action.NOAAClient) (<-chan *v2action.LogMessage, <-chan error, <-chan v2action.ApplicationStateChange, <-chan string, <-chan error) 22 } 23 24 type RestartCommand struct { 25 RequiredArgs flag.AppName `positional-args:"yes"` 26 usage interface{} `usage:"CF_NAME restart APP_NAME"` 27 relatedCommands interface{} `related_commands:"restage, restart-app-instance"` 28 envCFStagingTimeout interface{} `environmentName:"CF_STAGING_TIMEOUT" environmentDescription:"Max wait time for buildpack staging, in minutes" environmentDefault:"15"` 29 envCFStartupTimeout interface{} `environmentName:"CF_STARTUP_TIMEOUT" environmentDescription:"Max wait time for app instance startup, in minutes" environmentDefault:"5"` 30 31 UI command.UI 32 Config command.Config 33 SharedActor command.SharedActor 34 Actor RestartActor 35 ApplicationSummaryActor shared.ApplicationSummaryActor 36 NOAAClient *consumer.Consumer 37 } 38 39 func (cmd *RestartCommand) Setup(config command.Config, ui command.UI) error { 40 cmd.UI = ui 41 cmd.Config = config 42 sharedActor := sharedaction.NewActor(config) 43 cmd.SharedActor = sharedActor 44 45 ccClient, uaaClient, err := shared.NewClients(config, ui, true) 46 if err != nil { 47 return err 48 49 } 50 ccClientV3, _, err := sharedV3.NewV3BasedClients(config, ui, true, "") 51 if err != nil { 52 return err 53 } 54 55 v2Actor := v2action.NewActor(ccClient, uaaClient, config) 56 v3Actor := v3action.NewActor(ccClientV3, config, sharedActor, nil) 57 58 cmd.Actor = v2Actor 59 cmd.ApplicationSummaryActor = v2v3action.NewActor(v2Actor, v3Actor) 60 61 cmd.Actor = v2action.NewActor(ccClient, uaaClient, config) 62 cmd.ApplicationSummaryActor = v2v3action.NewActor(v2Actor, v3Actor) 63 cmd.NOAAClient = shared.NewNOAAClient(ccClient.DopplerEndpoint(), config, uaaClient, ui) 64 65 return nil 66 } 67 68 func (cmd RestartCommand) Execute(args []string) error { 69 err := cmd.SharedActor.CheckTarget(true, true) 70 if err != nil { 71 return err 72 } 73 74 user, err := cmd.Config.CurrentUser() 75 if err != nil { 76 return err 77 } 78 79 cmd.UI.DisplayTextWithFlavor("Restarting app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.CurrentUser}}...", 80 map[string]interface{}{ 81 "AppName": cmd.RequiredArgs.AppName, 82 "OrgName": cmd.Config.TargetedOrganization().Name, 83 "SpaceName": cmd.Config.TargetedSpace().Name, 84 "CurrentUser": user.Name, 85 }) 86 87 app, warnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID) 88 cmd.UI.DisplayWarnings(warnings) 89 if err != nil { 90 return err 91 } 92 93 messages, logErrs, appState, apiWarnings, errs := cmd.Actor.RestartApplication(app, cmd.NOAAClient) 94 err = shared.PollStart(cmd.UI, cmd.Config, messages, logErrs, appState, apiWarnings, errs) 95 if err != nil { 96 return err 97 } 98 99 cmd.UI.DisplayNewline() 100 log.WithField("v3_api_version", cmd.ApplicationSummaryActor.CloudControllerV3APIVersion()).Debug("using v3 for app display") 101 appSummary, v3Warnings, err := cmd.ApplicationSummaryActor.GetApplicationSummaryByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID, true) 102 cmd.UI.DisplayWarnings(v3Warnings) 103 if err != nil { 104 return err 105 } 106 shared.NewAppSummaryDisplayer2(cmd.UI).AppDisplay(appSummary, true) 107 return nil 108 }