WIP: reuse RESTMapper

Author: justinsb

pkg/patterns/declarative/pkg/watch/dynamic.go

@@ -27,7 +27,6 @@ import (
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/apimachinery/pkg/watch"
 	"k8s.io/client-go/dynamic"
-	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/event"
 	"sigs.k8s.io/controller-runtime/pkg/log"
@@ -36,18 +35,15 @@ import (
 // WatchDelay is the time between a Watch being dropped and attempting to resume it
 const WatchDelay = 30 * time.Second
 
-func NewDynamicWatch(mgr ctrl.Manager) (*dynamicWatch, chan event.GenericEvent, error) {
-	dw := &dynamicWatch{events: make(chan event.GenericEvent)}
-
-	restMapper := mgr.GetRESTMapper()
-
-	client, err := dynamic.NewForConfig(mgr.GetConfig())
-	if err != nil {
-		return nil, nil, err
+// NewDynamicWatch constructs a watcher for unstructured objects.
+// Deprecated: avoid using directly; will move to internal in future.
+func NewDynamicWatch(restMapper meta.RESTMapper, client dynamic.Interface) (*dynamicWatch, chan event.GenericEvent, error) {
+	dw := &dynamicWatch{
+		events:     make(chan event.GenericEvent),
+		restMapper: restMapper,
+		client:     client,
 	}
 
-	dw.restMapper = restMapper
-	dw.client = client
 	return dw, dw.events, nil
 }
 

pkg/patterns/declarative/watch.go

@@ -21,11 +21,14 @@ import (
 	"fmt"
 	"sync"
 
+	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/labels"
 	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/client-go/dynamic"
 	"k8s.io/client-go/rest"
 	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
 	"sigs.k8s.io/controller-runtime/pkg/controller"
 	"sigs.k8s.io/controller-runtime/pkg/handler"
 	"sigs.k8s.io/controller-runtime/pkg/log"
@@ -85,7 +88,31 @@ func WatchChildren(options WatchChildrenOptions) (chan struct{}, error) {
 		return nil, fmt.Errorf("labelMaker is required to scope watches")
 	}
 
-	dw, events, err := watch.NewDynamicWatch(*options.RESTConfig)
+	if options.RESTConfig == nil {
+		if options.Manager != nil {
+			options.RESTConfig = options.Manager.GetConfig()
+		} else {
+			return nil, fmt.Errorf("RESTConfig or Manager should be set")
+		}
+	}
+
+	var restMapper meta.RESTMapper
+	if options.Manager != nil {
+		restMapper = options.Manager.GetRESTMapper()
+	} else {
+		rm, err := apiutil.NewDiscoveryRESTMapper(options.RESTConfig)
+		if err != nil {
+			return nil, err
+		}
+		restMapper = rm
+	}
+
+	client, err := dynamic.NewForConfig(options.RESTConfig)
+	if err != nil {
+		return nil, err
+	}
+
+	dw, events, err := watch.NewDynamicWatch(restMapper, client)
 	if err != nil {
 		return nil, fmt.Errorf("creating dynamic watch: %v", err)
 	}