-
Notifications
You must be signed in to change notification settings - Fork 929
/
create_org_quota_command.go
110 lines (91 loc) · 4.31 KB
/
create_org_quota_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
103
104
105
106
107
108
109
110
package v7
import (
"code.cloudfoundry.org/cli/actor/sharedaction"
"code.cloudfoundry.org/cli/actor/v7action"
"code.cloudfoundry.org/cli/api/cloudcontroller/ccerror"
"code.cloudfoundry.org/cli/command"
"code.cloudfoundry.org/cli/command/flag"
"code.cloudfoundry.org/cli/command/v7/shared"
"code.cloudfoundry.org/cli/types"
"code.cloudfoundry.org/clock"
)
//go:generate counterfeiter . CreateOrgQuotaActor
type CreateOrgQuotaActor interface {
CreateOrganizationQuota(name string, limits v7action.QuotaLimits) (v7action.Warnings, error)
}
type CreateOrgQuotaCommand struct {
RequiredArgs flag.OrganizationQuota `positional-args:"yes"`
NumAppInstances flag.IntegerLimit `short:"a" description:"Total number of application instances. (Default: unlimited)."`
PaidServicePlans bool `long:"allow-paid-service-plans" description:"Allow provisioning instances of paid service plans. (Default: disallowed)."`
PerProcessMemory flag.MemoryWithUnlimited `short:"i" description:"Maximum amount of memory a process can have (e.g. 1024M, 1G, 10G). (Default: unlimited)."`
TotalMemory flag.MemoryWithUnlimited `short:"m" description:"Total amount of memory all processes can have (e.g. 1024M, 1G, 10G). -1 represents an unlimited amount. (Default: 0)."`
TotalRoutes flag.IntegerLimit `short:"r" description:"Total number of routes. -1 represents an unlimited amount. (Default: 0)."`
TotalReservedPorts flag.IntegerLimit `long:"reserved-route-ports" description:"Maximum number of routes that may be created with ports. -1 represents an unlimited amount. (Default: 0)."`
TotalServiceInstances flag.IntegerLimit `short:"s" description:"Total number of service instances. -1 represents an unlimited amount. (Default: 0)."`
usage interface{} `usage:"CF_NAME create-org-quota ORG_QUOTA [-m TOTAL_MEMORY] [-i INSTANCE_MEMORY] [-r ROUTES] [-s SERVICE_INSTANCES] [-a APP_INSTANCES] [--allow-paid-service-plans] [--reserved-route-ports RESERVED_ROUTE_PORTS]"`
relatedCommands interface{} `related_commands:"create-org, org-quotas, set-org-quota"`
UI command.UI
Config command.Config
Actor CreateOrgQuotaActor
SharedActor command.SharedActor
}
func (cmd *CreateOrgQuotaCommand) Setup(config command.Config, ui command.UI) error {
cmd.UI = ui
cmd.Config = config
sharedActor := sharedaction.NewActor(config)
cmd.SharedActor = sharedActor
ccClient, uaaClient, err := shared.GetNewClientsAndConnectToCF(config, ui, "")
if err != nil {
return err
}
cmd.Actor = v7action.NewActor(ccClient, config, sharedActor, uaaClient, clock.NewClock())
return nil
}
func (cmd CreateOrgQuotaCommand) Execute(args []string) error {
err := cmd.SharedActor.CheckTarget(false, false)
if err != nil {
return err
}
user, err := cmd.Config.CurrentUser()
if err != nil {
return err
}
orgQuotaName := cmd.RequiredArgs.OrganizationQuotaName
cmd.UI.DisplayTextWithFlavor("Creating org quota {{.OrganizationQuotaName}} as {{.User}}...",
map[string]interface{}{
"User": user.Name,
"OrganizationQuotaName": orgQuotaName,
})
warnings, err := cmd.Actor.CreateOrganizationQuota(orgQuotaName, v7action.QuotaLimits{
TotalMemoryInMB: convertMegabytesFlagToNullInt(cmd.TotalMemory),
PerProcessMemoryInMB: convertMegabytesFlagToNullInt(cmd.PerProcessMemory),
TotalInstances: convertIntegerLimitFlagToNullInt(cmd.NumAppInstances),
PaidServicesAllowed: &cmd.PaidServicePlans,
TotalServiceInstances: convertIntegerLimitFlagToNullInt(cmd.TotalServiceInstances),
TotalRoutes: convertIntegerLimitFlagToNullInt(cmd.TotalRoutes),
TotalReservedPorts: convertIntegerLimitFlagToNullInt(cmd.TotalReservedPorts),
})
cmd.UI.DisplayWarnings(warnings)
if _, ok := err.(ccerror.QuotaAlreadyExists); ok {
cmd.UI.DisplayWarning(err.Error())
cmd.UI.DisplayOK()
return nil
}
if err != nil {
return err
}
cmd.UI.DisplayOK()
return nil
}
func convertMegabytesFlagToNullInt(flag flag.MemoryWithUnlimited) *types.NullInt {
if !flag.IsSet {
return nil
}
return &types.NullInt{IsSet: true, Value: flag.Value}
}
func convertIntegerLimitFlagToNullInt(flag flag.IntegerLimit) *types.NullInt {
if !flag.IsSet {
return nil
}
return &types.NullInt{IsSet: true, Value: flag.Value}
}