github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+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.GetNewClientsAndConnectToCF(config, ui)
    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  }