From cbb618c6694b31ff879cc791028829a1451eb61a Mon Sep 17 00:00:00 2001
From: Hao Luo <howlowck@gmail.com>
Date: Thu, 13 Oct 2022 16:44:59 -0500
Subject: [PATCH] added taskWorkers back in

Signed-off-by: Hao Luo <howlowck@gmail.com>
---
 plugins/scaffolder-backend/api-report.md           | 3 +++
 plugins/scaffolder-backend/src/ScaffolderPlugin.ts | 3 +++
 plugins/scaffolder-backend/src/service/router.ts   | 2 ++
 3 files changed, 8 insertions(+)

diff --git a/plugins/scaffolder-backend/api-report.md b/plugins/scaffolder-backend/api-report.md
index 2ea9884b99..96a3d33fa7 100644
--- a/plugins/scaffolder-backend/api-report.md
+++ b/plugins/scaffolder-backend/api-report.md
@@ -562,6 +562,8 @@ export interface RouterOptions {
   scheduler?: PluginTaskScheduler;
   // (undocumented)
   taskBroker?: TaskBroker;
+  // (undocumented)
+  taskWorkers?: number;
 }
 
 // @public (undocumented)
@@ -597,6 +599,7 @@ export const scaffolderPlugin: (
 // @alpha
 export type ScaffolderPluginOptions = {
   actions?: TemplateAction<any>[];
+  taskWorkers?: number;
   taskBroker?: TaskBroker;
   additionalTemplateFilters?: Record<string, TemplateFilter>;
   additionalTemplateGlobals?: Record<string, TemplateGlobal>;
diff --git a/plugins/scaffolder-backend/src/ScaffolderPlugin.ts b/plugins/scaffolder-backend/src/ScaffolderPlugin.ts
index 84bf355156..09d7bdcbb2 100644
--- a/plugins/scaffolder-backend/src/ScaffolderPlugin.ts
+++ b/plugins/scaffolder-backend/src/ScaffolderPlugin.ts
@@ -36,6 +36,7 @@ import { createRouter } from './service/router';
  */
 export type ScaffolderPluginOptions = {
   actions?: TemplateAction<any>[];
+  taskWorkers?: number;
   taskBroker?: TaskBroker;
   additionalTemplateFilters?: Record<string, TemplateFilter>;
   additionalTemplateGlobals?: Record<string, TemplateGlobal>;
@@ -104,6 +105,7 @@ export const scaffolderPlugin = createBackendPlugin({
         const {
           additionalTemplateFilters,
           taskBroker,
+          taskWorkers,
           additionalTemplateGlobals,
         } = options;
         const log = loggerToWinstonLogger(logger);
@@ -133,6 +135,7 @@ export const scaffolderPlugin = createBackendPlugin({
           reader,
           actions,
           taskBroker,
+          taskWorkers,
           additionalTemplateFilters,
           additionalTemplateGlobals,
         });
diff --git a/plugins/scaffolder-backend/src/service/router.ts b/plugins/scaffolder-backend/src/service/router.ts
index ddef93e396..6fe914202e 100644
--- a/plugins/scaffolder-backend/src/service/router.ts
+++ b/plugins/scaffolder-backend/src/service/router.ts
@@ -67,6 +67,7 @@ export interface RouterOptions {
   scheduler?: PluginTaskScheduler;
 
   actions?: TemplateAction<any>[];
+  taskWorkers?: number;
   taskBroker?: TaskBroker;
   additionalTemplateFilters?: Record<string, TemplateFilter>;
   additionalTemplateGlobals?: Record<string, TemplateGlobal>;
@@ -158,6 +159,7 @@ export async function createRouter(
     database,
     catalogClient,
     actions,
+    taskWorkers,
     scheduler,
     additionalTemplateFilters,
     additionalTemplateGlobals,