Merge pull request kosmos-io#195 from wuyingjun-lucky/main

modify leaf_resource_manager to singleton

Author: kosmos-robot

cmd/clustertree/cluster-manager/app/manager.go

@@ -58,7 +58,7 @@ func NewAgentCommand(ctx context.Context) *cobra.Command {
 }
 
 func run(ctx context.Context, opts *options.Options) error {
-	globalleafManager := leafUtils.NewLeafResourceManager()
+	globalleafManager := leafUtils.GetGlobalLeafResourceManager()
 
 	config, err := clientcmd.BuildConfigFromFlags(opts.KubernetesOptions.Master, opts.KubernetesOptions.KubeConfig)
 	if err != nil {

pkg/clustertree/cluster-manager/utils/leaf_resource_manager.go

@@ -9,6 +9,11 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/client"
 )
 
+var (
+	instance LeafResourceManager
+	once     sync.Once
+)
+
 type LeafResource struct {
 	Client               client.Client
 	DynamicClient        dynamic.Interface
@@ -74,8 +79,12 @@ func (l *leafResourceManager) ListNodeNames() []string {
 	return keys
 }
 
-func NewLeafResourceManager() LeafResourceManager {
-	return &leafResourceManager{
-		resourceMap: make(map[string]*LeafResource),
-	}
+func GetGlobalLeafResourceManager() LeafResourceManager {
+	once.Do(func() {
+		instance = &leafResourceManager{
+			resourceMap: make(map[string]*LeafResource),
+		}
+	})
+
+	return instance
 }