Merge pull request #57 from elixir-cloud-aai/fix-for-rahti

fix Sets the requests equals to limits.
elixir-cloud-aai · Aug 7, 2024 · 12754b8 · 12754b8
2 parents eceb7b4 + 61aa043
commit 12754b8
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/src/main/java/uk/ac/ebi/tsc/tesk/k8s/convert/TesKubernetesConverter.java b/src/main/java/uk/ac/ebi/tsc/tesk/k8s/convert/TesKubernetesConverter.java
@@ -189,6 +189,7 @@ public V1Job fromTesExecutorToK8sJob(String generatedTaskId, String tesTaskName,
             executor.getEnv().forEach((key, value) -> container.addEnvItem(new V1EnvVar().name(key).value(value)));
         }
         container.setWorkingDir(executor.getWorkdir());
+        Optional.ofNullable(resources).map(TesResources::getCpuCores).ifPresent(cpuCores -> container.getResources().putLimitsItem(RESOURCE_CPU_KEY, new QuantityFormatter().parse(cpuCores.toString())));
         Optional.ofNullable(resources).map(TesResources::getCpuCores).ifPresent(cpuCores -> container.getResources().putRequestsItem(RESOURCE_CPU_KEY, new QuantityFormatter().parse(cpuCores.toString())));
 	// Limit number of decimals to 6
 	Optional.ofNullable(resources).map(TesResources::getRamGb).ifPresent(ramGb -> container.getResources().putRequestsItem(RESOURCE_MEM_KEY, new QuantityFormatter().parse(String.format("%.6f",ramGb)+RESOURCE_MEM_UNIT)));

diff --git a/src/test/java/uk/ac/ebi/tsc/tesk/k8s/convert/TesKubernetesConverterTest.java b/src/test/java/uk/ac/ebi/tsc/tesk/k8s/convert/TesKubernetesConverterTest.java
@@ -241,6 +241,7 @@ public void fromTesTaskToK8sConfigMap() throws IOException {
         taskMasterInputJson.extractingJsonPathStringValue("executors[1].spec.template.spec.containers[0].name").isEqualTo("task-35605447-ex-01");
 
         taskMasterInputJson.extractingJsonPathArrayValue("executors[*].spec.template.spec.restartPolicy").containsOnly("Never").hasSize(2);
+        taskMasterInputJson.extractingJsonPathArrayValue("executors[*].spec.template.spec.containers[0].resources.limits.cpu").containsOnly("4").hasSize(2);
         taskMasterInputJson.extractingJsonPathArrayValue("executors[*].spec.template.spec.containers[0].resources.requests.cpu").containsOnly("4").hasSize(2);
         taskMasterInputJson.extractingJsonPathArrayValue("executors[*].spec.template.spec.containers[0].resources.requests.memory").containsOnly("16106127360").hasSize(2);
 

diff --git a/src/test/resources/fromTesToK8s/taskmaster_param.json b/src/test/resources/fromTesToK8s/taskmaster_param.json
@@ -69,6 +69,9 @@
                   "requests": {
                     "cpu": "4",
                     "memory": "16106127360"
+                  },
+                  "limits": {
+                    "cpu": "4"
                   }
                 },
                 "volumeMounts": [
@@ -143,6 +146,9 @@
                   "requests": {
                     "cpu": "4",
                     "memory": "16106127360"
+                  },
+                  "limits": {
+                    "cpu": "4"
                   }
                 },
                 "workingDir": "/starthere",