github.com/franc20/ayesa_sap@v7.0.0-beta.28.0.20200124003224-302d4d52fa6c+incompatible/command/v6/restart_command.go (about)

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