Skip to content

Commit e5eb800

Browse files
yash97yash97
authored
Node termination cleaner refactor and adding node id from node object (#567)
* fetching instance id from k8s node object * refactored cninode controller, removed calling finalizer manager, handling all patches locally. * removing finalizer manager method * adding context and separate go routine to cleanup node resources
1 parent 35f491e commit e5eb800

37 files changed

+680
-310
lines changed

‎controllers/crds/cninode_controller.go‎

Lines changed: 80 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,9 @@ import (
2525
"github.com/aws/amazon-vpc-resource-controller-k8s/pkg/utils"
2626
"github.com/go-logr/logr"
2727
"github.com/prometheus/client_golang/prometheus"
28+
"golang.org/x/sync/semaphore"
2829
v1 "k8s.io/api/core/v1"
29-
"k8s.io/apimachinery/pkg/api/errors"
30+
apierrors "k8s.io/apimachinery/pkg/api/errors"
3031
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3132
"k8s.io/apimachinery/pkg/runtime"
3233
"k8s.io/apimachinery/pkg/types"
@@ -35,6 +36,7 @@ import (
3536
ctrl "sigs.k8s.io/controller-runtime"
3637
"sigs.k8s.io/controller-runtime/pkg/client"
3738
"sigs.k8s.io/controller-runtime/pkg/controller"
39+
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
3840
"sigs.k8s.io/controller-runtime/pkg/metrics"
3941
)
4042

@@ -75,7 +77,8 @@ type CNINodeReconciler struct {
7577
clusterName string
7678
vpcId string
7779
finalizerManager k8s.FinalizerManager
78-
newResourceCleaner func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string) cleanup.ResourceCleaner
80+
deletePool *semaphore.Weighted
81+
newResourceCleaner func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string, log logr.Logger) cleanup.ResourceCleaner
7982
}
8083

