fix: made type-metadata.storage.ts much faster

Changed the data modeling of type-metadata storage to store the data in maps instead of arrays.
This results in a dramatic improvement of application startup time since many of the lookups for metadata are now in O(1) instead of O(n)

Author: roypeled

packages/graphql/lib/schema-builder/collections/array.collection.ts

@@ -0,0 +1,15 @@
+export class ArrayCollection<T> extends Array<T> {
+  constructor(private globalArray: Array<T>) {
+    super();
+  }
+
+  push(...items): number {
+    this.globalArray.push(...items);
+    return super.push(...items);
+  }
+
+  unshift(...items): number {
+    this.globalArray.unshift(...items);
+    return super.unshift(...items);
+  }
+}

packages/graphql/lib/schema-builder/collections/field.directive.collection.ts

@@ -0,0 +1,18 @@
+import { MetadataListByNameCollection } from './metadata.list.by.name.collection';
+import { PropertyDirectiveMetadata } from '../metadata';
+
+export class FieldDirectiveCollection extends MetadataListByNameCollection<PropertyDirectiveMetadata> {
+  sdls = new Set<string>();
+  fieldNames = new Set<string>();
+
+  add(value: PropertyDirectiveMetadata) {
+    if (this.sdls.has(value.sdl) && this.fieldNames.has(value.fieldName))
+      return;
+
+    super.add(value, value.fieldName);
+
+    this.sdls.add(value.sdl);
+    this.fieldNames.add(value.fieldName);
+    this.globalArray && this.globalArray.push(value);
+  }
+}

packages/graphql/lib/schema-builder/collections/index.ts

@@ -0,0 +1,7 @@
+export * from './metadata.storage.collection.list';
+export * from './field.directive.collection';
+export * from './array.collection';
+export * from './metada.collection.model.interface';
+export * from './metadata.by.name.collection';
+export * from './metadata.list.by.name.collection';
+export * from './metadata.storage.collection';

packages/graphql/lib/schema-builder/collections/metada.collection.model.interface.ts

@@ -0,0 +1,14 @@
+import { ClassMetadata, ResolverClassMetadata } from '../metadata';
+import { ObjectTypeMetadata } from '../metadata/object-type.metadata';
+
+export interface MetadataCollectionModelInterface {
+  argumentType: ClassMetadata[];
+  interface: ClassMetadata[];
+  inputType: ClassMetadata[];
+  objectType: ObjectTypeMetadata[];
+  resolver: ResolverClassMetadata[];
+  classDirectives: [];
+  classExtensions: [];
+  fieldDirectives: [];
+  fieldExtensions: [];
+}

packages/graphql/lib/schema-builder/collections/metadata.by.name.collection.ts

@@ -0,0 +1,26 @@
+export class MetadataByNameCollection<T> {
+  protected map = new Map<string, T>();
+  protected all: (T extends any[] ? T[number] : T)[] = [];
+
+  getAll() {
+    return this.all;
+  }
+
+  getByName(name: string) {
+    return this.map.get(name);
+  }
+
+  add(value: T extends any[] ? T[number] : T, name: string) {
+    if (this.map.has(name)) return;
+
+    this.map.set(name, value);
+    this.all.push(value);
+  }
+
+  unshift(value: T extends any[] ? T[number] : T, name: string) {
+    if (this.map.has(name)) return;
+
+    this.map.set(name, value);
+    this.all.unshift(value);
+  }
+}

packages/graphql/lib/schema-builder/collections/metadata.list.by.name.collection.ts

@@ -0,0 +1,37 @@
+import { MetadataByNameCollection } from './metadata.by.name.collection';
+
+export class MetadataListByNameCollection<T> extends MetadataByNameCollection<
+  T[]
+> {
+  constructor(protected globalArray: Array<T> = null) {
+    super();
+  }
+
+  getByName(name: string): T[] {
+    return super.getByName(name) || [];
+  }
+
+  add(value: T, name: string) {
+    let arrayResult = super.getByName(name);
+    if (!arrayResult) {
+      arrayResult = [];
+      this.map.set(name, arrayResult);
+    }
+
+    arrayResult.push(value);
+    this.all.push(value);
+    this.globalArray && this.globalArray.push(value);
+  }
+
+  unshift(value: T, name: string) {
+    let arrayResult = super.getByName(name);
+    if (!arrayResult) {
+      arrayResult = [];
+      this.map.set(name, arrayResult);
+    }
+
+    arrayResult.unshift(value);
+    this.all.push(value);
+    this.globalArray && this.globalArray.unshift(value);
+  }
+}

