modify leaf_resource_manager to singleton

Signed-off-by: wuyingjun <[email protected]>
wuyingjun-lucky · Nov 2, 2023 · 1eef7d2 · 1eef7d2
1 parent 344d998
commit 1eef7d2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/cmd/clustertree/cluster-manager/app/manager.go b/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 {

diff --git a/pkg/clustertree/cluster-manager/utils/leaf_resource_manager.go b/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
 }