-
Notifications
You must be signed in to change notification settings - Fork 473
/
nodes.go
171 lines (148 loc) · 6.52 KB
/
nodes.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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
// Copyright 2020 SAP SE or an SAP affiliate company. All rights reserved. This file is licensed under the Apache Software License, v. 2 except as noted otherwise in the LICENSE file
//
// 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 worker
import (
"context"
"fmt"
"time"
machinev1alpha1 "github.com/gardener/machine-controller-manager/pkg/apis/machine/v1alpha1"
"github.com/go-logr/logr"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/types"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/log"
"github.com/gardener/gardener/extensions/pkg/controller/healthcheck"
gardencorev1beta1 "github.com/gardener/gardener/pkg/apis/core/v1beta1"
)
const (
// AnnotationKeyNotManagedByMCM is a constant for an annotation on the node resource that indicates that
// the node is not handled by MCM.
AnnotationKeyNotManagedByMCM = "node.machine.sapcloud.io/not-managed-by-mcm"
)
// DefaultHealthChecker all the information for the Worker HealthCheck.
// This check assumes that the MachineControllerManager (https://github.com/gardener/machine-controller-manager) has been
// deployed by the Worker extension controller.
type DefaultHealthChecker struct {
logger logr.Logger
// Needs to be set by actuator before calling the Check function
seedClient client.Client
// make sure shoot client is instantiated
shootClient client.Client
// scaleUpProgressingThreshold is the progressing threshold when the health check detects a scale-up situation.
scaleUpProgressingThreshold *time.Duration
// scaleDownProgressingThreshold is the progressing threshold when the health check detects a scale-down situation.
scaleDownProgressingThreshold *time.Duration
}
// NewNodesChecker is a health check function which performs certain checks about the nodes registered in the cluster.
// It implements the healthcheck.HealthCheck interface.
func NewNodesChecker() *DefaultHealthChecker {
scaleUpProgressingThreshold := 5 * time.Minute
scaleDownProgressingThreshold := 15 * time.Minute
return &DefaultHealthChecker{
scaleUpProgressingThreshold: &scaleUpProgressingThreshold,
scaleDownProgressingThreshold: &scaleDownProgressingThreshold,
}
}
// WithScaleUpProgressingThreshold sets the scaleUpProgressingThreshold property.
func (h *DefaultHealthChecker) WithScaleUpProgressingThreshold(d time.Duration) *DefaultHealthChecker {
h.scaleUpProgressingThreshold = &d
return h
}
// WithScaleDownProgressingThreshold sets the scaleDownProgressingThreshold property.
func (h *DefaultHealthChecker) WithScaleDownProgressingThreshold(d time.Duration) *DefaultHealthChecker {
h.scaleDownProgressingThreshold = &d
return h
}
// InjectSeedClient injects the seed client.
func (h *DefaultHealthChecker) InjectSeedClient(seedClient client.Client) {
h.seedClient = seedClient
}
// InjectShootClient injects the shoot client.
func (h *DefaultHealthChecker) InjectShootClient(shootClient client.Client) {
h.shootClient = shootClient
}
// SetLoggerSuffix injects the logger.
func (h *DefaultHealthChecker) SetLoggerSuffix(provider, extension string) {
h.logger = log.Log.WithName(fmt.Sprintf("%s-%s-healthcheck-nodes", provider, extension))
}
// DeepCopy clones the healthCheck struct by making a copy and returning the pointer to that new copy.
// Actually, it does not perform a *deep* copy.
func (h *DefaultHealthChecker) DeepCopy() healthcheck.HealthCheck {
shallowCopy := *h
return &shallowCopy
}
// Check executes the health check.
func (h *DefaultHealthChecker) Check(ctx context.Context, request types.NamespacedName) (*healthcheck.SingleCheckResult, error) {
machineDeploymentList := &machinev1alpha1.MachineDeploymentList{}
if err := h.seedClient.List(ctx, machineDeploymentList, client.InNamespace(request.Namespace)); err != nil {
err := fmt.Errorf("unable to check nodes. Failed to list machine deployments in namespace %q: %w", request.Namespace, err)
h.logger.Error(err, "Health check failed")
return nil, err
}
nodeList := &corev1.NodeList{}
if err := h.shootClient.List(ctx, nodeList); err != nil {
err := fmt.Errorf("unable to check nodes. Failed to list shoot nodes: %w", err)
h.logger.Error(err, "Health check failed")
return nil, err
}
var (
readyNodes int
registeredNodes = len(nodeList.Items)
desiredMachines = getDesiredMachineCount(machineDeploymentList.Items)
nodeNotManagedByMCM int
)
for _, node := range nodeList.Items {
if metav1.HasAnnotation(node.ObjectMeta, AnnotationKeyNotManagedByMCM) && node.Annotations[AnnotationKeyNotManagedByMCM] == "1" {
nodeNotManagedByMCM++
continue
}
if node.Spec.Unschedulable {
continue
}
for _, condition := range node.Status.Conditions {
if condition.Type == corev1.NodeReady && condition.Status == corev1.ConditionTrue {
readyNodes++
}
}
}
// only nodes that are managed by MCM is considered
registeredNodes = registeredNodes - nodeNotManagedByMCM
machineList := &machinev1alpha1.MachineList{}
if registeredNodes != desiredMachines || readyNodes != desiredMachines {
if err := h.seedClient.List(ctx, machineList, client.InNamespace(request.Namespace)); err != nil {
err := fmt.Errorf("unable to check nodes. Failed to list machines in namespace %q: %w", request.Namespace, err)
h.logger.Error(err, "Health check failed")
return nil, err
}
}
for _, deployment := range machineDeploymentList.Items {
for _, failedMachine := range deployment.Status.FailedMachines {
err := fmt.Errorf("machine %q failed: %s", failedMachine.Name, failedMachine.LastOperation.Description)
h.logger.Error(err, "Health check failed")
return &healthcheck.SingleCheckResult{
Status: gardencorev1beta1.ConditionFalse,
Detail: err.Error(),
}, nil
}
}
if isHealthy, err := checkMachineDeploymentsHealthy(machineDeploymentList.Items); !isHealthy {
h.logger.Error(err, "Health check failed")
return &healthcheck.SingleCheckResult{
Status: gardencorev1beta1.ConditionFalse,
Detail: err.Error(),
}, nil
}
return &healthcheck.SingleCheckResult{Status: gardencorev1beta1.ConditionTrue}, nil
}