feat: queue support concurrency config (#1231)

* feat: queue support concurrency config * refactor: remove redundant code
midwayjs · Aug 16, 2021 · 67bb7f5 · 67bb7f5
1 parent 5de5284
commit 67bb7f5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/packages/task/src/config/config.default.ts b/packages/task/src/config/config.default.ts
@@ -6,4 +6,5 @@ export const taskConfig = {
       tz: 'Asia/Shanghai',
     },
   },
+  concurrency: 1,
 };
diff --git a/packages/task/src/configuration.ts b/packages/task/src/configuration.ts
@@ -177,11 +177,12 @@ export class AutoConfiguration {
     const modules = listModule(MODULE_TASK_QUEUE_KEY);
     const queueMap = {};
     const config = JSON.parse(JSON.stringify(this.taskConfig));
+    const concurrency = config.concurrency || 1;
     delete config.defaultJobOptions.repeat;
     for (const module of modules) {
       const rule = getClassMetadata(MODULE_TASK_QUEUE_OPTIONS, module);
       const queue = new Bull(`${rule.name}:execute`, config);
-      queue.process(async job => {
+      queue.process(concurrency, async job => {
         const ctx = this.getContext({
           type: 'Queue',
           id: job.id,

diff --git a/packages/task/test/fixtures/base-app/src/config/config.default.ts b/packages/task/test/fixtures/base-app/src/config/config.default.ts
@@ -6,4 +6,5 @@ export const taskConfig = {
       tz: 'Asia/Shanghai',
     },
   },
+  concurrency: 1,
 };