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,44 @@
+import { type AuthenticatedEnvironment } from "~/services/apiAuth.server";
+import { marqs } from "./marqs/index.server";
+import { engine } from "./runEngine.server";
+
+//This allows us to update MARQS and the RunQueue
+
+/** Updates MARQS and the RunQueue limits */
+export async function updateEnvConcurrencyLimits(
+  environment: AuthenticatedEnvironment,
+  maximumConcurrencyLimit?: number
+) {
+  let updatedEnvironment = environment;
+  if (maximumConcurrencyLimit !== undefined) {
+    updatedEnvironment.maximumConcurrencyLimit = maximumConcurrencyLimit;
+  }
+
+  await Promise.allSettled([
+    marqs?.updateEnvConcurrencyLimits(updatedEnvironment),
+    engine.runQueue.updateEnvConcurrencyLimits(updatedEnvironment),
+  ]);
+}
+
+/** Updates MARQS and the RunQueue limits for a queue */
+export async function updateQueueConcurrencyLimits(
+  environment: AuthenticatedEnvironment,
+  queueName: string,
+  concurrency: number
+) {
+  await Promise.allSettled([
+    marqs?.updateQueueConcurrencyLimits(environment, queueName, concurrency),
+    engine.runQueue.updateQueueConcurrencyLimits(environment, queueName, concurrency),
+  ]);
+}
+
+/** Removes MARQS and the RunQueue limits for a queue */
+export async function removeQueueConcurrencyLimits(
+  environment: AuthenticatedEnvironment,
+  queueName: string
+) {
+  await Promise.allSettled([
+    marqs?.removeQueueConcurrencyLimits(environment, queueName),
+    engine.runQueue.removeQueueConcurrencyLimits(environment, queueName),
+  ]);
+}