mirror of https://github.com/kubernetes/kops.git
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
/*
|
|
Copyright 2020 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 components
|
|
|
|
import (
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
"k8s.io/kops/pkg/apis/kops"
|
|
"k8s.io/kops/upup/pkg/fi"
|
|
"k8s.io/kops/upup/pkg/fi/loader"
|
|
)
|
|
|
|
// NodeTerminationHandlerOptionsBuilder adds options for the node termination handler to the model.
|
|
type NodeTerminationHandlerOptionsBuilder struct {
|
|
*OptionsContext
|
|
}
|
|
|
|
var _ loader.OptionsBuilder = &NodeTerminationHandlerOptionsBuilder{}
|
|
|
|
func (b *NodeTerminationHandlerOptionsBuilder) BuildOptions(o interface{}) error {
|
|
clusterSpec := o.(*kops.ClusterSpec)
|
|
if clusterSpec.CloudProvider.AWS == nil {
|
|
return nil
|
|
}
|
|
if clusterSpec.CloudProvider.AWS.NodeTerminationHandler == nil {
|
|
clusterSpec.CloudProvider.AWS.NodeTerminationHandler = &kops.NodeTerminationHandlerSpec{}
|
|
}
|
|
nth := clusterSpec.CloudProvider.AWS.NodeTerminationHandler
|
|
if nth.Enabled == nil {
|
|
nth.Enabled = fi.PtrTo(true)
|
|
}
|
|
if nth.EnableSpotInterruptionDraining == nil {
|
|
nth.EnableSpotInterruptionDraining = fi.PtrTo(true)
|
|
}
|
|
if nth.EnableScheduledEventDraining == nil {
|
|
nth.EnableScheduledEventDraining = fi.PtrTo(true)
|
|
}
|
|
if nth.EnableRebalanceMonitoring == nil {
|
|
nth.EnableRebalanceMonitoring = fi.PtrTo(false)
|
|
}
|
|
if nth.EnableRebalanceDraining == nil {
|
|
nth.EnableRebalanceDraining = fi.PtrTo(false)
|
|
}
|
|
|
|
if nth.EnablePrometheusMetrics == nil {
|
|
nth.EnablePrometheusMetrics = fi.PtrTo(false)
|
|
}
|
|
|
|
if nth.ExcludeFromLoadBalancers == nil {
|
|
nth.ExcludeFromLoadBalancers = fi.PtrTo(true)
|
|
}
|
|
|
|
if nth.ManagedASGTag == nil {
|
|
nth.ManagedASGTag = fi.PtrTo("aws-node-termination-handler/managed")
|
|
}
|
|
|
|
if nth.CPURequest == nil {
|
|
defaultCPURequest := resource.MustParse("50m")
|
|
nth.CPURequest = &defaultCPURequest
|
|
}
|
|
|
|
if nth.MemoryRequest == nil {
|
|
defaultMemoryRequest := resource.MustParse("64Mi")
|
|
nth.MemoryRequest = &defaultMemoryRequest
|
|
}
|
|
|
|
if nth.Version == nil {
|
|
nth.Version = fi.PtrTo("v1.18.3")
|
|
}
|
|
|
|
return nil
|
|
}
|