Fix[LocalStorage]: fix scheduler no enough nodes bug

Signed-off-by: jie.li <jie.li@daocloud.io>
angel0507 · Aug 11, 2022 · b14e3f1 · b14e3f1
1 parent 90b5abd
commit b14e3f1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/pkg/local-storage/member/controller/scheduler/resources.go b/pkg/local-storage/member/controller/scheduler/resources.go
@@ -71,6 +71,21 @@ func (r *resources) init(apiClient client.Client, informerCache runtimecache.Cac
 	})
 }
 
+func (r *resources) initilizeTotalStorage() {
+	nodeList := &apisv1alpha1.LocalStorageNodeList{}
+	if err := r.apiClient.List(context.TODO(), nodeList); err != nil {
+		r.logger.WithError(err).Fatal("Failed to list LocalStorageNodes")
+	}
+	// initialize total capacity
+	for i := range nodeList.Items {
+		if nodeList.Items[i].Status.State == apisv1alpha1.NodeStateReady {
+			r.addTotalStorage(&nodeList.Items[i])
+		} else {
+			r.delTotalStorage(&nodeList.Items[i])
+		}
+	}
+}
+
 func (r *resources) initilizeResources() {
 	r.logger.Debug("Initializing resources ...")
 	volList := &apisv1alpha1.LocalVolumeList{}

diff --git a/pkg/local-storage/member/controller/scheduler/scheduler.go b/pkg/local-storage/member/controller/scheduler/scheduler.go
@@ -43,13 +43,10 @@ func (s *scheduler) Init() {
 }
 
 func (s *scheduler) initResources() {
-
 	s.once.Do(func() {
 		s.resourceCollections.apiClient = s.apiClient
 	})
-
-	s.resourceCollections.initilizeResources()
-
+	s.resourceCollections.initilizeTotalStorage()
 }
 
 // GetNodeCandidates gets available nodes for the volume, used by K8s scheduler