-
Notifications
You must be signed in to change notification settings - Fork 18
/
sks_nodepool_scale.go
90 lines (75 loc) · 2.11 KB
/
sks_nodepool_scale.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
package cmd
import (
"errors"
"fmt"
"strconv"
exov2 "github.com/exoscale/egoscale/v2"
exoapi "github.com/exoscale/egoscale/v2/api"
"github.com/spf13/cobra"
)
var sksNodepoolScaleCmd = &cobra.Command{
Use: "scale CLUSTER-NAME|ID NODEPOOL-NAME|ID SIZE",
Short: "Scale a SKS cluster Nodepool size",
Long: `This command scales a SKS cluster Nodepool size up (growing) or down
(shrinking).
In case of a scale-down, operators should use the "exo sks nodepool evict"
variant, allowing them to specify which specific Nodes should be evicted from
the pool rather than leaving the decision to the SKS manager.`,
PreRunE: func(cmd *cobra.Command, args []string) error {
if len(args) != 3 {
cmdExitOnUsageError(cmd, "invalid arguments")
}
cmdSetZoneFlagFromDefault(cmd)
return cmdCheckRequiredFlags(cmd, []string{"zone"})
},
RunE: func(cmd *cobra.Command, args []string) error {
var (
c = args[0]
np = args[1]
nodepool *exov2.SKSNodepool
)
size, err := strconv.Atoi(args[2])
if err != nil {
return fmt.Errorf("invalid size %q", args[2])
}
if size <= 0 {
return errors.New("minimum Nodepool size is 1")
}
z, err := cmd.Flags().GetString("zone")
if err != nil {
return err
}
zone, err := getZoneByNameOrID(z)
if err != nil {
return fmt.Errorf("error retrieving zone: %s", err)
}
ctx := exoapi.WithEndpoint(gContext, exoapi.NewReqEndpoint(gCurrentAccount.Environment, zone.Name))
cluster, err := lookupSKSCluster(ctx, zone.Name, c)
if err != nil {
return err
}
for _, n := range cluster.Nodepools {
if n.ID == np || n.Name == np {
nodepool = n
break
}
}
if nodepool == nil {
return errors.New("Nodepool not found") // nolint:golint
}
decorateAsyncOperation(fmt.Sprintf("Scaling Nodepool %q...", np), func() {
err = cluster.ScaleNodepool(ctx, nodepool, int64(size))
})
if err != nil {
return err
}
if !gQuiet {
return output(showSKSNodepool(zone, cluster.ID, nodepool.ID))
}
return nil
},
}
func init() {
sksNodepoolScaleCmd.Flags().StringP("zone", "z", "", "SKS cluster zone")
sksNodepoolCmd.AddCommand(sksNodepoolScaleCmd)
}