github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+incompatible/command/v6/v3_zdt_restart_command.go (about)

     1  package v6
     2  
     3  import (
     4  	"code.cloudfoundry.org/cli/actor/sharedaction"
     5  	"code.cloudfoundry.org/cli/actor/v3action"
     6  	"code.cloudfoundry.org/cli/api/cloudcontroller/ccversion"
     7  	"code.cloudfoundry.org/cli/command"
     8  	"code.cloudfoundry.org/cli/command/flag"
     9  	"code.cloudfoundry.org/cli/command/v6/shared"
    10  )
    11  
    12  //go:generate counterfeiter . V3ZeroDowntimeRestartActor
    13  
    14  type V3ZeroDowntimeRestartActor interface {
    15  	ZeroDowntimePollStart(appGUID string, warningsChannel chan<- v3action.Warnings) error
    16  	CreateDeployment(appGUID, dropletGUID string) (string, v3action.Warnings, error)
    17  
    18  	CloudControllerAPIVersion() string
    19  	GetApplicationByNameAndSpace(appName string, spaceGUID string) (v3action.Application, v3action.Warnings, error)
    20  	StartApplication(appGUID string) (v3action.Warnings, error)
    21  }
    22  
    23  type V3ZeroDowntimeRestartCommand struct {
    24  	RequiredArgs flag.AppName `positional-args:"yes"`
    25  	usage        interface{}  `usage:"CF_NAME v3-zdt-restart APP_NAME"`
    26  
    27  	UI          command.UI
    28  	Config      command.Config
    29  	SharedActor command.SharedActor
    30  	Actor       V3ZeroDowntimeRestartActor
    31  }
    32  
    33  func (cmd *V3ZeroDowntimeRestartCommand) Setup(config command.Config, ui command.UI) error {
    34  	cmd.UI = ui
    35  	cmd.Config = config
    36  	cmd.SharedActor = sharedaction.NewActor(config)
    37  
    38  	ccClient, _, err := shared.NewV3BasedClients(config, ui, true)
    39  	if err != nil {
    40  		return err
    41  	}
    42  	cmd.Actor = v3action.NewActor(ccClient, config, nil, nil)
    43  
    44  	return nil
    45  }
    46  
    47  func (cmd V3ZeroDowntimeRestartCommand) Execute(args []string) error {
    48  	cmd.UI.DisplayWarning(command.ExperimentalWarning)
    49  
    50  	err := command.MinimumCCAPIVersionCheck(cmd.Actor.CloudControllerAPIVersion(), ccversion.MinVersionZeroDowntimePushV3)
    51  	if err != nil {
    52  		return err
    53  	}
    54  
    55  	err = cmd.SharedActor.CheckTarget(true, true)
    56  	if err != nil {
    57  		return err
    58  	}
    59  
    60  	user, err := cmd.Config.CurrentUser()
    61  	if err != nil {
    62  		return err
    63  	}
    64  
    65  	app, warnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, cmd.Config.TargetedSpace().GUID)
    66  	cmd.UI.DisplayWarnings(warnings)
    67  	if err != nil {
    68  		return err
    69  	}
    70  
    71  	if app.Stopped() {
    72  		cmd.UI.DisplayTextWithFlavor("Starting app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.Username}}...", map[string]interface{}{
    73  			"AppName":   cmd.RequiredArgs.AppName,
    74  			"OrgName":   cmd.Config.TargetedOrganization().Name,
    75  			"SpaceName": cmd.Config.TargetedSpace().Name,
    76  			"Username":  user.Name,
    77  		})
    78  
    79  		warnings, err = cmd.Actor.StartApplication(app.GUID)
    80  		cmd.UI.DisplayWarnings(warnings)
    81  		if err != nil {
    82  			return err
    83  		}
    84  	} else {
    85  		cmd.UI.DisplayTextWithFlavor("Starting deployment for app {{.AppName}} in org {{.CurrentOrg}} / space {{.CurrentSpace}} as {{.CurrentUser}}...", map[string]interface{}{
    86  			"AppName":      cmd.RequiredArgs.AppName,
    87  			"CurrentSpace": cmd.Config.TargetedSpace().Name,
    88  			"CurrentOrg":   cmd.Config.TargetedOrganization().Name,
    89  			"CurrentUser":  user.Name,
    90  		})
    91  
    92  		_, warnings, err = cmd.Actor.CreateDeployment(app.GUID, "")
    93  		cmd.UI.DisplayWarnings(warnings)
    94  		if err != nil {
    95  			return err
    96  		}
    97  
    98  		cmd.UI.DisplayText("Waiting for app to start...")
    99  
   100  		warnings := make(chan v3action.Warnings)
   101  		done := make(chan bool)
   102  		go func() {
   103  			for {
   104  				select {
   105  				case message := <-warnings:
   106  					cmd.UI.DisplayWarnings(message) // untested
   107  				case <-done:
   108  					return
   109  				}
   110  			}
   111  		}()
   112  
   113  		err = cmd.Actor.ZeroDowntimePollStart(app.GUID, warnings)
   114  		done <- true
   115  		if err != nil {
   116  			return err
   117  		}
   118  	}
   119  
   120  	cmd.UI.DisplayOK()
   121  
   122  	return nil
   123  }