@tsed/mikro-orm 8.0.0 → 8.0.2

package/src/services/MikroOrmRegistry.ts

@@ -0,0 +1,62 @@
+import {IDatabaseDriver as DatabaseDriver, MikroORM, Options} from "@mikro-orm/core";
+import {catchAsyncError, getValue} from "@tsed/core";
+import {Inject, Injectable} from "@tsed/di";
+import {Logger} from "@tsed/logger";
+
+import {MikroOrmFactory} from "./MikroOrmFactory.js";
+
+@Injectable()
+export class MikroOrmRegistry {
+  private readonly DEFAULT_CONTEXT_NAME = "default";
+  private readonly instances = new Map<string, MikroORM>();
+
+  constructor(
+    @Inject() private readonly logger: Logger,
+    @Inject() private readonly mikroOrmFactory: MikroOrmFactory
+  ) {}
+
+  public async register<T extends DatabaseDriver>(options: Options<T>): Promise<MikroORM> {
+    const contextName = getValue<string>(options, "contextName", this.DEFAULT_CONTEXT_NAME);
+
+    if (this.has(contextName)) {
+      return this.get(contextName)!;
+    }
+
+    this.logger.info(`Create connection with MikroOrm to database: %s`, contextName);
+    this.logger.debug(`options: %j`, options);
+
+    const instance = await this.mikroOrmFactory.create(options);
+
+    this.instances.set(contextName, instance);
+
+    this.logger.info(`Connected with MikroOrm to database: %s`, contextName);
+
+    return instance;
+  }
+
+  public get(contextName?: string): MikroORM | undefined {
+    return this.instances.get(contextName ?? this.DEFAULT_CONTEXT_NAME);
+  }
+
+  public has(contextName?: string): boolean {
+    return this.instances.has(contextName ?? this.DEFAULT_CONTEXT_NAME);
+  }
+
+  public values(): IterableIterator<MikroORM> {
+    return this.instances.values();
+  }
+
+  public async clear(): Promise<void> {
+    const instances = [...this.instances.values()];
+
+    await Promise.all(instances.map((instance: MikroORM) => catchAsyncError(() => this.dispose(instance))));
+
+    this.instances.clear();
+  }
+
+  private async dispose(instance: MikroORM): Promise<void> {
+    if (await instance.isConnected()) {
+      await instance.close(false);
+    }
+  }
+}