Add a 10 second timeout for in _poolCreate method for tedious

qiuguoqiang · Jul 23, 2018 · 89784e7 · 89784e7
1 parent 528bb62
commit 89784e7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/tedious.js b/lib/tedious.js
@@ -231,8 +231,10 @@ class ConnectionPool extends base.ConnectionPool {
       IDS.add(tedious, 'Connection')
       debug('pool(%d): connection #%d created', IDS.get(this), IDS.get(tedious))
       debug('connection(%d): establishing', IDS.get(tedious))
+      const emittedConnectEvent = false
 
       tedious.once('connect', err => {
+        emittedConnectEvent = true
         if (err) {
           err = new base.ConnectionError(err)
           return reject(err)
@@ -242,6 +244,12 @@ class ConnectionPool extends base.ConnectionPool {
         resolve(tedious)
       })
 
+      setTimeout(() => {
+        if (!emittedConnectEvent) {
+          reject(new base.ConnectionError('Tedious connection timed out after 10 seconds'))
+        }
+      }, 10 * 1000)
+
       tedious.on('error', err => {
         if (err.code === 'ESOCKET') {
           tedious.hasError = true