Remove the concurrencyLimit from a queue if the limit is null

Author: matt-aitken

apps/webapp/app/v3/marqs/index.server.ts

@@ -80,6 +80,10 @@ export class MarQS {
     return this.redis.set(this.keys.queueConcurrencyLimitKey(env, queue), concurrency);
   }
 
+  public async removeQueueConcurrencyLimits(env: AuthenticatedEnvironment, queue: string) {
+    return this.redis.del(this.keys.queueConcurrencyLimitKey(env, queue));
+  }
+
   public async updateEnvConcurrencyLimits(env: AuthenticatedEnvironment) {
     await this.#callUpdateGlobalConcurrencyLimits({
       envConcurrencyLimitKey: this.keys.envConcurrencyLimitKey(env),

apps/webapp/app/v3/services/createBackgroundWorker.server.ts

@@ -178,6 +178,8 @@ export async function createBackgroundTasks(
           taskQueue.name,
           taskQueue.concurrencyLimit
         );
+      } else {
+        await marqs?.removeQueueConcurrencyLimits(environment, taskQueue.name);
       }
     } catch (error) {
       if (error instanceof Prisma.PrismaClientKnownRequestError) {

apps/webapp/app/v3/services/triggerTask.server.ts

@@ -219,6 +219,8 @@ export class TriggerTaskService extends BaseService {
                   taskQueue.name,
                   taskQueue.concurrencyLimit
                 );
+              } else {
+                await marqs?.removeQueueConcurrencyLimits(environment, taskQueue.name);
               }
             }
 

references/v3-catalog/src/queues.ts

@@ -1,5 +1,6 @@
 import dotenv from "dotenv";
 import { simpleChildTask } from "./trigger/subtasks";
+import { wait } from "@trigger.dev/sdk/v3";
 
 dotenv.config();
 
@@ -54,6 +55,18 @@ export async function run() {
       },
     },
   ]);
+
+  await wait.for({ seconds: 5 });
+
+  //this should set the concurrencyLimit back to none
+  await simpleChildTask.trigger(
+    { message: "Simple alt queue 2" },
+    {
+      queue: {
+        name: "queue-concurrency-3",
+      },
+    }
+  );
 }
 
 run().catch(console.error);