Merge pull request #1 from CSdread/314-label-selector

initial pass at adding label selector to watch

Author: CSdread

pkg/sdk/api.go

@@ -53,7 +53,7 @@ func Watch(apiVersion, kind, namespace string, resyncPeriod int, opts ...watchOp
 	}
 	o := newWatchOp()
 	o.applyOpts(opts)
-	informer := NewInformer(resourcePluralName, namespace, resourceClient, resyncPeriod, collector, o.numWorkers)
+	informer := NewInformer(resourcePluralName, namespace, resourceClient, resyncPeriod, collector, o.numWorkers, o.labelSelector)
 	informers = append(informers, informer)
 }
 

pkg/sdk/informer.go

@@ -46,7 +46,7 @@ type informer struct {
 	numWorkers          int
 }
 
-func NewInformer(resourcePluralName, namespace string, resourceClient dynamic.ResourceInterface, resyncPeriod int, c *metrics.Collector, n int) Informer {
+func NewInformer(resourcePluralName, namespace string, resourceClient dynamic.ResourceInterface, resyncPeriod int, c *metrics.Collector, n int, labelSelector string) Informer {
 	i := &informer{
 		resourcePluralName: resourcePluralName,
 		queue:              workqueue.NewNamedRateLimitingQueue(workqueue.DefaultControllerRateLimiter(), resourcePluralName),
@@ -58,7 +58,7 @@ func NewInformer(resourcePluralName, namespace string, resourceClient dynamic.Re
 
 	resyncDuration := time.Duration(resyncPeriod) * time.Second
 	i.sharedIndexInformer = cache.NewSharedIndexInformer(
-		newListWatcherFromResourceClient(resourceClient), &unstructured.Unstructured{}, resyncDuration, cache.Indexers{},
+		newListWatcherFromResourceClient(resourceClient, labelSelector), &unstructured.Unstructured{}, resyncDuration, cache.Indexers{},
 	)
 	i.sharedIndexInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
 		AddFunc:    i.handleAddResourceEvent,
@@ -68,11 +68,17 @@ func NewInformer(resourcePluralName, namespace string, resourceClient dynamic.Re
 	return i
 }
 
-func newListWatcherFromResourceClient(resourceClient dynamic.ResourceInterface) *cache.ListWatch {
+func newListWatcherFromResourceClient(resourceClient dynamic.ResourceInterface, labelSelector string) *cache.ListWatch {
 	listFunc := func(options metav1.ListOptions) (runtime.Object, error) {
+		if labelSelector != "" {
+			options.LabelSelector = labelSelector
+		}
 		return resourceClient.List(options)
 	}
 	watchFunc := func(options metav1.ListOptions) (watch.Interface, error) {
+		if labelSelector != "" {
+			options.LabelSelector = labelSelector
+		}
 		return resourceClient.Watch(options)
 	}
 	return &cache.ListWatch{ListFunc: listFunc, WatchFunc: watchFunc}

pkg/sdk/watch-opt.go

@@ -16,7 +16,8 @@ package sdk
 
 // WatchOp wraps all the options for Watch().
 type watchOp struct {
-	numWorkers int
+	numWorkers    int
+	labelSelector string
 }
 
 // NewWatchOp create a new deafult WatchOp
@@ -47,3 +48,9 @@ func WithNumWorkers(numWorkers int) watchOption {
 		op.numWorkers = numWorkers
 	}
 }
+
+func WithLabelSelector(labelSelector string) watchOption {
+	return func(op *watchOp) {
+		op.labelSelector = labelSelector
+	}
+}