remove some client global vars

Signed-off-by: rambohe-ch <linbo.hlb@alibaba-inc.com>
openyurtio · Aug 15, 2023 · 243d7dc · 243d7dc
1 parent 271f774
commit 243d7dc
Show file tree

Hide file tree

Showing 13 changed files with 36 additions and 268 deletions.
diff --git a/cmd/yurt-manager/app/manager.go b/cmd/yurt-manager/app/manager.go
@@ -37,7 +37,6 @@ import (
 	"github.com/openyurtio/openyurt/cmd/yurt-manager/app/config"
 	"github.com/openyurtio/openyurt/cmd/yurt-manager/app/options"
 	"github.com/openyurtio/openyurt/pkg/apis"
-	extclient "github.com/openyurtio/openyurt/pkg/client"
 	"github.com/openyurtio/openyurt/pkg/controller"
 	"github.com/openyurtio/openyurt/pkg/profile"
 	"github.com/openyurtio/openyurt/pkg/projectinfo"
@@ -151,16 +150,8 @@ func Run(c *config.CompletedConfig, stopCh <-chan struct{}) error {
 
 	ctx := ctrl.SetupSignalHandler()
 	cfg := ctrl.GetConfigOrDie()
-
 	setRestConfig(cfg, c)
 
-	setupLog.Info("new clientset registry")
-	err := extclient.NewRegistry(cfg)
-	if err != nil {
-		setupLog.Error(err, "unable to init yurt-manager clientset and informer")
-		os.Exit(1)
-	}
-
 	mgr, err := ctrl.NewManager(cfg, ctrl.Options{
 		Scheme:                     scheme,
 		MetricsBindAddress:         c.ComponentConfig.Generic.MetricsAddr,

diff --git a/pkg/client/generic_client.go b/pkg/client/generic_client.go
diff --git a/pkg/client/registry.go b/pkg/client/registry.go
diff --git a/pkg/controller/daemonpodupdater/daemon_pod_updater_controller.go b/pkg/controller/daemonpodupdater/daemon_pod_updater_controller.go
@@ -51,7 +51,6 @@ import (
 	appconfig "github.com/openyurtio/openyurt/cmd/yurt-manager/app/config"
 	k8sutil "github.com/openyurtio/openyurt/pkg/controller/daemonpodupdater/kubernetes"
 	utilclient "github.com/openyurtio/openyurt/pkg/util/client"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 func init() {
@@ -103,9 +102,6 @@ func Format(format string, args ...interface{}) string {
 // Add creates a new Daemonpodupdater Controller and adds it to the Manager with default RBAC. The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *appconfig.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		return nil
-	}
 	klog.Infof("daemonupdater-controller add controller %s", controllerKind.String())
 	return add(mgr, newReconciler(c, mgr))
 }

diff --git a/pkg/controller/platformadmin/platformadmin_controller.go b/pkg/controller/platformadmin/platformadmin_controller.go
@@ -49,8 +49,6 @@ import (
 	iotv1alpha2 "github.com/openyurtio/openyurt/pkg/apis/iot/v1alpha2"
 	"github.com/openyurtio/openyurt/pkg/controller/platformadmin/config"
 	util "github.com/openyurtio/openyurt/pkg/controller/platformadmin/utils"
-	utilclient "github.com/openyurtio/openyurt/pkg/util/client"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 func init() {
@@ -64,7 +62,7 @@ func Format(format string, args ...interface{}) string {
 
 var (
 	concurrentReconciles = 3
-	controllerKind       = iotv1alpha2.SchemeGroupVersion.WithKind("PlatformAdmin")
+	controllerResource   = iotv1alpha2.SchemeGroupVersion.WithResource("platformadmins")
 )
 
 const (
@@ -121,18 +119,19 @@ var _ reconcile.Reconciler = &ReconcilePlatformAdmin{}
 // Add creates a new PlatformAdmin Controller and adds it to the Manager with default RBAC. The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *appconfig.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		return nil
+	if _, err := mgr.GetRESTMapper().KindFor(controllerResource); err != nil {
+		klog.Infof("resource %s doesn't exist", controllerResource.String())
+		return err
 	}
 
-	klog.Infof("platformadmin-controller add controller %s", controllerKind.String())
+	klog.Infof("platformadmin-controller add controller %s", controllerResource.String())
 	return add(mgr, newReconciler(c, mgr))
 }
 
 // newReconciler returns a new reconcile.Reconciler
 func newReconciler(c *appconfig.CompletedConfig, mgr manager.Manager) reconcile.Reconciler {
 	return &ReconcilePlatformAdmin{
-		Client:         utilclient.NewClientFromManager(mgr, ControllerName),
+		Client:         mgr.GetClient(),
 		scheme:         mgr.GetScheme(),
 		recorder:       mgr.GetEventRecorderFor(ControllerName),
 		yamlSerializer: kjson.NewSerializerWithOptions(kjson.DefaultMetaFactory, scheme.Scheme, scheme.Scheme, kjson.SerializerOptions{Yaml: true, Pretty: true}),

diff --git a/pkg/controller/raven/gateway/gateway_controller.go b/pkg/controller/raven/gateway/gateway_controller.go
@@ -43,12 +43,10 @@ import (
 	"github.com/openyurtio/openyurt/pkg/controller/raven/config"
 	"github.com/openyurtio/openyurt/pkg/controller/raven/utils"
 	nodeutil "github.com/openyurtio/openyurt/pkg/controller/util/node"
-	utilclient "github.com/openyurtio/openyurt/pkg/util/client"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 var (
-	controllerKind = ravenv1alpha1.SchemeGroupVersion.WithKind("Gateway")
+	controllerResource = ravenv1alpha1.SchemeGroupVersion.WithResource("gateways")
 )
 
 func Format(format string, args ...interface{}) string {
@@ -59,14 +57,15 @@ func Format(format string, args ...interface{}) string {
 // Add creates a new Gateway Controller and adds it to the Manager with default RBAC. The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *appconfig.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		return nil
+	if _, err := mgr.GetRESTMapper().KindFor(controllerResource); err != nil {
+		klog.Infof("resource %s doesn't exist", controllerResource.String())
+		return err
 	}
+
 	// init global variables
-	cfg := c.ComponentConfig.Generic
-	ravenv1alpha1.ServiceNamespacedName.Namespace = cfg.WorkingNamespace
+	ravenv1alpha1.ServiceNamespacedName.Namespace = c.ComponentConfig.Generic.WorkingNamespace
 
-	klog.Infof("ravenl3-gateway-controller add controller %s", controllerKind.String())
+	klog.Infof("ravenl3-gateway-controller add controller %s", controllerResource.String())
 	return add(mgr, newReconciler(c, mgr))
 }
 
@@ -83,7 +82,7 @@ type ReconcileGateway struct {
 // newReconciler returns a new reconcile.Reconciler
 func newReconciler(c *appconfig.CompletedConfig, mgr manager.Manager) reconcile.Reconciler {
 	return &ReconcileGateway{
-		Client:       utilclient.NewClientFromManager(mgr, common.ControllerName),
+		Client:       mgr.GetClient(),
 		scheme:       mgr.GetScheme(),
 		recorder:     mgr.GetEventRecorderFor(common.ControllerName),
 		Configration: c.ComponentConfig.GatewayController,

diff --git a/pkg/controller/raven/service/service_controller.go b/pkg/controller/raven/service/service_controller.go
@@ -41,8 +41,6 @@ import (
 	ravenv1alpha1 "github.com/openyurtio/openyurt/pkg/apis/raven/v1alpha1"
 	common "github.com/openyurtio/openyurt/pkg/controller/raven"
 	"github.com/openyurtio/openyurt/pkg/controller/raven/utils"
-	utilclient "github.com/openyurtio/openyurt/pkg/util/client"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 var (
@@ -57,10 +55,6 @@ func Format(format string, args ...interface{}) string {
 // Add creates a new Service Controller and adds it to the Manager with default RBAC. The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *appconfig.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		return nil
-	}
-
 	klog.Infof("ravenl3-service-controller add controller %s", controllerKind.String())
 	return add(mgr, newReconciler(c, mgr))
 }
@@ -77,7 +71,7 @@ type ReconcileService struct {
 // newReconciler returns a new reconcile.Reconciler
 func newReconciler(c *appconfig.CompletedConfig, mgr manager.Manager) reconcile.Reconciler {
 	return &ReconcileService{
-		Client:   utilclient.NewClientFromManager(mgr, common.ControllerName),
+		Client:   mgr.GetClient(),
 		scheme:   mgr.GetScheme(),
 		recorder: mgr.GetEventRecorderFor(common.ControllerName),
 	}

diff --git a/pkg/controller/servicetopology/endpoints/endpoints_controller.go b/pkg/controller/servicetopology/endpoints/endpoints_controller.go
@@ -33,7 +33,6 @@ import (
 	appconfig "github.com/openyurtio/openyurt/cmd/yurt-manager/app/config"
 	common "github.com/openyurtio/openyurt/pkg/controller/servicetopology"
 	"github.com/openyurtio/openyurt/pkg/controller/servicetopology/adapter"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 func init() {
@@ -53,10 +52,6 @@ func Format(format string, args ...interface{}) string {
 // Add creates a new Servicetopology endpoints Controller and adds it to the Manager with default RBAC. The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *appconfig.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		return nil
-	}
-
 	klog.Infof("servicetopology-endpoints-controller add controller %s", controllerKind.String())
 	return add(mgr, newReconciler(c, mgr))
 }

diff --git a/pkg/controller/yurtappdaemon/yurtappdaemon_controller.go b/pkg/controller/yurtappdaemon/yurtappdaemon_controller.go
@@ -41,12 +41,11 @@ import (
 	unitv1alpha1 "github.com/openyurtio/openyurt/pkg/apis/apps/v1alpha1"
 	"github.com/openyurtio/openyurt/pkg/controller/util"
 	"github.com/openyurtio/openyurt/pkg/controller/yurtappdaemon/workloadcontroller"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 var (
 	concurrentReconciles = 3
-	controllerKind       = unitv1alpha1.SchemeGroupVersion.WithKind("YurtAppDaemon")
+	controllerResource   = unitv1alpha1.SchemeGroupVersion.WithResource("yurtappdaemons")
 )
 
 const (
@@ -74,12 +73,12 @@ func Format(format string, args ...interface{}) string {
 // The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *config.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		klog.Errorf(Format("DiscoverGVK error"))
-		return nil
+	if _, err := mgr.GetRESTMapper().KindFor(controllerResource); err != nil {
+		klog.Infof("resource %s doesn't exist", controllerResource.String())
+		return err
 	}
 
-	klog.Infof("yurtappdaemon-controller add controller %s", controllerKind.String())
+	klog.Infof("yurtappdaemon-controller add controller %s", controllerResource.String())
 	return add(mgr, newReconciler(mgr))
 }
 

diff --git a/pkg/controller/yurtappset/yurtappset_controller.go b/pkg/controller/yurtappset/yurtappset_controller.go
@@ -43,7 +43,6 @@ import (
 	"github.com/openyurtio/openyurt/cmd/yurt-manager/app/config"
 	unitv1alpha1 "github.com/openyurtio/openyurt/pkg/apis/apps/v1alpha1"
 	"github.com/openyurtio/openyurt/pkg/controller/yurtappset/adapter"
-	utildiscovery "github.com/openyurtio/openyurt/pkg/util/discovery"
 )
 
 func init() {
@@ -52,7 +51,7 @@ func init() {
 
 var (
 	concurrentReconciles = 3
-	controllerKind       = unitv1alpha1.SchemeGroupVersion.WithKind("YurtAppSet")
+	controllerResource   = unitv1alpha1.SchemeGroupVersion.WithResource("yurtappsets")
 )
 
 const (
@@ -75,19 +74,17 @@ func Format(format string, args ...interface{}) string {
 // Add creates a new YurtAppSet Controller and adds it to the Manager with default RBAC. The Manager will set fields on the Controller
 // and Start it when the Manager is Started.
 func Add(c *config.CompletedConfig, mgr manager.Manager) error {
-	if !utildiscovery.DiscoverGVK(controllerKind) {
-		klog.Errorf(Format("DiscoverGVK error"))
-		return nil
+	if _, err := mgr.GetRESTMapper().KindFor(controllerResource); err != nil {
+		klog.Infof("resource %s doesn't exist", controllerResource.String())
+		return err
 	}
 
-	klog.Infof("yurtappset-controller add controller %s", controllerKind.String())
+	klog.Infof("yurtappset-controller add controller %s", controllerResource.String())
 	return add(mgr, newReconciler(c, mgr))
 }
 
 // newReconciler returns a new reconcile.Reconciler
 func newReconciler(c *config.CompletedConfig, mgr manager.Manager) reconcile.Reconciler {
-	klog.Infof("yurtappset-controller newReconciler %s", controllerKind.String())
-
 	return &ReconcileYurtAppSet{
 		Client: mgr.GetClient(),
 		scheme: mgr.GetScheme(),