-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
windows.go
122 lines (107 loc) · 3.22 KB
/
windows.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
111
112
113
114
115
116
117
118
119
120
121
122
package nodebootstrap
import (
"encoding/base64"
"fmt"
"strconv"
"strings"
"github.com/kris-nova/logger"
api "github.com/weaveworks/eksctl/pkg/apis/eksctl.io/v1alpha5"
"github.com/weaveworks/eksctl/pkg/nodebootstrap/powershell"
"github.com/weaveworks/eksctl/pkg/nodebootstrap/utils"
)
type Windows struct {
clusterConfig *api.ClusterConfig
np api.NodePool
clusterDNS string
}
func NewWindowsBootstrapper(clusterConfig *api.ClusterConfig, np api.NodePool, clusterDNS string) *Windows {
return &Windows{
clusterConfig: clusterConfig,
np: np,
clusterDNS: clusterDNS,
}
}
func (b *Windows) UserData() (string, error) {
ng := b.np.BaseNodeGroup()
bootstrapCommands := append([]string{
`<powershell>
[string]$EKSBootstrapScriptFile = "$env:ProgramFiles\Amazon\EKS\Start-EKSBootstrap.ps1"`,
}, ng.PreBootstrapCommands...)
if ng.OverrideBootstrapCommand != nil {
bootstrapCommands = append(bootstrapCommands,
b.makeBootstrapParams(true),
*ng.OverrideBootstrapCommand,
)
} else {
bootstrapCommands = append(bootstrapCommands, fmt.Sprintf("& $EKSBootstrapScriptFile %s 3>&1 4>&1 5>&1 6>&1", b.makeBootstrapParams(false)))
}
bootstrapCommands = append(bootstrapCommands, "</powershell>")
userData := base64.StdEncoding.EncodeToString([]byte(strings.Join(bootstrapCommands, "\n")))
logger.Debug("user-data = %s", userData)
return userData, nil
}
func (b *Windows) makeBootstrapParams(hasBootstrapCommand bool) string {
params := []powershell.KeyValue{
{
Key: "EKSClusterName",
Value: b.clusterConfig.Metadata.Name,
},
{
Key: "APIServerEndpoint",
Value: b.clusterConfig.Status.Endpoint,
},
{
Key: "Base64ClusterCA",
Value: base64.StdEncoding.EncodeToString(b.clusterConfig.Status.CertificateAuthorityData),
},
{
Key: "ServiceCIDR",
Value: b.clusterConfig.Status.KubernetesNetworkConfig.ServiceIPv4CIDR,
},
}
if unmanaged, ok := b.np.(*api.NodeGroup); ok {
// DNSClusterIP is only configurable for self-managed nodegroups.
if b.clusterDNS != "" {
params = append(params, powershell.KeyValue{
Key: "DNSClusterIP",
Value: b.clusterDNS,
})
}
// ContainerRuntime is only configurable for self-managed nodegroups.
if unmanaged.ContainerRuntime != nil {
params = append(params, powershell.KeyValue{
Key: "ContainerRuntime",
Value: *unmanaged.ContainerRuntime,
})
}
}
kubeletOptions := b.makeKubeletOptions()
params = append(params, powershell.KeyValue{
Key: "KubeletExtraArgs",
Value: powershell.ToCLIArgs(kubeletOptions),
})
if hasBootstrapCommand {
return powershell.JoinVariables(powershell.FormatStringVariables(params), powershell.FormatHashTable(kubeletOptions, "KubeletExtraArgsMap"))
}
return powershell.FormatParams(params)
}
func (b *Windows) makeKubeletOptions() []powershell.KeyValue {
ng := b.np.BaseNodeGroup()
kubeletOptions := []powershell.KeyValue{
{
Key: "node-labels",
Value: formatLabels(ng.Labels),
},
{
Key: "register-with-taints",
Value: utils.FormatTaints(b.np.NGTaints()),
},
}
if ng.MaxPodsPerNode != 0 {
kubeletOptions = append(kubeletOptions, powershell.KeyValue{
Key: "max-pods",
Value: strconv.Itoa(ng.MaxPodsPerNode),
})
}
return kubeletOptions
}