feat!: make ServerlessClient EventEmitter

Author: perrin4869

index.d.ts

@@ -1,4 +1,6 @@
+import EventEmitter = require("events");
 import stream = require("stream");
+import pg = require("pg");
 
 declare interface TlsOptions {
   rejectUnauthorized?: boolean;
@@ -75,7 +77,7 @@ declare namespace ServerlessClient {
   export { TlsOptions, Config };
 }
 
-declare class ServerlessClient {
+declare class ServerlessClient extends EventEmitter {
   constructor(config: Config)
 
   clean(): Promise<number | undefined>
@@ -88,7 +90,7 @@ declare class ServerlessClient {
 
   end(): Promise<any>
 
-  on(...args): void
+  on(event: "init", listener: (client: pg.Client) => void): this
 }
 
 export = ServerlessClient

package-lock.json

@@ -6,10 +6,13 @@
  * @license MIT
  */
 
+const EventEmitter = require("events");
 const {isValidStrategy, type, validateNum, isWithinRange} = require("./utils");
 const Postgres = require("./postgres");
 
 function ServerlessClient(config) {
+  EventEmitter.call(this);
+
   this._client = null;
   if (config.plugin) {
     this._plugin = config.plugin
@@ -20,6 +23,9 @@ function ServerlessClient(config) {
   this.setConfig(config)
 }
 
+ServerlessClient.prototype = Object.create(EventEmitter.prototype);
+ServerlessClient.prototype.constructor = ServerlessClient;
+
 ServerlessClient.prototype.constructor = ServerlessClient;
 ServerlessClient.prototype._sleep = delay =>
   new Promise(resolve => {
@@ -201,6 +207,7 @@ ServerlessClient.prototype._init = async function () {
     await this._setMaxConnections(this)
   }
 
+  this.emit("init", this._client);
   this._logger("Max connections: ", this._maxConns.cache.total)
 }
 
@@ -423,8 +430,4 @@ ServerlessClient.prototype.end = async function () {
   this._client = null
 }
 
-ServerlessClient.prototype.on = function (...args) {
-  this._client.on(...args)
-}
-
 module.exports = {ServerlessClient};