Start workload controller if scanner is installed (#194)

Signed-off-by: Tamal Saha <[email protected]>

Author: tamalsaha

pkg/apiserver/apiserver.go

@@ -27,7 +27,6 @@ import (
 	scannerscheme "kubeops.dev/scanner/client/clientset/versioned/scheme"
 	identityinstall "kubeops.dev/ui-server/apis/identity/install"
 	identityv1alpha1 "kubeops.dev/ui-server/apis/identity/v1alpha1"
-	scannercontrollers "kubeops.dev/ui-server/pkg/controllers/scanner"
 	"kubeops.dev/ui-server/pkg/graph"
 	"kubeops.dev/ui-server/pkg/menu"
 	"kubeops.dev/ui-server/pkg/metricshandler"
@@ -187,6 +186,10 @@ func (c completedConfig) New(ctx context.Context) (*UIServer, error) {
 	log.SetLogger(klogr.New())
 	setupLog := log.Log.WithName("setup")
 
+	if !c.ExtraConfig.DisableImageCache {
+		shared.InitImageCache(c.ExtraConfig.CacheSize, c.ExtraConfig.CacheTTL)
+	}
+
 	cfg := c.ExtraConfig.ClientConfig
 	mgr, err := manager.New(cfg, manager.Options{
 		Scheme:                 Scheme,
@@ -240,17 +243,6 @@ func (c completedConfig) New(ctx context.Context) (*UIServer, error) {
 		os.Exit(1)
 	}
 
-	if !c.ExtraConfig.DisableImageCache {
-		shared.InitImageCache(c.ExtraConfig.CacheSize, c.ExtraConfig.CacheTTL)
-	}
-
-	if err = (&scannercontrollers.WorkloadReconciler{
-		Client: mgr.GetClient(),
-	}).SetupWithManager(mgr); err != nil {
-		setupLog.Error(err, "unable to create controller", "controller", "Workload")
-		os.Exit(1)
-	}
-
 	s := &UIServer{
 		GenericAPIServer: genericServer,
 		Manager:          mgr,

pkg/graph/setup.go

@@ -22,6 +22,9 @@ import (
 	"strings"
 	"time"
 
+	scannerapi "kubeops.dev/scanner/apis/scanner/v1alpha1"
+	scannercontrollers "kubeops.dev/ui-server/pkg/controllers/scanner"
+
 	"github.com/graphql-go/graphql"
 	"github.com/pkg/errors"
 	"gomodules.xyz/sets"
@@ -108,6 +111,15 @@ func SetupGraphReconciler(mgr manager.Manager) func(ctx context.Context) error {
 			}).SetupWithManager(mgr); err != nil {
 				return err
 			}
+
+			if rid.Group == scannerapi.SchemeGroupVersion.Group &&
+				rid.Kind == scannerapi.ResourceKindImageScanRequest {
+				if err := (&scannercontrollers.WorkloadReconciler{
+					Client: mgr.GetClient(),
+				}).SetupWithManager(mgr); err != nil {
+					return err
+				}
+			}
 		}
 		return nil
 	}