Merge pull request #1849 from zimnx/mz/scheme-builder

Add Scheme Builder
scylladb · Mar 20, 2024 · 1520cf7 · 1520cf7
2 parents 8b32358 + 28af2b0
commit 1520cf7
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/pkg/scheme/scheme.go b/pkg/scheme/scheme.go
@@ -25,15 +25,17 @@ var (
 			Strict: true,
 		},
 	)
+	localSchemeBuilder = runtime.SchemeBuilder{
+		kscheme.AddToScheme,
+		scyllav1.Install,
+		scyllav1alpha1.Install,
+		cqlclientv1alpha1.Install,
+		monitoringv1.Install,
+	}
+
+	AddToScheme = localSchemeBuilder.AddToScheme
 )
 
 func init() {
-	utilruntime.Must(kscheme.AddToScheme(Scheme))
-
-	utilruntime.Must(scyllav1.Install(Scheme))
-	utilruntime.Must(scyllav1alpha1.Install(Scheme))
-
-	utilruntime.Must(cqlclientv1alpha1.Install(Scheme))
-
-	utilruntime.Must(monitoringv1.Install(Scheme))
+	utilruntime.Must(AddToScheme(Scheme))
 }