fix(resources): improve ressource accounting. Fixes argoproj#12468 (a…

…rgoproj#12492) Signed-off-by: Aloÿs Augustin <aloys.augustin@aqemia.com>
isubasinghe · Mar 12, 2024 · 5761695 · 5761695
1 parent 574c6ec
commit 5761695
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/util/resource/duration_test.go b/util/resource/duration_test.go
@@ -44,7 +44,7 @@ func TestDurationForPod(t *testing.T) {
 			corev1.ResourceCPU:    wfv1.NewResourceDuration(2 * time.Minute),
 			corev1.ResourceMemory: wfv1.NewResourceDuration(1 * time.Minute),
 		}},
-		{"ContainerWithCPURequest", &corev1.Pod{
+		{"ContainerWithGPULimit", &corev1.Pod{
 			Spec: corev1.PodSpec{Containers: []corev1.Container{{Name: "main", Resources: corev1.ResourceRequirements{
 				Requests: corev1.ResourceList{
 					corev1.ResourceCPU: resource.MustParse("2000m"),

diff --git a/util/resource/summary.go b/util/resource/summary.go
@@ -28,9 +28,14 @@ func (ss Summaries) Duration() wfv1.ResourcesDuration {
 	// Add container states.
 	d := wfv1.ResourcesDuration{}
 	for _, s := range ss {
+		// age is converted to seconds, otherwise the multiplication below is very likely to overflow
 		age := int64(s.age().Seconds())
 		for n, q := range s.ResourceList {
-			d = d.Add(wfv1.ResourcesDuration{n: wfv1.NewResourceDuration(time.Duration(q.Value() * age / wfv1.ResourceQuantityDenominator(n).Value() * int64(time.Second)))})
+			d = d.Add(wfv1.ResourcesDuration{
+				n: wfv1.NewResourceDuration(time.Duration(
+					q.MilliValue()*age/wfv1.ResourceQuantityDenominator(n).MilliValue(),
+				) * time.Second),
+			})
 		}
 	}
 	return d