Implement support for specifying container when getting logs

If a pod have more than one container the current call will fail. This commit fix that issue, while perserving backward compability in the api by making the container id parameter optional. The get logs call will behave as before if container id is null or empty
statisticsnorway · Aug 21, 2024 · 5d39329 · 5d39329
1 parent 5e48651
commit 5d39329
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/onyxia-api/src/main/java/fr/insee/onyxia/api/controller/api/mylab/MyLabController.java b/onyxia-api/src/main/java/fr/insee/onyxia/api/controller/api/mylab/MyLabController.java
@@ -298,10 +298,11 @@ public String getLogs(
             @Parameter(hidden = true) Region region,
             @Parameter(hidden = true) Project project,
             @RequestParam("serviceId") String serviceId,
-            @RequestParam("taskId") String taskId) {
+            @RequestParam("taskId") String taskId,
+            @RequestParam("containerId") Optional<String> containerId) {
         if (Service.ServiceType.KUBERNETES.equals(region.getServices().getType())) {
             return helmAppsService.getLogs(
-                    region, project, userProvider.getUser(region), serviceId, taskId);
+                    region, project, userProvider.getUser(region), serviceId, taskId, containerId.orElse(null));
         }
         return null;
     }

diff --git a/onyxia-api/src/main/java/fr/insee/onyxia/api/services/AppsService.java b/onyxia-api/src/main/java/fr/insee/onyxia/api/services/AppsService.java
@@ -49,7 +49,7 @@ Service getUserService(Region region, Project project, User user, String service
     UninstallService destroyService(
             Region region, Project project, User user, String path, boolean bulk) throws Exception;
 
-    String getLogs(Region region, Project project, User user, String serviceId, String taskId);
+    String getLogs(Region region, Project project, User user, String serviceId, String taskId, String containerId);
 
     Watch getEvents(Region region, Project project, User user, Watcher<Event> watcher)
             throws HelmInstallService.MultipleServiceFound, ParseException;

diff --git a/onyxia-api/src/main/java/fr/insee/onyxia/api/services/impl/HelmAppsService.java b/onyxia-api/src/main/java/fr/insee/onyxia/api/services/impl/HelmAppsService.java
@@ -198,13 +198,14 @@ public CompletableFuture<ServicesListing> getUserServices(
 
     @Override
     public String getLogs(
-            Region region, Project project, User user, String serviceId, String taskId) {
+            Region region, Project project, User user, String serviceId, String taskId, String containerId) {
         KubernetesClient client = kubernetesClientProvider.getUserClient(region, user);
         return client.pods()
                 .inNamespace(
                         kubernetesService.determineNamespaceAndCreateIfNeeded(
                                 region, project, user))
                 .withName(taskId)
+                .inContainer(containerId)
                 .getLog();
     }