Adds onReserveConnection to Postgres and MySQL dialect configs (kys…

…ely-org#996)

Co-authored-by: Igal Klebanov <[email protected]>

src/dialect/mysql/mysql-dialect-config.ts

@@ -19,6 +19,11 @@ export interface MysqlDialectConfig {
    * Called once for each created connection.
    */
   onCreateConnection?: (connection: DatabaseConnection) => Promise<void>
+
+  /**
+   * Called every time a connection is acquired from the connection pool.
+   */
+  onReserveConnection?: (connection: DatabaseConnection) => Promise<void>
 }
 
 /**

src/dialect/mysql/mysql-driver.ts

@@ -47,6 +47,10 @@ export class MysqlDriver implements Driver {
       }
     }
 
+    if (this.#config?.onReserveConnection) {
+      await this.#config.onReserveConnection(connection)
+    }
+
     return connection
   }
 

src/dialect/postgres/postgres-dialect-config.ts

@@ -31,6 +31,11 @@ export interface PostgresDialectConfig {
    * Called once for each created connection.
    */
   onCreateConnection?: (connection: DatabaseConnection) => Promise<void>
+
+  /**
+   * Called every time a connection is acquired from the pool.
+   */
+  onReserveConnection?: (connection: DatabaseConnection) => Promise<void>
 }
 
 /**

src/dialect/postgres/postgres-driver.ts

@@ -48,6 +48,10 @@ export class PostgresDriver implements Driver {
       }
     }
 
+    if (this.#config.onReserveConnection) {
+      await this.#config.onReserveConnection(connection)
+    }
+
     return connection
   }