feat(database): overload connection with new drivers

AthennaIO · Nov 6, 2023 · c01f7da · c01f7da
1 parent e6d4c59
commit c01f7da
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/database/DatabaseImpl.ts b/src/database/DatabaseImpl.ts
@@ -9,6 +9,7 @@
 
 import type { Knex } from 'knex'
 import type { Driver } from '#src/drivers/Driver'
+import type { Collection, Connection } from 'mongoose'
 import { DriverFactory } from '#src/factories/DriverFactory'
 import { QueryBuilder } from '#src/database/builders/QueryBuilder'
 import { ConnectionFactory } from '#src/factories/ConnectionFactory'
@@ -35,6 +36,21 @@ export class DatabaseImpl<Client = any, QB = any> {
     this.connect(athennaDbOpts)
   }
 
+  public connection(
+    connection: 'mongo',
+    options?: ConnectionOptions
+  ): DatabaseImpl<Connection, Collection>
+
+  public connection(
+    connection: 'mysql',
+    options?: ConnectionOptions
+  ): DatabaseImpl<Knex, Knex.QueryBuilder>
+
+  public connection(
+    connection: 'sqlite',
+    options?: ConnectionOptions
+  ): DatabaseImpl<Knex, Knex.QueryBuilder>
+
   public connection(
     connection: 'postgres',
     options?: ConnectionOptions