fix: incorrect node resources

Signed-off-by: wangyizhi1 <[email protected]>
wangyizhi1 · Nov 21, 2023 · 39c2b0e · 39c2b0e
1 parent 0f957aa
commit 39c2b0e
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 7 deletions.
diff --git a/pkg/clustertree/cluster-manager/controllers/node_resources_controller.go b/pkg/clustertree/cluster-manager/controllers/node_resources_controller.go
@@ -126,10 +126,7 @@ func (c *NodeResourcesController) Reconcile(ctx context.Context, request reconci
 			}, err
 		}
 
-		clusterResources := utils.CalculateClusterResources(nodesInLeaf, pods)
 		clone := nodeInRoot.DeepCopy()
-		clone.Status.Allocatable = clusterResources
-		clone.Status.Capacity = clusterResources
 		clone.Status.Conditions = utils.NodeConditions()
 
 		// Node2Node mode should sync leaf node's labels and annotations to root nodeInRoot
@@ -156,6 +153,10 @@ func (c *NodeResourcesController) Reconcile(ctx context.Context, request reconci
 			}
 		}
 
+		clusterResources := utils.CalculateClusterResources(nodesInLeaf, pods)
+		clone.Status.Allocatable = clusterResources
+		clone.Status.Capacity = clusterResources
+
 		patch, err := utils.CreateMergePatch(nodeInRoot, clone)
 		if err != nil {
 			klog.Errorf("Could not CreateMergePatch,Error: %v", err)

diff --git a/pkg/clustertree/cluster-manager/controllers/pv/oneway_pv_controller.go b/pkg/clustertree/cluster-manager/controllers/pv/oneway_pv_controller.go
@@ -87,8 +87,8 @@ func (c *OnewayPVController) Reconcile(ctx context.Context, request reconcile.Re
 	vp, err := c.RootDynamic.Resource(VolumePathGVR).Get(ctx, request.Name, metav1.GetOptions{})
 	if err != nil {
 		if errors.IsNotFound(err) {
-			klog.Warningf("vp %s not found", request.Name)
-			return reconcile.Result{}, nil
+			klog.V(4).Infof("vp %s not found", request.Name)
+			return reconcile.Result{RequeueAfter: requeueTime}, nil
 		}
 		klog.Errorf("get volumePath %s error: %v", request.Name, err)
 		return reconcile.Result{RequeueAfter: requeueTime}, nil

diff --git a/pkg/clustertree/cluster-manager/controllers/pvc/oneway_pvc_controller.go b/pkg/clustertree/cluster-manager/controllers/pvc/oneway_pvc_controller.go
@@ -96,8 +96,8 @@ func (c *OnewayPVCController) Reconcile(ctx context.Context, request reconcile.R
 	vp, err := c.RootDynamic.Resource(VolumePathGVR).Get(ctx, request.Name, metav1.GetOptions{})
 	if err != nil {
 		if errors.IsNotFound(err) {
-			klog.Warningf("vp %s not found", request.Name)
-			return reconcile.Result{}, nil
+			klog.V(4).Infof("vp %s not found", request.Name)
+			return reconcile.Result{RequeueAfter: requeueTime}, nil
 		}
 		klog.Errorf("get volumePath %s error: %v", request.Name, err)
 		return reconcile.Result{RequeueAfter: requeueTime}, nil

diff --git a/pkg/utils/resources.go b/pkg/utils/resources.go
@@ -2,13 +2,20 @@ package utils
 
 import (
 	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/api/resource"
 	v1resource "k8s.io/kubernetes/pkg/api/v1/resource"
 )
 
+const (
+	podResourceName corev1.ResourceName = "pods"
+)
+
 func CalculateClusterResources(nodes *corev1.NodeList, pods *corev1.PodList) corev1.ResourceList {
 	base := GetNodesTotalResources(nodes)
 	reqs, _ := GetPodsTotalRequestsAndLimits(pods)
+	podNums := GetUsedPodNums(pods)
 	SubResourceList(base, reqs)
+	SubResourceList(base, podNums)
 	return base
 }
 
@@ -70,3 +77,18 @@ func GetPodsTotalRequestsAndLimits(podList *corev1.PodList) (reqs corev1.Resourc
 	}
 	return
 }
+
+func GetUsedPodNums(podList *corev1.PodList) (res corev1.ResourceList) {
+	podQuantity := resource.Quantity{}
+	res = corev1.ResourceList{}
+	for _, p := range podList.Items {
+		pod := p
+		if IsVirtualPod(&pod) {
+			continue
+		}
+		q := resource.MustParse("1")
+		podQuantity.Add(q)
+	}
+	res[podResourceName] = podQuantity
+	return
+}