Performance: fetch all docker container statuses in one request (#7092)

Co-authored-by: shamoon <4887959+shamoon@users.noreply.github.com>

src/pages/api/docker/status/[...service].js

@@ -1,116 +0,0 @@
-import Docker from "dockerode";
-
-import getDockerArguments from "utils/config/docker";
-import createLogger from "utils/logger";
-
-const logger = createLogger("dockerStatusService");
-
-export default async function handler(req, res) {
-  const { service } = req.query;
-  const [containerName, containerServer] = service;
-
-  if (!containerName && !containerServer) {
-    return res.status(400).send({
-      error: "docker query parameters are required",
-    });
-  }
-
-  try {
-    const dockerArgs = getDockerArguments(containerServer);
-    const docker = new Docker(dockerArgs.conn);
-    const containers = await docker.listContainers({
-      all: true,
-    });
-
-    // bad docker connections can result in a <Buffer ...> object?
-    // in any case, this ensures the result is the expected array
-    if (!Array.isArray(containers)) {
-      return res.status(500).send({
-        error: "query failed",
-      });
-    }
-
-    const containerNames = containers.flatMap((container) => container.Names.map((name) => name.replace(/^\//, "")));
-    const containerExists = containerNames.includes(containerName);
-
-    if (containerExists) {
-      const container = docker.getContainer(containerName);
-      const info = await container.inspect();
-
-      return res.status(200).json({
-        status: info.State.Status,
-        health: info.State.Health?.Status,
-      });
-    }
-
-    if (dockerArgs.swarm) {
-      const serviceInfo = await docker
-        .getService(containerName)
-        .inspect()
-        .catch(() => undefined);
-
-      if (!serviceInfo) {
-        return res.status(404).send({
-          status: "not found",
-        });
-      }
-
-      const tasks = await docker
-        .listTasks({
-          filters: {
-            service: [containerName],
-            "desired-state": ["running"],
-          },
-        })
-        .catch(() => []);
-
-      if (serviceInfo.Spec.Mode?.Replicated) {
-        // Replicated service, check n replicas
-        const replicas = parseInt(serviceInfo.Spec.Mode?.Replicated?.Replicas, 10);
-        if (tasks.length === replicas) {
-          return res.status(200).json({
-            status: `running ${tasks.length}/${replicas}`,
-          });
-        }
-        if (tasks.length > 0) {
-          return res.status(200).json({
-            status: `partial ${tasks.length}/${replicas}`,
-          });
-        }
-      } else {
-        // Global service, prefer 'local' containers
-        const localContainerIDs = containers.map((c) => c.Id);
-        const task =
-          tasks.find((t) => localContainerIDs.includes(t.Status?.ContainerStatus?.ContainerID)) ?? tasks.at(0);
-        const taskContainerId = task?.Status?.ContainerStatus?.ContainerID;
-
-        if (taskContainerId) {
-          try {
-            const container = docker.getContainer(taskContainerId);
-            const info = await container.inspect();
-
-            return res.status(200).json({
-              status: info.State.Status,
-              health: info.State.Health?.Status,
-            });
-          } catch (e) {
-            if (task) {
-              return res.status(200).json({
-                status: task.Status.State,
-              });
-            }
-          }
-        }
-      }
-    }
-
-    return res.status(404).send({
-      status: "not found",
-    });
-  } catch (e) {
-    if (e) logger.error(e);
-    return res.status(500).send({
-      error: { message: e?.message ?? "Unknown error" },
-    });
-  }
-}

src/pages/api/docker/statuses.js

@@ -0,0 +1,21 @@
+import { getDockerStatuses } from "utils/docker/status";
+import createLogger from "utils/logger";
+
+const logger = createLogger("dockerStatuses");
+
+export default async function handler(req, res) {
+  try {
+    const result = await getDockerStatuses(req.query.server);
+
+    if (result.error) {
+      return res.status(500).send({ error: result.error });
+    }
+
+    return res.status(200).json(result);
+  } catch (e) {
+    if (e) logger.error(e);
+    return res.status(500).send({
+      error: { message: e?.message ?? "Unknown error" },
+    });
+  }
+}