Add storage to deployment resource calculation for podGroup (#416)

spotify · Jun 10, 2022 · afef85e · afef85e
1 parent aaf29e1
commit afef85e
Show file tree

Hide file tree

Showing 2 changed files with 294 additions and 116 deletions.
diff --git a/internal/batchscheduler/volcano/volcano.go b/internal/batchscheduler/volcano/volcano.go
@@ -241,6 +241,12 @@ func getDeploymentResources(spec *appsv1.DeploymentSpec) *corev1.ResourceRequire
 	for i := int32(0); i < *spec.Replicas; i++ {
 		tmResource := getPodResource(&spec.Template.Spec)
 		addResourceRequirements(reqs, tmResource)
+
+		for _, volume := range spec.Template.Spec.Volumes {
+			if volume.Ephemeral != nil && volume.Ephemeral.VolumeClaimTemplate != nil {
+				addResourceRequirements(reqs, &volume.Ephemeral.VolumeClaimTemplate.Spec.Resources)
+			}
+		}
 	}
 	return reqs
 }