chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,58 @@
+import { type AuthenticatedEnvironment } from "~/services/apiAuth.server";
+import { marqs } from "~/v3/marqs/index.server";
+import { engine } from "~/v3/runEngine.server";
+import { getQueueSizeLimit } from "~/v3/utils/queueLimits.server";
+import { BasePresenter } from "./basePresenter.server";
+
+export type Environment = {
+  running: number;
+  queued: number;
+  concurrencyLimit: number;
+  burstFactor: number;
+  runsEnabled: boolean;
+  queueSizeLimit: number | null;
+};
+
+export class EnvironmentQueuePresenter extends BasePresenter {
+  async call(environment: AuthenticatedEnvironment): Promise<Environment> {
+    const [engineV1Executing, engineV2Executing, engineV1Queued, engineV2Queued] =
+      await Promise.all([
+        marqs.currentConcurrencyOfEnvironment(environment),
+        engine.concurrencyOfEnvQueue(environment),
+        marqs.lengthOfEnvQueue(environment),
+        engine.lengthOfEnvQueue(environment),
+      ]);
+
+    const running = (engineV1Executing ?? 0) + (engineV2Executing ?? 0);
+    const queued = (engineV1Queued ?? 0) + (engineV2Queued ?? 0);
+
+    const organization = await this._replica.organization.findFirst({
+      where: {
+        id: environment.organizationId,
+      },
+      select: {
+        runsEnabled: true,
+        maximumDevQueueSize: true,
+        maximumDeployedQueueSize: true,
+      },
+    });
+
+    if (!organization) {
+      throw new Error("Organization not found");
+    }
+
+    const queueSizeLimit = getQueueSizeLimit(environment.type, organization);
+
+    return {
+      running,
+      queued,
+      concurrencyLimit: environment.maximumConcurrencyLimit,
+      burstFactor:
+        typeof environment.concurrencyLimitBurstFactor === "number"
+          ? environment.concurrencyLimitBurstFactor
+          : environment.concurrencyLimitBurstFactor.toNumber(),
+      runsEnabled: environment.type === "DEVELOPMENT" || organization.runsEnabled,
+      queueSizeLimit,
+    };
+  }
+}