Split msnodesqlv8

Author: dhensby

lib/msnodesqlv8/connection-pool.js

@@ -0,0 +1,86 @@
+'use strict'
+
+const msnodesql = require('msnodesqlv8')
+const debug = require('debug')('mssql:msv8')
+const BaseConnectionPool = require('../base/connection-pool')
+const { IDS, INCREMENT } = require('../utils')
+const shared = require('../shared')
+const ConnectionError = require('../error/connection-error')
+
+const CONNECTION_STRING_PORT = 'Driver=SQL Server Native Client 11.0;Server=#{server},#{port};Database=#{database};Uid=#{user};Pwd=#{password};Trusted_Connection=#{trusted};Encrypt=#{encrypt};'
+const CONNECTION_STRING_NAMED_INSTANCE = 'Driver=SQL Server Native Client 11.0;Server=#{server}\\#{instance};Database=#{database};Uid=#{user};Pwd=#{password};Trusted_Connection=#{trusted};Encrypt=#{encrypt};'
+
+class ConnectionPool extends BaseConnectionPool {
+  _poolCreate () {
+    return new shared.Promise((resolve, reject) => {
+      let defaultConnectionString = CONNECTION_STRING_PORT
+
+      if (this.config.options.instanceName != null) {
+        defaultConnectionString = CONNECTION_STRING_NAMED_INSTANCE
+      }
+
+      if (this.config.requestTimeout == null) {
+        this.config.requestTimeout = 15000
+      }
+
+      const cfg = {
+        conn_str: this.config.connectionString || defaultConnectionString,
+        conn_timeout: (this.config.connectionTimeout || 15000) / 1000
+      }
+
+      cfg.conn_str = cfg.conn_str.replace(new RegExp('#{([^}]*)}', 'g'), (p) => {
+        const key = p.substr(2, p.length - 3)
+
+        switch (key) {
+          case 'instance':
+            return this.config.options.instanceName
+          case 'trusted':
+            return this.config.options.trustedConnection ? 'Yes' : 'No'
+          case 'encrypt':
+            return this.config.options.encrypt ? 'Yes' : 'No'
+          default:
+            return this.config[key] != null ? this.config[key] : ''
+        }
+      })
+
+      const connedtionId = INCREMENT.Connection++
+      debug('pool(%d): connection #%d created', IDS.get(this), connedtionId)
+      debug('connection(%d): establishing', connedtionId)
+
+      if (typeof this.config.beforeConnect === 'function') {
+        this.config.beforeConnect(cfg)
+      }
+
+      msnodesql.open(cfg, (err, tds) => {
+        if (err) {
+          err = new ConnectionError(err)
+          return reject(err)
+        }
+
+        IDS.add(tds, 'Connection', connedtionId)
+        debug('connection(%d): established', IDS.get(tds))
+        resolve(tds)
+      })
+    })
+  }
+
+  _poolValidate (tds) {
+    return tds && !tds.hasError
+  }
+
+  _poolDestroy (tds) {
+    return new shared.Promise((resolve, reject) => {
+      if (!tds) {
+        resolve()
+        return
+      }
+      debug('connection(%d): destroying', IDS.get(tds))
+      tds.close(() => {
+        resolve()
+      })
+      debug('connection(%d): destroyed', IDS.get(tds))
+    })
+  }
+}
+
+module.exports = ConnectionPool

lib/msnodesqlv8/index.js

@@ -0,0 +1,28 @@
+'use strict'
+
+const base = require('../base')
+const ConnectionPool = require('./connection-pool')
+const Transaction = require('./transaction')
+const Request = require('./request')
+
+module.exports = Object.assign({
+  ConnectionPool,
+  Transaction,
+  Request,
+  PreparedStatement: base.PreparedStatement
+}, base.exports)
+
+Object.defineProperty(module.exports, 'Promise', {
+  enumerable: true,
+  get: () => {
+    return base.Promise
+  },
+  set: (value) => {
+    base.Promise = value
+  }
+})
+
+base.driver.name = 'msnodesqlv8'
+base.driver.ConnectionPool = ConnectionPool
+base.driver.Transaction = Transaction
+base.driver.Request = Request