typeorm 0.3.18-dev.ff6e875 → 0.3.18

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();
         }
     }
@@ -1924,7 +1931,7 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                             ? "STORED"
                             : "VIRTUAL";
                     // We cannot relay on information_schema.columns.generation_expression, because it is formatted different.
-                    const asExpressionQuery = await this.selectTypeormMetadataSql({
+                    const asExpressionQuery = this.selectTypeormMetadataSql({
                         database: dbTable["TABLE_CATALOG"],
                         schema: dbTable["TABLE_SCHEMA"],
                         table: dbTable["TABLE_NAME"],
@@ -2371,6 +2378,7 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
      * Converts MssqlParameter into real mssql parameter type.
      */
     mssqlParameterToNativeParameter(parameter) {
+        var _a, _b, _c;
         switch (this.driver.normalizeType({ type: parameter.type })) {
             case "bit":
                 return this.driver.mssql.Bit;
@@ -2395,12 +2403,24 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
             case "tinyint":
                 return this.driver.mssql.TinyInt;
             case "char":
+                if ((_a = this.driver.options.options) === null || _a === void 0 ? void 0 : _a.disableAsciiToUnicodeParamConversion) {
+                    return this.driver.mssql.Char(...parameter.params);
+                }
+                return this.driver.mssql.NChar(...parameter.params);
             case "nchar":
                 return this.driver.mssql.NChar(...parameter.params);
             case "text":
+                if ((_b = this.driver.options.options) === null || _b === void 0 ? void 0 : _b.disableAsciiToUnicodeParamConversion) {
+                    return this.driver.mssql.Text;
+                }
+                return this.driver.mssql.Ntext;
             case "ntext":
                 return this.driver.mssql.Ntext;
             case "varchar":
+                if ((_c = this.driver.options.options) === null || _c === void 0 ? void 0 : _c.disableAsciiToUnicodeParamConversion) {
+                    return this.driver.mssql.VarChar(...parameter.params);
+                }
+                return this.driver.mssql.NVarChar(...parameter.params);
             case "nvarchar":
                 return this.driver.mssql.NVarChar(...parameter.params);
             case "xml":
@@ -2451,6 +2471,12 @@ class SqlServerQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                 return ISOLATION_LEVEL.READ_COMMITTED;
         }
     }
+    /**
+     * Change table comment.
+     */
+    changeTableComment(tableOrName, comment) {
+        throw new error_1.TypeORMError(`sqlserver driver does not support change table comment.`);
+    }
 }
 exports.SqlServerQueryRunner = SqlServerQueryRunner;