/
install_vpc_controllers.go
80 lines (62 loc) · 2.03 KB
/
install_vpc_controllers.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
package utils
import (
"context"
"time"
"github.com/kris-nova/logger"
"github.com/spf13/cobra"
"github.com/spf13/pflag"
"github.com/weaveworks/eksctl/pkg/eks"
"github.com/weaveworks/eksctl/pkg/utils/tasks"
api "github.com/weaveworks/eksctl/pkg/apis/eksctl.io/v1alpha5"
"github.com/weaveworks/eksctl/pkg/ctl/cmdutils"
)
func installWindowsVPCController(cmd *cmdutils.Cmd) {
cfg := api.NewClusterConfig()
cmd.ClusterConfig = cfg
cmd.SetDescription("install-vpc-controllers", "Install Windows VPC controller to support running Windows workloads", "")
cmd.CobraCommand.RunE = func(_ *cobra.Command, args []string) error {
cmd.NameArg = cmdutils.GetNameArg(args)
return doInstallWindowsVPCController(cmd)
}
cmd.FlagSetGroup.InFlagSet("General", func(fs *pflag.FlagSet) {
cmdutils.AddClusterFlagWithDeprecated(fs, cfg.Metadata)
cmdutils.AddRegionFlag(fs, &cmd.ProviderConfig)
cmdutils.AddConfigFileFlag(fs, &cmd.ClusterConfigFile)
cmdutils.AddApproveFlag(fs, cmd)
cmdutils.AddTimeoutFlag(fs, &cmd.ProviderConfig.WaitTimeout)
})
cmdutils.AddCommonFlagsForAWS(cmd, &cmd.ProviderConfig, false)
}
func doInstallWindowsVPCController(cmd *cmdutils.Cmd) error {
if err := cmdutils.NewMetadataLoader(cmd).Load(); err != nil {
return err
}
cfg := cmd.ClusterConfig
meta := cmd.ClusterConfig.Metadata
parentCtx := context.TODO()
ctl, err := cmd.NewProviderForExistingCluster(parentCtx)
if err != nil {
return err
}
logger.Info("using region %s", meta.Region)
if ok, err := ctl.CanUpdate(cfg); !ok {
return err
}
ctx, cancel := context.WithDeadline(parentCtx, time.Now().Add(cmd.ProviderConfig.WaitTimeout))
defer cancel()
vpcControllerTask := &eks.VPCControllerTask{
Info: "install Windows VPC controller",
Context: ctx,
ClusterConfig: cfg,
ClusterProvider: ctl,
PlanMode: cmd.Plan,
}
taskTree := &tasks.TaskTree{
Tasks: []tasks.Task{vpcControllerTask},
}
if errs := taskTree.DoAllSync(); len(errs) > 0 {
return errs[0]
}
cmdutils.LogPlanModeWarning(cmd.Plan)
return nil
}