cache mlflow crd status

Signed-off-by: Alyssa Goins <agoins@redhat.com>

Author: alyssacgoins

main.go

@@ -43,6 +43,8 @@ import (
 	"k8s.io/apimachinery/pkg/selection"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	clientgoscheme "k8s.io/client-go/kubernetes/scheme"
+	"k8s.io/client-go/discovery"
+	"k8s.io/client-go/rest"
 
 	// Import all Kubernetes client auth plugins (e.g. Azure, GCP, OIDC, etc.)
 	// to ensure that exec-entrypoint and run can make use of them.
@@ -153,6 +155,15 @@ func initConfig(configPath string) error {
 	return nil
 }
 
+func isCRDAvailable(cfg *rest.Config, groupVersion string) bool {
+	dc, err := discovery.NewDiscoveryClientForConfig(cfg)
+	if err != nil {
+		return false
+	}
+	_, err = dc.ServerResourcesForGroupVersion(groupVersion)
+	return err == nil
+}
+
 func main() {
 	var metricsAddr string
 	var enableLeaderElection bool
@@ -207,7 +218,9 @@ func main() {
 	dspSelector := labels.NewSelector().Add(*dspLabelReq)
 	dspFilter := cache.ByObject{Label: dspSelector}
 
-	mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{
+	restCfg := ctrl.GetConfigOrDie()
+
+	mgrOpts := ctrl.Options{
 		Scheme: scheme,
 		Metrics: metricsserver.Options{
 			BindAddress: metricsAddr,
@@ -247,22 +260,32 @@ func main() {
 				// (odh-trusted-ca-bundle, service-ca secrets) that lack the dsp-version
 				// label. Strip data payloads instead of filtering by label so the
 				// informer still detects changes to those external resources.
-			&corev1.ConfigMap{}: {Transform: stripConfigMapData},
-			&corev1.Secret{}:    {Transform: stripSecretData},
+				&corev1.ConfigMap{}: {Transform: stripConfigMapData},
+				&corev1.Secret{}:    {Transform: stripSecretData},
 			},
 		},
 		// ConfigMap and Secret client reads bypass the cache entirely for
 		// direct API reads, since the cache strips data payloads.
 		Client: client.Options{
 			Cache: &client.CacheOptions{
-			DisableFor: []client.Object{
-				&corev1.ConfigMap{},
-				&corev1.Secret{},
-				&mlflowv1.MLflow{},
-			},
+				DisableFor: []client.Object{
+					&corev1.ConfigMap{},
+					&corev1.Secret{},
+				},
 			},
 		},
-	})
+	}
+
+	if isCRDAvailable(restCfg, mlflowv1.GroupVersion.String()) {
+		mlflowSyncPeriod := 5 * time.Minute
+		mgrOpts.Cache.ByObject[&mlflowv1.MLflow{}] = cache.ByObject{SyncPeriod: &mlflowSyncPeriod}
+		setupLog.Info("MLflow CRD detected, caching MLflow resources", "syncPeriod", mlflowSyncPeriod)
+	} else {
+		mgrOpts.Client.Cache.DisableFor = append(mgrOpts.Client.Cache.DisableFor, &mlflowv1.MLflow{})
+		setupLog.Info("MLflow CRD not found, MLflow cache disabled")
+	}
+
+	mgr, err := ctrl.NewManager(restCfg, mgrOpts)
 	if err != nil {
 		setupLog.Error(err, "unable to start manager")
 		os.Exit(1)