8184
func NewCNINodeReconciler(
@@ -88,7 +91,8 @@ func NewCNINodeReconciler(
8891
clusterName string,
8992
vpcId string,
9093
finalizerManager k8s.FinalizerManager,
91-
newResourceCleaner func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string) cleanup.ResourceCleaner,
94+
maxConcurrentWorkers int,
95+
newResourceCleaner func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string, log logr.Logger) cleanup.ResourceCleaner,
9296
) *CNINodeReconciler {
9397
return &CNINodeReconciler{
9498
Client: client,
@@ -100,6 +104,7 @@ func NewCNINodeReconciler(
100104
clusterName: clusterName,
101105
vpcId: vpcId,
102106
finalizerManager: finalizerManager,
107+
deletePool: semaphore.NewWeighted(int64(maxConcurrentWorkers)),
103108
newResourceCleaner: newResourceCleaner,
104109
}
105110
}
@@ -118,7 +123,7 @@ func (r *CNINodeReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ct
118123
nodeFound := true
119124
node := &v1.Node{}
120125
if err := r.Client.Get(ctx, req.NamespacedName, node); err != nil {
121-
if errors.IsNotFound(err) {
126+
if apierrors.IsNotFound(err) {
122127
nodeFound = false
123128
} else {
124129
r.log.Error(err, "failed to get the node object in CNINode reconciliation, will retry")
@@ -128,66 +133,50 @@ func (r *CNINodeReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ct
128133
}
129134

130135
if cniNode.GetDeletionTimestamp().IsZero() {
131-
shouldPatch := false
132136
cniNodeCopy := cniNode.DeepCopy()
133-
// Add cluster name tag if it does not exist
134-
val, ok := cniNode.Spec.Tags[config.VPCCNIClusterNameKey]
135-
if !ok || val != r.clusterName {
136-
if len(cniNodeCopy.Spec.Tags) != 0 {
137-
cniNodeCopy.Spec.Tags[config.VPCCNIClusterNameKey] = r.clusterName
138-
} else {
139-
cniNodeCopy.Spec.Tags = map[string]string{
140-
config.VPCCNIClusterNameKey: r.clusterName,
141-
}
142-
}
143-
shouldPatch = true
144-
}
145-
// if node exists, get & add OS label if it does not exist on CNINode
146-
if nodeFound {
147-
nodeLabelOS := node.ObjectMeta.Labels[config.NodeLabelOS]
148-
val, ok = cniNode.ObjectMeta.Labels[config.NodeLabelOS]
149-
if !ok || val != nodeLabelOS {
150-
if len(cniNodeCopy.ObjectMeta.Labels) != 0 {
151-
cniNodeCopy.ObjectMeta.Labels[config.NodeLabelOS] = nodeLabelOS
152-
} else {
153-
cniNodeCopy.ObjectMeta.Labels = map[string]string{
154-
config.NodeLabelOS: nodeLabelOS,
155-
}
156-
}
157-
shouldPatch = true
158-
}
159-
}
137+
shouldPatch, err := r.ensureTagsAndLabels(cniNodeCopy, node)
138+
shouldPatch = controllerutil.AddFinalizer(cniNodeCopy, config.NodeTerminationFinalizer) || shouldPatch
160139

161140
if shouldPatch {
162-
r.log.Info("patching CNINode to add required fields Tags and Labels", "cninode", cniNode.Name)
163-
return ctrl.Result{}, r.Client.Patch(ctx, cniNodeCopy, client.MergeFromWithOptions(cniNode, client.MergeFromWithOptimisticLock{}))
164-
}
165-
166-
// Add finalizer if it does not exist
167-
if err := r.finalizerManager.AddFinalizers(ctx, cniNode, config.NodeTerminationFinalizer); err != nil {
168-
r.log.Error(err, "failed to add finalizer on CNINode, will retry", "cniNode", cniNode.Name, "finalizer", config.NodeTerminationFinalizer)
169-
return ctrl.Result{}, err
141+
r.log.Info("patching CNINode to add fields Tags, Labels and finalizer", "cninode", cniNode.Name)
142+
if err := r.Client.Patch(ctx, cniNodeCopy, client.MergeFromWithOptions(cniNode, client.MergeFromWithOptimisticLock{})); err != nil {
143+
if apierrors.IsConflict(err) {
144+
r.log.Info("failed to update cninode", "cninode", cniNode.Name, "error", err)
145+
return ctrl.Result{Requeue: true}, nil
146+
}
147+
return ctrl.Result{}, err
148+
}
170149
}
171-
return ctrl.Result{}, nil
172-
150+
return ctrl.Result{}, err
173151
} else { // CNINode is marked for deletion
174152
if !nodeFound {
175153
// node is also deleted, proceed with running the cleanup routine and remove the finalizer
176-
177154
// run cleanup for Linux nodes only
178155
if val, ok := cniNode.ObjectMeta.Labels[config.NodeLabelOS]; ok && val == config.OSLinux {
179156
r.log.Info("running the finalizer routine on cniNode", "cniNode", cniNode.Name)
180157
// run cleanup when node id is present
181158
if nodeID, ok := cniNode.Spec.Tags[config.NetworkInterfaceNodeIDKey]; ok && nodeID != "" {
182-
if err := r.newResourceCleaner(nodeID, r.eC2Wrapper, r.vpcId).DeleteLeakedResources(); err != nil {
183-
r.log.Error(err, "failed to cleanup resources during node termination")
184-
ec2API.NodeTerminationENICleanupFailure.Inc()
159+
if !r.deletePool.TryAcquire(1) {
160+
r.log.Info("d, will requeue request")
161+
return ctrl.Result{RequeueAfter: 10 * time.Second}, nil
185162
}
163+
go func(nodeID string) {
164+
defer r.deletePool.Release(1)
165+
childCtx, cancel := context.WithTimeout(ctx, config.NodeTerminationTimeout)
166+
defer cancel()
167+
if err := r.newResourceCleaner(nodeID, r.eC2Wrapper, r.vpcId, r.log).DeleteLeakedResources(childCtx); err != nil {
168+
r.log.Error(err, "failed to cleanup resources during node termination")
169+
ec2API.NodeTerminationENICleanupFailure.Inc()
170+
}
171+
}(nodeID)
186172
}
187173
}
188174

189-
if err := r.finalizerManager.RemoveFinalizers(ctx, cniNode, config.NodeTerminationFinalizer); err != nil {
175+
if err := r.removeFinalizer(ctx, cniNode, config.NodeTerminationFinalizer); err != nil {
190176
r.log.Error(err, "failed to remove finalizer on CNINode, will retry", "cniNode", cniNode.Name, "finalizer", config.NodeTerminationFinalizer)
177+
if apierrors.IsConflict(err) {
178+
return ctrl.Result{Requeue: true}, nil
179+
}
191180
return ctrl.Result{}, err
192181
}
193182
return ctrl.Result{}, nil
@@ -207,7 +196,7 @@ func (r *CNINodeReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ct
207196
Spec: cniNode.Spec,
208197
}
209198

210-
if err := r.finalizerManager.RemoveFinalizers(ctx, cniNode, config.NodeTerminationFinalizer); err != nil {
199+
if err := r.removeFinalizer(ctx, cniNode, config.NodeTerminationFinalizer); err != nil {
211200
r.log.Error(err, "failed to remove finalizer on CNINode, will retry")
212201
return ctrl.Result{}, err
213202
}
@@ -252,7 +241,7 @@ func (r *CNINodeReconciler) waitTillCNINodeDeleted(nameSpacedCNINode types.Names
252241
oldCNINode := &v1alpha1.CNINode{}
253242

254243
return wait.PollUntilContextTimeout(context.TODO(), 500*time.Millisecond, time.Second*3, true, func(ctx context.Context) (bool, error) {
255-
if err := r.Client.Get(ctx, nameSpacedCNINode, oldCNINode); err != nil && errors.IsNotFound(err) {
244+
if err := r.Client.Get(ctx, nameSpacedCNINode, oldCNINode); err != nil && apierrors.IsNotFound(err) {
256245
return true, nil
257246
}
258247
return false, nil
@@ -266,3 +255,45 @@ func (r *CNINodeReconciler) createCNINodeFromObj(ctx context.Context, newCNINode
266255
return r.Client.Create(ctx, newCNINode)
267256
})
268257
}
258+
259+
func (r *CNINodeReconciler) ensureTagsAndLabels(cniNode *v1alpha1.CNINode, node *v1.Node) (bool, error) {
260+
shouldPatch := false
261+
var err error
262+
if cniNode.Spec.Tags == nil {
263+
cniNode.Spec.Tags = make(map[string]string)
264+
}
265+
// add cluster name tag if it does not exist
266+
if cniNode.Spec.Tags[config.VPCCNIClusterNameKey] != r.clusterName {
267+
cniNode.Spec.Tags[config.VPCCNIClusterNameKey] = r.clusterName
268+
shouldPatch = true
269+
}
270+
if node != nil {
271+
var nodeID string
272+
nodeID, err = utils.GetNodeID(node)
273+
274+
if nodeID != "" && cniNode.Spec.Tags[config.NetworkInterfaceNodeIDKey] != nodeID {
275+
cniNode.Spec.Tags[config.NetworkInterfaceNodeIDKey] = nodeID
276+
shouldPatch = true
277+
}
278+
279+
// add node label if it does not exist
280+
if cniNode.ObjectMeta.Labels == nil {
281+
cniNode.ObjectMeta.Labels = make(map[string]string)
282+
}
283+
if cniNode.ObjectMeta.Labels[config.NodeLabelOS] != node.ObjectMeta.Labels[config.NodeLabelOS] {
284+
cniNode.ObjectMeta.Labels[config.NodeLabelOS] = node.ObjectMeta.Labels[config.NodeLabelOS]
285+
shouldPatch = true
286+
}
287+
}
288+
return shouldPatch, err
289+
}
290+
291+
func (r *CNINodeReconciler) removeFinalizer(ctx context.Context, cniNode *v1alpha1.CNINode, finalizer string) error {
292+
cniNodeCopy := cniNode.DeepCopy()
293+
294+
if controllerutil.RemoveFinalizer(cniNodeCopy, finalizer) {
295+
r.log.Info("removing finalizer for cninode", "name", cniNode.GetName(), "finalizer", finalizer)
296+
return r.Client.Patch(ctx, cniNodeCopy, client.MergeFromWithOptions(cniNode, client.MergeFromWithOptimisticLock{}))
297+
}
298+
return nil
299+
}

‎controllers/crds/cninode_controller_test.go‎

Lines changed: 44 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,10 @@ import (
1111
ec2API "github.com/aws/amazon-vpc-resource-controller-k8s/pkg/aws/ec2/api"
1212
"github.com/aws/amazon-vpc-resource-controller-k8s/pkg/aws/ec2/api/cleanup"
1313
"github.com/aws/amazon-vpc-resource-controller-k8s/pkg/config"
14+
"github.com/go-logr/logr"
1415
"github.com/golang/mock/gomock"
1516
"github.com/stretchr/testify/assert"
17+
"golang.org/x/sync/semaphore"
1618
corev1 "k8s.io/api/core/v1"
1719
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
1820
"k8s.io/apimachinery/pkg/runtime"
@@ -37,6 +39,9 @@ var (
3739
config.NodeLabelOS: "linux",
3840
},
3941
},
42+
Spec: corev1.NodeSpec{
43+
ProviderID: "aws:///us-west-2a/i-0123456789abcdef0",
44+
},
4045
}
4146
reconcileRequest = reconcile.Request{
4247
NamespacedName: types.NamespacedName{
@@ -57,6 +62,7 @@ func NewCNINodeMock(ctrl *gomock.Controller, mockObjects ...client.Object) *CNIN
5762
log: zap.New(),
5863
clusterName: mockClusterName,
5964
vpcId: "vpc-000000000000",
65+
deletePool: semaphore.NewWeighted(10),
6066
},
6167
}
6268
}
@@ -80,7 +86,7 @@ func TestCNINodeReconcile(t *testing.T) {
8086
asserts func(reconcile.Result, error, *v1alpha1.CNINode)
8187
}{
8288
{
83-
name: "verify clusterName tag and labels are added if missing",
89+
name: "verify clusterName, instanceID, os label are added if missing",
8490
args: args{
8591
mockNode: mockNodeWithLabel,
8692
mockCNINode: &v1alpha1.CNINode{
@@ -94,7 +100,7 @@ func TestCNINodeReconcile(t *testing.T) {
94100
assert.NoError(t, err)
95101
assert.Equal(t, res, reconcile.Result{})
96102
assert.Equal(t, cniNode.Labels, map[string]string{config.NodeLabelOS: "linux"})
97-
assert.Equal(t, cniNode.Spec.Tags, map[string]string{config.VPCCNIClusterNameKey: mockClusterName})
103+
assert.Equal(t, cniNode.Spec.Tags, map[string]string{config.VPCCNIClusterNameKey: mockClusterName, config.NetworkInterfaceNodeIDKey: "i-0123456789abcdef0"})
98104
},
99105
},
100106
{
@@ -113,14 +119,10 @@ func TestCNINodeReconcile(t *testing.T) {
113119
},
114120
},
115121
prepare: func(f *fields) {
116-
f.mockCNINode.Reconciler.newResourceCleaner = func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string) cleanup.ResourceCleaner {
122+
f.mockCNINode.Reconciler.newResourceCleaner = func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string, log logr.Logger) cleanup.ResourceCleaner {
117123
return f.mockResourceCleaner
118124
}
119-
f.mockResourceCleaner.EXPECT().DeleteLeakedResources().Times(0)
120-
121-
f.mockFinalizerManager.EXPECT().
122-
RemoveFinalizers(gomock.Any(), gomock.Any(), config.NodeTerminationFinalizer).
123-
Return(nil)
125+
f.mockResourceCleaner.EXPECT().DeleteLeakedResources(gomock.Any()).Times(0)
124126
},
125127
asserts: func(res reconcile.Result, err error, cniNode *v1alpha1.CNINode) {
126128
assert.NoError(t, err)
@@ -142,27 +144,50 @@ func TestCNINodeReconcile(t *testing.T) {
142144
},
143145
Spec: v1alpha1.CNINodeSpec{
144146
Tags: map[string]string{
145-
config.NetworkInterfaceNodeIDKey: "i-1234567890",
147+
config.NetworkInterfaceNodeIDKey: "i-0123456789abcdef0",
146148
},
147149
},
148150
},
149151
},
150152
prepare: func(f *fields) {
151-
f.mockCNINode.Reconciler.newResourceCleaner = func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string) cleanup.ResourceCleaner {
152-
assert.Equal(t, "i-1234567890", nodeID)
153+
f.mockCNINode.Reconciler.newResourceCleaner = func(nodeID string, eC2Wrapper ec2API.EC2Wrapper, vpcID string, log logr.Logger) cleanup.ResourceCleaner {
154+
assert.Equal(t, "i-0123456789abcdef0", nodeID)
153155
return f.mockResourceCleaner
154156
}
155-
f.mockResourceCleaner.EXPECT().DeleteLeakedResources().Times(1).Return(nil)
156-
f.mockFinalizerManager.EXPECT().
157-
RemoveFinalizers(gomock.Any(), gomock.Any(), config.NodeTerminationFinalizer).
158-
Return(nil)
157+
f.mockResourceCleaner.EXPECT().DeleteLeakedResources(gomock.Any()).Times(1).Return(nil)
159158

160159
},
161160
asserts: func(res reconcile.Result, err error, cniNode *v1alpha1.CNINode) {
162161
assert.NoError(t, err)
163162
assert.Equal(t, res, reconcile.Result{})
164163
},
165164
},
165+
{
166+
name: "verify finalizer is added when labels and tags are present",
167+
args: args{
168+
mockNode: mockNodeWithLabel,
169+
mockCNINode: &v1alpha1.CNINode{
170+
ObjectMeta: metav1.ObjectMeta{
171+
Name: mockName,
172+
Labels: map[string]string{
173+
config.NodeLabelOS: "linux",
174+
},
175+
},
176+
Spec: v1alpha1.CNINodeSpec{
177+
Tags: map[string]string{
178+
config.VPCCNIClusterNameKey: mockClusterName,
179+
config.NetworkInterfaceNodeIDKey: "i-0123456789abcdef0",
180+
},
181+
},
182+
},
183+
},
184+
prepare: nil,
185+
asserts: func(res reconcile.Result, err error, cniNode *v1alpha1.CNINode) {
186+
assert.NoError(t, err)
187+
assert.Equal(t, res, reconcile.Result{})
188+
assert.Contains(t, cniNode.Finalizers, config.NodeTerminationFinalizer)
189+
},
190+
},
166191
}
167192
for _, tt := range tests {
168193
t.Run(tt.name, func(t *testing.T) {
@@ -188,8 +213,10 @@ func TestCNINodeReconcile(t *testing.T) {
188213
res, err := mock.Reconciler.Reconcile(context.Background(), reconcileRequest)
189214

190215
cniNode := &v1alpha1.CNINode{}
191-
getErr := mock.Reconciler.Client.Get(context.Background(), reconcileRequest.NamespacedName, cniNode)
192-
assert.NoError(t, getErr)
216+
if tt.args.mockCNINode.GetDeletionTimestamp() == nil {
217+
getErr := mock.Reconciler.Client.Get(context.Background(), reconcileRequest.NamespacedName, cniNode)
218+
assert.NoError(t, getErr)
219+
}
193220

194221
if tt.asserts != nil {
195222
tt.asserts(res, err, cniNode)

‎go.mod‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ require (
2727
github.com/prometheus/common v0.62.0
2828
github.com/stretchr/testify v1.10.0
2929
go.uber.org/zap v1.27.0
30+
golang.org/x/sync v0.13.0
3031
golang.org/x/time v0.11.0
3132
gomodules.xyz/jsonpatch/v2 v2.4.0
3233
k8s.io/api v0.33.0
@@ -50,7 +51,6 @@ require (
5051
github.com/gorilla/websocket v1.5.4-0.20250319132907-e064f32e3674 // indirect
5152
github.com/mxk/go-flowrate v0.0.0-20140419014527-cca7078d478f // indirect
5253
github.com/x448/float16 v0.8.4 // indirect
53-
golang.org/x/sync v0.13.0 // indirect
5454
gopkg.in/evanphx/json-patch.v4 v4.12.0 // indirect
5555
sigs.k8s.io/randfill v1.0.0 // indirect
5656
)

‎main.go‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -439,6 +439,7 @@ func main() {
439439
clusterName,
440440
vpcID,
441441
finalizerManager,
442+
maxNodeConcurrentReconciles,
442443
cleanup.NewNodeResourceCleaner,
443444
).SetupWithManager(mgr, maxNodeConcurrentReconciles)); err != nil {
444445
setupLog.Error(err, "unable to create controller", "controller", "CNINode")

0 commit comments

Comments
 (0)