add ownerUID index for controllerrevisions

Signed-off-by: shiyan2016 <[email protected]>
openkruise · Sep 9, 2024 · 0f20137 · 0f20137
1 parent 4918768
commit 0f20137
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/pkg/util/fieldindex/register.go b/pkg/util/fieldindex/register.go
@@ -23,6 +23,7 @@ import (
 	appsv1alpha1 "github.com/openkruise/kruise/apis/apps/v1alpha1"
 	utildiscovery "github.com/openkruise/kruise/pkg/util/discovery"
 
+	appsv1 "k8s.io/api/apps/v1"
 	batchv1 "k8s.io/api/batch/v1"
 	v1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -65,6 +66,11 @@ func RegisterFieldIndexes(c cache.Cache) error {
 		if err = c.IndexField(context.TODO(), &v1.PersistentVolumeClaim{}, IndexNameForOwnerRefUID, ownerIndexFunc); err != nil {
 			return
 		}
+		// controller revision ownerReference
+		if err = c.IndexField(context.TODO(), &appsv1.ControllerRevision{}, IndexNameForOwnerRefUID, ownerIndexFunc); err != nil {
+			return
+		}
+
 		// ImagePullJob ownerReference
 		if err = c.IndexField(context.TODO(), &appsv1alpha1.ImagePullJob{}, IndexNameForOwnerRefUID, ownerIndexFunc); err != nil {
 			return

diff --git a/pkg/util/history/controller_history.go b/pkg/util/history/controller_history.go
@@ -24,12 +24,15 @@ import (
 	apps "k8s.io/api/apps/v1"
 	"k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/fields"
 	"k8s.io/apimachinery/pkg/labels"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/client-go/util/retry"
 	"k8s.io/kubernetes/pkg/controller/history"
 	"sigs.k8s.io/controller-runtime/pkg/client"
+
+	"github.com/openkruise/kruise/pkg/util/fieldindex"
 )
 
 // NewHistory returns an an instance of Interface that uses client to communicate with the API Server and lister to list ControllerRevisions.
@@ -44,16 +47,14 @@ type realHistory struct {
 func (rh *realHistory) ListControllerRevisions(parent metav1.Object, selector labels.Selector) ([]*apps.ControllerRevision, error) {
 	// List all revisions in the namespace that match the selector
 	revisions := apps.ControllerRevisionList{}
-	err := rh.List(context.TODO(), &revisions, &client.ListOptions{Namespace: parent.GetNamespace(), LabelSelector: selector})
-	if err != nil {
-		return nil, err
+	opts := &client.ListOptions{
+		Namespace:     parent.GetNamespace(),
+		FieldSelector: fields.SelectorFromSet(fields.Set{fieldindex.IndexNameForOwnerRefUID: string(parent.GetUID())}),
 	}
+	err := rh.List(context.TODO(), &revisions, opts)
 	var owned []*apps.ControllerRevision
 	for i := range revisions.Items {
-		ref := metav1.GetControllerOf(&revisions.Items[i])
-		if ref == nil || ref.UID == parent.GetUID() {
-			owned = append(owned, &revisions.Items[i])
-		}
+		owned = append(owned, &revisions.Items[i])
 	}
 	return owned, err
 }