workflow-cli/parser/autoscale.go at a0a3ef34c22179f5f4eccc5c4ab349118572df39 · drycc/workflow-cli · GitHub

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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package parser

import (
	docopt "github.com/docopt/docopt-go"
	"github.com/drycc/workflow-cli/cmd"
)

// Autoscale displays all relevant commands for `drycc autoscale`.
func Autoscale(argv []string, cmdr cmd.Commander) error {
	usage := `
Valid commands for autoscale:

autoscale:list   list autoscale options of an application
autoscale:set    turn on autoscale for an app
autoscale:unset  turn off autoscale for an app

Use 'drycc help [command]' to learn more.
`

	switch argv[0] {
	case "autoscale:list":
		return autoscaleList(argv, cmdr)
	case "autoscale:set":
		return autoscaleSet(argv, cmdr)
	case "autoscale:unset":
		return autoscaleUnset(argv, cmdr)
	default:
		if printHelp(argv, usage) {
			return nil
		}

		if argv[0] == "autoscale" {
			argv[0] = "autoscale:list"
			return autoscaleList(argv, cmdr)
		}

		PrintUsage(cmdr)
		return nil
	}
}

func autoscaleList(argv []string, cmdr cmd.Commander) error {
	usage := `
Prints a list of autoscale options for the application.

Usage: drycc autoscale:list [options]

Options:
  -a --app=<app>
    the uniquely identifiable name for the application.
`

	args, err := docopt.ParseArgs(usage, argv, "")

	if err != nil {
		return err
	}

	return cmdr.AutoscaleList(safeGetString(args, "--app"))
}

func autoscaleSet(argv []string, cmdr cmd.Commander) error {
	usage := `
Set autoscale option per process type for an app.

Usage: drycc autoscale:set <process-type> --min=<min> --max=<max> --cpu-percent=<percent> [options]

Arguments:
  <process-type>
    the process type to add to the application's autoscale settings.
  --min=<min>
	minimum replicas to keep around
  --max=<max>
	max replicas to scale up to
  --cpu-percent=<cpu-percent>
	target CPU utilization

Options:
  -a --app=<app>
    the uniquely identifiable name of the application.
`

	args, err := docopt.ParseArgs(usage, argv, "")

	if err != nil {
		return err
	}

	processType := args["<process-type>"].(string)
	app := safeGetString(args, "--app")
	min := safeGetInt(args, "--min")
	max := safeGetInt(args, "--max")
	CPUPercent := safeGetInt(args, "--cpu-percent")

	return cmdr.AutoscaleSet(app, processType, min, max, CPUPercent)
}

func autoscaleUnset(argv []string, cmdr cmd.Commander) error {
	usage := `
Unset autoscale per process type for an app.

Usage: drycc autoscale:unset <process-type> [options]

Arguments:
  <process-type>
    the process type to remove from the application's autoscale settings.

Options:
  -a --app=<app>
    the uniquely identifiable name of the application.
`

	args, err := docopt.ParseArgs(usage, argv, "")

	if err != nil {
		return err
	}

	processType := args["<process-type>"].(string)
	app := safeGetString(args, "--app")

	return cmdr.AutoscaleUnset(app, processType)
}