packages/graphql/lib/schema-builder/collections/metadata.storage.collection.list.ts

@@ -0,0 +1,44 @@
+import { MetadataStorageCollection } from './metadata.storage.collection';
+import { MetadataCollectionModelInterface } from './metada.collection.model.interface';
+
+export class MetadataStorageCollectionList {
+  private storageMap = new Map<Function, MetadataStorageCollection>();
+  private storageList = new Array<MetadataStorageCollection>();
+
+  public all: MetadataCollectionModelInterface = {
+    argumentType: [],
+    interface: [],
+    inputType: [],
+    objectType: [],
+    resolver: [],
+    classDirectives: [],
+    classExtensions: [],
+    fieldDirectives: [],
+    fieldExtensions: [],
+  };
+
+  get(target: Function) {
+    let metadata = this.storageMap.get(target);
+
+    if (!metadata) {
+      metadata = new MetadataStorageCollection(this.all);
+      this.storageMap.set(target, metadata);
+      this.storageList.push(metadata);
+    }
+
+    return metadata;
+  }
+
+  compile() {
+    this.reversePredicate((t) => t.classDirectives);
+    this.reversePredicate((t) => t.classExtensions);
+    this.reversePredicate((t) => t.fieldDirectives.getAll());
+    this.reversePredicate((t) => t.fieldExtensions.getAll());
+  }
+
+  private reversePredicate<V>(
+    predicate: (t: MetadataStorageCollection) => Array<V>,
+  ) {
+    this.storageList.forEach((t) => predicate(t).reverse());
+  }
+}

packages/graphql/lib/schema-builder/collections/metadata.storage.collection.ts

@@ -0,0 +1,79 @@
+import { MetadataByNameCollection } from './metadata.by.name.collection';
+import {
+  ClassDirectiveMetadata,
+  ClassExtensionsMetadata,
+  ClassMetadata,
+  MethodArgsMetadata,
+  PropertyExtensionsMetadata,
+  PropertyMetadata,
+  ResolverClassMetadata,
+} from '../metadata';
+import { MetadataListByNameCollection } from './metadata.list.by.name.collection';
+import { FieldDirectiveCollection } from './field.directive.collection';
+import { ObjectTypeMetadata } from '../metadata/object-type.metadata';
+import { ArrayCollection } from './array.collection';
+import { MetadataCollectionModelInterface } from './metada.collection.model.interface';
+
+export class MetadataStorageCollection {
+  constructor(private all: MetadataCollectionModelInterface) {}
+
+  fields = new MetadataByNameCollection<PropertyMetadata>();
+  params = new MetadataListByNameCollection<MethodArgsMetadata>();
+  fieldDirectives = new FieldDirectiveCollection(this.all.fieldDirectives);
+  fieldExtensions =
+    new MetadataListByNameCollection<PropertyExtensionsMetadata>(
+      this.all.fieldExtensions,
+    );
+  classDirectives = new ArrayCollection<ClassDirectiveMetadata>(
+    this.all.classDirectives,
+  );
+  classExtensions = new ArrayCollection<ClassExtensionsMetadata>(
+    this.all.classExtensions,
+  );
+
+  set argumentType(val: ClassMetadata) {
+    this._argumentType = val;
+    this.all.argumentType.push(val);
+  }
+  get argumentType() {
+    return this._argumentType;
+  }
+
+  set interface(val: ClassMetadata) {
+    this._interface = val;
+    this.all.interface.push(val);
+  }
+  get interface() {
+    return this._interface;
+  }
+
+  set inputType(val: ClassMetadata) {
+    this._inputType = val;
+    this.all.inputType.push(val);
+  }
+  get inputType() {
+    return this._inputType;
+  }
+
+  set objectType(val: ObjectTypeMetadata) {
+    this._objectType = val;
+    this.all.objectType.push(val);
+  }
+  get objectType() {
+    return this._objectType;
+  }
+
+  set resolver(val: ResolverClassMetadata) {
+    this._resolver = val;
+    this.all.resolver.push(val);
+  }
+  get resolver() {
+    return this._resolver;
+  }
+
+  _argumentType: ClassMetadata;
+  _interface: ClassMetadata;
+  _inputType: ClassMetadata;
+  _objectType: ObjectTypeMetadata;
+  _resolver: ResolverClassMetadata;
+}