kubernetes/client: add more options for cache client

iawia002 · Mar 1, 2024 · f41db03 · f41db03
1 parent 40d14a2
commit f41db03
Showing 1 changed file with 41 additions and 10 deletions.
diff --git a/kubernetes/client/generic/client.go b/kubernetes/client/generic/client.go
@@ -10,19 +10,23 @@ import (
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/client-go/kubernetes/scheme"
 	"k8s.io/client-go/rest"
+	toolscache "k8s.io/client-go/tools/cache"
 	"sigs.k8s.io/controller-runtime/pkg/cache"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
 )
 
 // Options defines options needed to generate a client.
 type Options struct {
-	syncPeriod  *time.Duration
-	scheme      *runtime.Scheme
-	cacheReader bool
-	ctx         context.Context
-	httpClient  *http.Client
-	mapper      meta.RESTMapper
+	syncPeriod        *time.Duration
+	scheme            *runtime.Scheme
+	cacheReader       bool
+	ctx               context.Context
+	httpClient        *http.Client
+	mapper            meta.RESTMapper
+	defaultNamespaces map[string]cache.Config
+	defaultTransform  toolscache.TransformFunc
+	byObject          map[client.Object]cache.ByObject
 }
 
 // WithSyncPeriod sets the SyncPeriod time option.
@@ -71,6 +75,27 @@ func WithMapper(mapper meta.RESTMapper) func(opts *Options) {
 	}
 }
 
+// WithDefaultNamespaces sets the DefaultNamespaces for the cache client.
+func WithDefaultNamespaces(defaultNamespaces map[string]cache.Config) func(opts *Options) {
+	return func(opts *Options) {
+		opts.defaultNamespaces = defaultNamespaces
+	}
+}
+
+// WithDefaultTransform sets the DefaultTransform for the cache client.
+func WithDefaultTransform(defaultTransform toolscache.TransformFunc) func(opts *Options) {
+	return func(opts *Options) {
+		opts.defaultTransform = defaultTransform
+	}
+}
+
+// WithByObject sets the ByObject for the cache client.
+func WithByObject(byObject map[client.Object]cache.ByObject) func(opts *Options) {
+	return func(opts *Options) {
+		opts.byObject = byObject
+	}
+}
+
 // NewCache returns a controller-runtime cache client implementation.
 func NewCache(config *rest.Config, options ...func(*Options)) (cache.Cache, error) {
 	opts := &Options{
@@ -97,10 +122,13 @@ func NewCache(config *rest.Config, options ...func(*Options)) (cache.Cache, erro
 	}
 
 	cacheClient, err := cache.New(config, cache.Options{
-		HTTPClient: opts.httpClient,
-		Scheme:     opts.scheme,
-		Mapper:     opts.mapper,
-		SyncPeriod: opts.syncPeriod,
+		HTTPClient:        opts.httpClient,
+		Scheme:            opts.scheme,
+		Mapper:            opts.mapper,
+		SyncPeriod:        opts.syncPeriod,
+		DefaultNamespaces: opts.defaultNamespaces,
+		DefaultTransform:  opts.defaultTransform,
+		ByObject:          opts.byObject,
 	})
 	if err != nil {
 		return nil, err
@@ -151,6 +179,9 @@ func NewClient(config *rest.Config, options ...func(*Options)) (client.Client, e
 			WithMapper(opts.mapper),
 			WithSyncPeriod(opts.syncPeriod),
 			WithContext(opts.ctx),
+			WithDefaultNamespaces(opts.defaultNamespaces),
+			WithDefaultTransform(opts.defaultTransform),
+			WithByObject(opts.byObject),
 		)
 		if err != nil {
 			return nil, err