Add filter on namespaces

Author: jrauh01

cmd/icinga-kubernetes/main.go

@@ -66,26 +66,26 @@ func main() {
 		}
 	}
 
-	informers := kinformers.NewSharedInformerFactory(k, 0)
+	informers := kinformers.NewSharedInformerFactoryWithOptions(k, 0, kinformers.WithNamespace(flags.Namespace))
 
 	g, ctx := errgroup.WithContext(ctx)
 
 	g.Go(func() error {
 		return sync.NewSync(
 			db, schema.NewNode, informers.Core().V1().Nodes().Informer(), logs.GetChildLogger("Nodes"),
-		).Run(ctx)
+		).Run(ctx, flags.Namespace)
 	})
 
 	g.Go(func() error {
 		return sync.NewSync(
 			db, schema.NewNamespace, informers.Core().V1().Namespaces().Informer(), logs.GetChildLogger("Namespaces"),
-		).Run(ctx)
+		).Run(ctx, flags.Namespace)
 	})
 
 	g.Go(func() error {
 		return sync.NewSync(
 			db, schema.NewPod, informers.Core().V1().Pods().Informer(), logs.GetChildLogger("Pods"),
-		).Run(ctx)
+		).Run(ctx, flags.Namespace)
 	})
 
 	if err := g.Wait(); err != nil {

internal/flags.go

@@ -4,4 +4,7 @@ package internal
 type Flags struct {
 	// Config is the path to the config file
 	Config string `short:"c" long:"config" description:"path to config file" required:"true" default:"./config.yml"`
+
+	// Namespace is the namespace to sync
+	Namespace string `short:"n" long:"namespace" description:"namespace to sync" required:"true" default:""`
 }

pkg/schema/pod.go

@@ -16,3 +16,9 @@ func NewPod() contracts.Resource {
 func (p *Pod) Obtain(kobject kmetav1.Object) {
 	p.kmetaWithNamespace.Obtain(kobject)
 }
+
+func (p *Pod) Scope() any {
+	return &struct {
+		Namespace string
+	}{}
+}

pkg/sync/sync.go

@@ -15,7 +15,7 @@ import (
 )
 
 type Sync interface {
-	Run(context.Context) error
+	Run(context.Context, string) error
 }
 
 type sync struct {
@@ -39,12 +39,12 @@ func NewSync(
 	}
 }
 
-func (s *sync) Run(ctx context.Context) error {
+func (s *sync) Run(ctx context.Context, namespace string) error {
 	s.logger.Info("Starting sync")
 
 	s.logger.Debug("Warming up")
 
-	err := s.Warmup(ctx)
+	err := s.Warmup(ctx, namespace)
 	if err != nil {
 		return errors.Wrap(err, "warmup failed")
 	}
@@ -139,15 +139,17 @@ func (s *sync) Run(ctx context.Context) error {
 	return g.Wait()
 }
 
-func (s *sync) Warmup(ctx context.Context) error {
+func (s *sync) Warmup(ctx context.Context, namespace string) error {
 	g, ctx := errgroup.WithContext(ctx)
 
 	resource := s.factory()
 	entities, err := s.db.YieldAll(ctx, func() database.Entity {
 		return s.factory()
-	}, s.db.BuildSelectStmt(resource, resource.Fingerprint()), struct{}{})
+	}, s.db.BuildSelectStmt(resource, resource.Fingerprint()), struct{ Namespace string }{Namespace: namespace})
 	com.ErrgroupReceive(ctx, g, err)
 
+	fmt.Println(s.db.BuildSelectStmt(resource, resource.Fingerprint()))
+
 	g.Go(func() error {
 		for {
 			select {