refactor: collection-factory create method

nestjs · Jan 29, 2023 · 7d5cc58 · 7d5cc58
1 parent b928e38
commit 7d5cc58
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/lib/schematics/collection.factory.ts b/lib/schematics/collection.factory.ts
@@ -7,17 +7,12 @@ import { NestCollection } from './nest.collection';
 
 export class CollectionFactory {
   public static create(collection: Collection | string): AbstractCollection {
-    switch (collection) {
-      case Collection.NESTJS:
-        return new NestCollection(
-          RunnerFactory.create(Runner.SCHEMATIC) as SchematicRunner,
-        );
+    const schematicRunner = RunnerFactory.create(Runner.SCHEMATIC) as SchematicRunner;
 
-      default:
-        return new CustomCollection(
-          collection,
-          RunnerFactory.create(Runner.SCHEMATIC) as SchematicRunner,
-        );
+    if (collection === Collection.NESTJS) {
+      return new NestCollection(schematicRunner);
+    } else {
+      return new CustomCollection(collection, schematicRunner);
     }
   }
 }