Commit iniziale

node_modules/mssql/lib/msnodesqlv8/transaction.js

@@ -0,0 +1,98 @@
+'use strict'
+
+const debug = require('debug')('mssql:msv8')
+const BaseTransaction = require('../base/transaction')
+const { IDS } = require('../utils')
+const Request = require('./request')
+const ISOLATION_LEVEL = require('../isolationlevel')
+const TransactionError = require('../error/transaction-error')
+
+const isolationLevelDeclaration = function (type) {
+  switch (type) {
+    case ISOLATION_LEVEL.READ_UNCOMMITTED: return 'READ UNCOMMITTED'
+    case ISOLATION_LEVEL.READ_COMMITTED: return 'READ COMMITTED'
+    case ISOLATION_LEVEL.REPEATABLE_READ: return 'REPEATABLE READ'
+    case ISOLATION_LEVEL.SERIALIZABLE: return 'SERIALIZABLE'
+    case ISOLATION_LEVEL.SNAPSHOT: return 'SNAPSHOT'
+    default: throw new TransactionError('Invalid isolation level.')
+  }
+}
+
+class Transaction extends BaseTransaction {
+  _begin (isolationLevel, callback) {
+    super._begin(isolationLevel, err => {
+      if (err) return callback(err)
+
+      debug('transaction(%d): begin', IDS.get(this))
+
+      this.parent.acquire(this, (err, connection, config) => {
+        if (err) return callback(err)
+
+        this._acquiredConnection = connection
+        this._acquiredConfig = config
+
+        const req = new Request(this)
+        req.stream = false
+        req.query(`set transaction isolation level ${isolationLevelDeclaration(this.isolationLevel)};begin tran;`, err => {
+          if (err) {
+            this.parent.release(this._acquiredConnection)
+            this._acquiredConnection = null
+            this._acquiredConfig = null
+
+            return callback(err)
+          }
+
+          debug('transaction(%d): begun', IDS.get(this))
+
+          callback(null)
+        })
+      })
+    })
+  }
+
+  _commit (callback) {
+    super._commit(err => {
+      if (err) return callback(err)
+
+      debug('transaction(%d): commit', IDS.get(this))
+
+      const req = new Request(this)
+      req.stream = false
+      req.query('commit tran', err => {
+        if (err) err = new TransactionError(err)
+
+        this.parent.release(this._acquiredConnection)
+        this._acquiredConnection = null
+        this._acquiredConfig = null
+
+        if (!err) debug('transaction(%d): commited', IDS.get(this))
+
+        callback(null)
+      })
+    })
+  }
+
+  _rollback (callback) {
+    super._commit(err => {
+      if (err) return callback(err)
+
+      debug('transaction(%d): rollback', IDS.get(this))
+
+      const req = new Request(this)
+      req.stream = false
+      req.query('rollback tran', err => {
+        if (err) err = new TransactionError(err)
+
+        this.parent.release(this._acquiredConnection)
+        this._acquiredConnection = null
+        this._acquiredConfig = null
+
+        if (!err) debug('transaction(%d): rolled back', IDS.get(this))
+
+        callback(null)
+      })
+    })
+  }
+}
+
+module.exports = Transaction