-
Notifications
You must be signed in to change notification settings - Fork 927
/
run_task_command.go
102 lines (83 loc) · 3.17 KB
/
run_task_command.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package v6
import (
"fmt"
"code.cloudfoundry.org/cli/actor/sharedaction"
"code.cloudfoundry.org/cli/actor/v3action"
"code.cloudfoundry.org/cli/command"
"code.cloudfoundry.org/cli/command/flag"
"code.cloudfoundry.org/cli/command/v6/shared"
)
//go:generate counterfeiter . RunTaskActor
type RunTaskActor interface {
GetApplicationByNameAndSpace(appName string, spaceGUID string) (v3action.Application, v3action.Warnings, error)
RunTask(appGUID string, task v3action.Task) (v3action.Task, v3action.Warnings, error)
}
type RunTaskCommand struct {
RequiredArgs flag.RunTaskArgs `positional-args:"yes"`
Disk flag.Megabytes `short:"k" description:"Disk limit (e.g. 256M, 1024M, 1G)"`
Memory flag.Megabytes `short:"m" description:"Memory limit (e.g. 256M, 1024M, 1G)"`
Name string `long:"name" description:"Name to give the task (generated if omitted)"`
usage interface{} `usage:"CF_NAME run-task APP_NAME COMMAND [-k DISK] [-m MEMORY] [--name TASK_NAME]\n\nTIP:\n Use 'cf logs' to display the logs of the app and all its tasks. If your task name is unique, grep this command's output for the task name to view task-specific logs.\n\nEXAMPLES:\n CF_NAME run-task my-app \"bundle exec rake db:migrate\" --name migrate"`
relatedCommands interface{} `related_commands:"logs, tasks, terminate-task"`
UI command.UI
Config command.Config
SharedActor command.SharedActor
Actor RunTaskActor
}
func (cmd *RunTaskCommand) Setup(config command.Config, ui command.UI) error {
cmd.UI = ui
cmd.Config = config
cmd.SharedActor = sharedaction.NewActor(config)
client, _, err := shared.NewV3BasedClients(config, ui, true)
if err != nil {
return err
}
cmd.Actor = v3action.NewActor(client, config, nil, nil)
return nil
}
func (cmd RunTaskCommand) Execute(args []string) error {
err := cmd.SharedActor.CheckTarget(true, true)
if err != nil {
return err
}
space := cmd.Config.TargetedSpace()
user, err := cmd.Config.CurrentUser()
if err != nil {
return err
}
application, warnings, err := cmd.Actor.GetApplicationByNameAndSpace(cmd.RequiredArgs.AppName, space.GUID)
cmd.UI.DisplayWarnings(warnings)
if err != nil {
return err
}
cmd.UI.DisplayTextWithFlavor("Creating task for app {{.AppName}} in org {{.OrgName}} / space {{.SpaceName}} as {{.CurrentUser}}...", map[string]interface{}{
"AppName": cmd.RequiredArgs.AppName,
"OrgName": cmd.Config.TargetedOrganization().Name,
"SpaceName": space.Name,
"CurrentUser": user.Name,
})
inputTask := v3action.Task{
Command: cmd.RequiredArgs.Command,
}
if cmd.Name != "" {
inputTask.Name = cmd.Name
}
if cmd.Disk.IsSet {
inputTask.DiskInMB = cmd.Disk.Value
}
if cmd.Memory.IsSet {
inputTask.MemoryInMB = cmd.Memory.Value
}
task, warnings, err := cmd.Actor.RunTask(application.GUID, inputTask)
cmd.UI.DisplayWarnings(warnings)
if err != nil {
return err
}
cmd.UI.DisplayOK()
cmd.UI.DisplayText("Task has been submitted successfully for execution.")
cmd.UI.DisplayKeyValueTable("", [][]string{
{cmd.UI.TranslateText("task name:"), task.Name},
{cmd.UI.TranslateText("task id:"), fmt.Sprint(task.SequenceID)},
}, 3)
return nil
}