typeorm 0.3.18-dev.c8ee5b1 → 0.3.18-dev.dd59524

package/driver/sqlserver/SqlServerQueryRunner.js

@@ -20,6 +20,7 @@ const error_1 = require("../../error");
 const QueryLock_1 = require("../../query-runner/QueryLock");
 const MetadataTableType_1 = require("../types/MetadataTableType");
 const InstanceChecker_1 = require("../../util/InstanceChecker");
+const BroadcasterResult_1 = require("../../subscriber/BroadcasterResult");
 /**
  * Runs queries on a single SQL Server database connection.
  */
@@ -79,6 +80,7 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                 ok();
             };
             if (this.transactionDepth === 0) {
+                this.transactionDepth += 1;
                 const pool = await (this.mode === "slave"
                     ? this.driver.obtainSlaveConnection()
                     : this.driver.obtainMasterConnection());
@@ -93,10 +95,10 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                 }
             }
             else {
-                await this.query(`SAVE TRANSACTION typeorm_${this.transactionDepth}`);
+                this.transactionDepth += 1;
+                await this.query(`SAVE TRANSACTION typeorm_${this.transactionDepth - 1}`);
                 ok();
             }
-            this.transactionDepth += 1;
         });
         await this.broadcaster.broadcast("AfterTransactionStart");
     }
@@ -112,6 +114,7 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
         await this.broadcaster.broadcast("BeforeTransactionCommit");
         if (this.transactionDepth === 1) {
             return new Promise((ok, fail) => {
+                this.transactionDepth -= 1;
                 this.databaseConnection.commit(async (err) => {
                     if (err)
                         return fail(err);
@@ -120,7 +123,6 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                     await this.broadcaster.broadcast("AfterTransactionCommit");
                     ok();
                     this.connection.logger.logQuery("COMMIT");
-                    this.transactionDepth -= 1;
                 });
             });
         }
@@ -137,11 +139,12 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
             throw new TransactionNotStartedError_1.TransactionNotStartedError();
         await this.broadcaster.broadcast("BeforeTransactionRollback");
         if (this.transactionDepth > 1) {
-            await this.query(`ROLLBACK TRANSACTION typeorm_${this.transactionDepth - 1}`);
             this.transactionDepth -= 1;
+            await this.query(`ROLLBACK TRANSACTION typeorm_${this.transactionDepth}`);
         }
         else {
             return new Promise((ok, fail) => {
+                this.transactionDepth -= 1;
                 this.databaseConnection.rollback(async (err) => {
                     if (err)
                         return fail(err);
@@ -150,7 +153,6 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                     await this.broadcaster.broadcast("AfterTransactionRollback");
                     ok();
                     this.connection.logger.logQuery("ROLLBACK");
-                    this.transactionDepth -= 1;
                 });
             });
         }
@@ -162,8 +164,10 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
         if (this.isReleased)
             throw new QueryRunnerAlreadyReleasedError_1.QueryRunnerAlreadyReleasedError();
         const release = await this.lock.acquire();
+        const broadcasterResult = new BroadcasterResult_1.BroadcasterResult();
         try {
             this.driver.connection.logger.logQuery(query, parameters, this);
+            this.broadcaster.broadcastBeforeQueryEvent(broadcasterResult, query, parameters);
             const pool = await (this.mode === "slave"
                 ? this.driver.obtainSlaveConnection()
                 : this.driver.obtainMasterConnection());
@@ -192,6 +196,7 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                     const maxQueryExecutionTime = this.driver.options.maxQueryExecutionTime;
                     const queryEndTime = +new Date();
                     const queryExecutionTime = queryEndTime - queryStartTime;
+                    this.broadcaster.broadcastAfterQueryEvent(broadcasterResult, query, parameters, true, queryExecutionTime, raw, undefined);
                     if (maxQueryExecutionTime &&
                         queryExecutionTime > maxQueryExecutionTime) {
                         this.driver.connection.logger.logQuerySlow(queryExecutionTime, query, parameters, this);
@@ -227,9 +232,11 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
         }
         catch (err) {
             this.driver.connection.logger.logQueryError(err, query, parameters, this);
+            this.broadcaster.broadcastAfterQueryEvent(broadcasterResult, query, parameters, false, undefined, undefined, err);
             throw err;
         }
         finally {
+            await broadcasterResult.wait();
             release();
         }
     }