From 435a03fa71f2091d4a7ed3ba67f0338bed2fd5f7 Mon Sep 17 00:00:00 2001 From: youys <1272586223@qq.com> Date: Thu, 7 Aug 2025 16:15:13 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BB=BB=E5=8A=A1=E5=88=97=E8=A1=A8?= =?UTF-8?q?=E6=94=BE=E5=87=BA=E6=99=AE=E9=80=9A=E5=AE=9E=E8=AE=AD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/internal/data/pod.go | 8 ++++++-- server/internal/service/container.go | 5 ++--- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/server/internal/data/pod.go b/server/internal/data/pod.go index 96c30c9..dc7c9c3 100644 --- a/server/internal/data/pod.go +++ b/server/internal/data/pod.go @@ -59,7 +59,8 @@ func (r *podRepo) onAddPod(obj interface{}) { return } nodeID, ok := pod.Annotations[util.AssignedNodeAnnotations] - if !ok { + tpiID := pod.Labels["tpi-id"] + if !ok && tpiID == "" { return } if biz.IsPodInTerminatedState(pod) { @@ -121,7 +122,10 @@ func (r *podRepo) fetchContainerInfo(pod *corev1.Pod) []*biz.Container { copier.Copy(&bizContainerDevices, pds) } if len(bizContainerDevices) < 1 { - return containers + for range pod.Spec.Containers { + bizContainerDevices = append(bizContainerDevices, biz.ContainerDevices{}) + } + //return containers } ctrIdMaps := map[string]string{} diff --git a/server/internal/service/container.go b/server/internal/service/container.go index ccba6f4..aaa4705 100644 --- a/server/internal/service/container.go +++ b/server/internal/service/container.go @@ -105,6 +105,8 @@ func (s *ContainerService) GetAllContainers(ctx context.Context, req *pb.GetAllC containerReply.NodeUid = container.NodeUID containerReply.Namespace = container.Namespace containerReply.Priority = container.Priority + containerReply.RequestedCpuCores = container.RequestedCpuCores + containerReply.RequestedMemory = container.RequestedMemory for _, containerDevice := range container.ContainerDevices { deviceID := containerDevice.UUID if device, err := s.node.FindDeviceByAliasId(containerDevice.UUID); err == nil { @@ -125,9 +127,6 @@ func (s *ContainerService) GetAllContainers(ctx context.Context, req *pb.GetAllC containerReply.Type = containerDevice.Type containerReply.AllocatedDevices++ } - if containerReply.DeviceIds == nil { - continue - } resourcePoolNames, err := database.QueryResourceNamesByNodeName(container.NodeName) if err != nil {