Skip to content

Commit 66caedd

Browse files
committed
✨ Migrate controllerutil AddFinalizer and RemoveFinalizer to controllerutil.Object and remove *WithError functions
1 parent 2f1457b commit 66caedd

File tree

2 files changed

+2
-38
lines changed

2 files changed

+2
-38
lines changed

pkg/controller/controllerutil/controllerutil.go

Lines changed: 2 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ import (
2222

2323
"k8s.io/apimachinery/pkg/api/equality"
2424
"k8s.io/apimachinery/pkg/api/errors"
25-
"k8s.io/apimachinery/pkg/api/meta"
2625
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
2726
"k8s.io/apimachinery/pkg/runtime"
2827
"k8s.io/apimachinery/pkg/runtime/schema"
@@ -238,7 +237,7 @@ func mutate(f MutateFn, key client.ObjectKey, obj runtime.Object) error {
238237
type MutateFn func() error
239238

240239
// AddFinalizer accepts a metav1 object and adds the provided finalizer if not present.
241-
func AddFinalizer(o metav1.Object, finalizer string) {
240+
func AddFinalizer(o Object, finalizer string) {
242241
f := o.GetFinalizers()
243242
for _, e := range f {
244243
if e == finalizer {
@@ -248,19 +247,8 @@ func AddFinalizer(o metav1.Object, finalizer string) {
248247
o.SetFinalizers(append(f, finalizer))
249248
}
250249

251-
// AddFinalizerWithError tries to convert a runtime object to a metav1 object and add the provided finalizer.
252-
// It returns an error if the provided object cannot provide an accessor.
253-
func AddFinalizerWithError(o runtime.Object, finalizer string) error {
254-
m, err := meta.Accessor(o)
255-
if err != nil {
256-
return err
257-
}
258-
AddFinalizer(m, finalizer)
259-
return nil
260-
}
261-
262250
// RemoveFinalizer accepts a metav1 object and removes the provided finalizer if present.
263-
func RemoveFinalizer(o metav1.Object, finalizer string) {
251+
func RemoveFinalizer(o Object, finalizer string) {
264252
f := o.GetFinalizers()
265253
for i := 0; i < len(f); i++ {
266254
if f[i] == finalizer {
@@ -271,17 +259,6 @@ func RemoveFinalizer(o metav1.Object, finalizer string) {
271259
o.SetFinalizers(f)
272260
}
273261

274-
// RemoveFinalizerWithError tries to convert a runtime object to a metav1 object and remove the provided finalizer.
275-
// It returns an error if the provided object cannot provide an accessor.
276-
func RemoveFinalizerWithError(o runtime.Object, finalizer string) error {
277-
m, err := meta.Accessor(o)
278-
if err != nil {
279-
return err
280-
}
281-
RemoveFinalizer(m, finalizer)
282-
return nil
283-
}
284-
285262
// ContainsFinalizer checks a metav1 object that the provided finalizer is present.
286263
func ContainsFinalizer(o Object, finalizer string) bool {
287264
f := o.GetFinalizers()

pkg/controller/controllerutil/controllerutil_test.go

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -407,21 +407,8 @@ var _ = Describe("Controllerutil", func() {
407407
})
408408

409409
Describe("Finalizers", func() {
410-
var obj runtime.Object = &errRuntimeObj{}
411410
var deploy *appsv1.Deployment
412411

413-
Describe("AddFinalizerWithError", func() {
414-
It("should return an error if object can't provide accessor", func() {
415-
Expect(controllerutil.AddFinalizerWithError(obj, testFinalizer)).To(HaveOccurred())
416-
})
417-
})
418-
419-
Describe("RemoveFinalizerWithError", func() {
420-
It("should return an error if object can't provide accessor", func() {
421-
Expect(controllerutil.RemoveFinalizerWithError(obj, testFinalizer)).To(HaveOccurred())
422-
})
423-
})
424-
425412
Describe("AddFinalizer", func() {
426413
deploy = &appsv1.Deployment{
427414
ObjectMeta: metav1.ObjectMeta{

0 commit comments

Comments
 (0)