From 2f8f2a63211e92a2a5dc4360b8b28b0add982864 Mon Sep 17 00:00:00 2001
From: Lei Zhao <leizha@uber.com>
Date: Tue, 4 Jul 2017 13:30:23 -0700
Subject: [PATCH] Pool.end() accepts an optional callback param

---
 types/pg/index.d.ts  | 2 +-
 types/pg/pg-tests.ts | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/types/pg/index.d.ts b/types/pg/index.d.ts
index 2b442b1227..6718fb3f12 100644
--- a/types/pg/index.d.ts
+++ b/types/pg/index.d.ts
@@ -71,7 +71,7 @@ export declare class Pool extends events.EventEmitter {
     connect(): Promise<Client>;
     connect(callback: (err: Error, client: Client, done: () => void) => void): void;
 
-    end(): Promise<void>;
+    end(callback?: () => void): Promise<void>;
 
     query(queryStream: QueryConfig & stream.Readable): stream.Readable;
     query(queryTextOrConfig: string | QueryConfig): Promise<QueryResult>;
diff --git a/types/pg/pg-tests.ts b/types/pg/pg-tests.ts
index 73c18ac1b7..4b4cfba7ec 100644
--- a/types/pg/pg-tests.ts
+++ b/types/pg/pg-tests.ts
@@ -74,3 +74,8 @@ pool.connect((err, client, done) => {
 pool.on('error', (err, client) => {
   console.error('idle client error', err.message, err.stack)
 })
+
+pool.end();
+pool.end(() => {
+    console.log("pool is closed");
+});