mirror of https://github.com/kubernetes/kops.git
101 lines
3.0 KiB
Go
101 lines
3.0 KiB
Go
/*
|
|
Copyright 2022 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package scalewaymodel
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/scaleway/scaleway-sdk-go/scw"
|
|
"k8s.io/kops/pkg/model"
|
|
"k8s.io/kops/upup/pkg/fi"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/scaleway"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/scalewaytasks"
|
|
)
|
|
|
|
var commercialTypesWithBlockStorageOnly = []string{"PRO", "PLAY", "ENT"}
|
|
|
|
const defaultNodeRootVolumeSizeGB = 50
|
|
const defaultControlPlaneRootVolumeSizeGB = 20
|
|
|
|
// InstanceModelBuilder configures instances for the cluster
|
|
type InstanceModelBuilder struct {
|
|
*ScwModelContext
|
|
|
|
BootstrapScriptBuilder *model.BootstrapScriptBuilder
|
|
Lifecycle fi.Lifecycle
|
|
}
|
|
|
|
var _ fi.CloudupModelBuilder = &InstanceModelBuilder{}
|
|
|
|
func (b *InstanceModelBuilder) Build(c *fi.CloudupModelBuilderContext) error {
|
|
for _, ig := range b.InstanceGroups {
|
|
name := ig.Name
|
|
zone, err := scw.ParseZone(ig.Spec.Subnets[0])
|
|
if err != nil {
|
|
return fmt.Errorf("error building instance task for %q: %w", name, err)
|
|
}
|
|
|
|
userData, err := b.BootstrapScriptBuilder.ResourceNodeUp(c, ig)
|
|
if err != nil {
|
|
return fmt.Errorf("error building bootstrap script for %q: %w", name, err)
|
|
}
|
|
|
|
instanceTags := []string{
|
|
scaleway.TagClusterName + "=" + b.Cluster.Name,
|
|
scaleway.TagInstanceGroup + "=" + ig.Name,
|
|
}
|
|
for k, v := range b.CloudTags(b.ClusterName(), false) {
|
|
instanceTags = append(instanceTags, fmt.Sprintf("%s=%s", k, v))
|
|
}
|
|
|
|
instance := scalewaytasks.Instance{
|
|
Count: int(fi.ValueOf(ig.Spec.MinSize)),
|
|
Name: fi.PtrTo(name),
|
|
Lifecycle: b.Lifecycle,
|
|
Zone: fi.PtrTo(string(zone)),
|
|
CommercialType: fi.PtrTo(ig.Spec.MachineType),
|
|
Image: fi.PtrTo(ig.Spec.Image),
|
|
UserData: &userData,
|
|
Tags: instanceTags,
|
|
}
|
|
|
|
if ig.IsControlPlane() {
|
|
instance.Tags = append(instance.Tags, scaleway.TagNameRolePrefix+"="+scaleway.TagRoleControlPlane)
|
|
instance.Role = fi.PtrTo(scaleway.TagRoleControlPlane)
|
|
} else {
|
|
instance.Role = fi.PtrTo(scaleway.TagRoleWorker)
|
|
}
|
|
|
|
// If the instance's commercial type is one that has no local storage, we have to specify for the
|
|
// block storage volume a big enough size (default size is 10GB)
|
|
for _, commercialType := range commercialTypesWithBlockStorageOnly {
|
|
if strings.HasPrefix(ig.Spec.MachineType, commercialType) {
|
|
if ig.IsControlPlane() {
|
|
instance.VolumeSize = fi.PtrTo(defaultControlPlaneRootVolumeSizeGB)
|
|
} else {
|
|
instance.VolumeSize = fi.PtrTo(defaultNodeRootVolumeSizeGB)
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
c.AddTask(&instance)
|
|
}
|
|
return nil
|
|
}
|