driver: add ephemeral-storage options to kuberentes-driver

Signed-off-by: Moleus <fafufuburr@gmail.com>
This commit is contained in:
Moleus 2024-04-01 12:33:58 +03:00
parent c690d460e8
commit ef2e30deba
No known key found for this signature in database
2 changed files with 32 additions and 10 deletions

View File

@ -133,10 +133,14 @@ func (f *factory) processDriverOpts(deploymentName string, namespace string, cfg
deploymentOpt.RequestsCPU = v deploymentOpt.RequestsCPU = v
case "requests.memory": case "requests.memory":
deploymentOpt.RequestsMemory = v deploymentOpt.RequestsMemory = v
case "requests.ephemeral-storage":
deploymentOpt.RequestsEphemeralStorage = v
case "limits.cpu": case "limits.cpu":
deploymentOpt.LimitsCPU = v deploymentOpt.LimitsCPU = v
case "limits.memory": case "limits.memory":
deploymentOpt.LimitsMemory = v deploymentOpt.LimitsMemory = v
case "limits.ephemeral-storage":
deploymentOpt.LimitsEphemeralStorage = v
case "rootless": case "rootless":
deploymentOpt.Rootless, err = strconv.ParseBool(v) deploymentOpt.Rootless, err = strconv.ParseBool(v)
if err != nil { if err != nil {

View File

@ -39,8 +39,10 @@ type DeploymentOpt struct {
Tolerations []corev1.Toleration Tolerations []corev1.Toleration
RequestsCPU string RequestsCPU string
RequestsMemory string RequestsMemory string
RequestsEphemeralStorage string
LimitsCPU string LimitsCPU string
LimitsMemory string LimitsMemory string
LimitsEphemeralStorage string
Platforms []v1.Platform Platforms []v1.Platform
} }
@ -205,6 +207,14 @@ func NewDeployment(opt *DeploymentOpt) (d *appsv1.Deployment, c []*corev1.Config
d.Spec.Template.Spec.Containers[0].Resources.Requests[corev1.ResourceMemory] = reqMemory d.Spec.Template.Spec.Containers[0].Resources.Requests[corev1.ResourceMemory] = reqMemory
} }
if opt.RequestsEphemeralStorage != "" {
reqEphemeralStorage, err := resource.ParseQuantity(opt.RequestsEphemeralStorage)
if err != nil {
return nil, nil, err
}
d.Spec.Template.Spec.Containers[0].Resources.Requests[corev1.ResourceEphemeralStorage] = reqEphemeralStorage
}
if opt.LimitsCPU != "" { if opt.LimitsCPU != "" {
limCPU, err := resource.ParseQuantity(opt.LimitsCPU) limCPU, err := resource.ParseQuantity(opt.LimitsCPU)
if err != nil { if err != nil {
@ -221,6 +231,14 @@ func NewDeployment(opt *DeploymentOpt) (d *appsv1.Deployment, c []*corev1.Config
d.Spec.Template.Spec.Containers[0].Resources.Limits[corev1.ResourceMemory] = limMemory d.Spec.Template.Spec.Containers[0].Resources.Limits[corev1.ResourceMemory] = limMemory
} }
if opt.LimitsEphemeralStorage != "" {
limEphemeralStorage, err := resource.ParseQuantity(opt.LimitsEphemeralStorage)
if err != nil {
return nil, nil, err
}
d.Spec.Template.Spec.Containers[0].Resources.Limits[corev1.ResourceEphemeralStorage] = limEphemeralStorage
}
return return
} }