@breautek/storm 4.5.0 → 4.6.1

package/src/MySQLConnection.ts

@@ -27,6 +27,8 @@ import * as SQLFormatter from 'sql-formatter';
 import { Logger, LogLevel } from '@arashi/logger';
 import { StormError } from './StormError';
 import { DeadLockError } from './DeadLockError';
+import { IsolationLevel } from './IsolationLevel';
+import {SetIsolationLevelQuery} from './private/SetIsolationLevelQuery';
 
 const DEFAULT_HIGH_WATERMARK: number = 512; // in number of result objects
 const TAG: string = 'MySQLConnection';
@@ -141,26 +143,28 @@ export class MySQLConnection extends DatabaseConnection<MySQL.PoolConnection> {
         return queryObject.stream(streamOptions);
     }
 
-    public startTransaction(): Promise<void> {
+    public override async startTransaction(isolationLevel?: IsolationLevel): Promise<void> {
         if (this.isReadOnly()) {
-            return Promise.reject(new Error('A readonly connection cannot start a transaction.'));
+            throw new Error('A readonly connection cannot start a transaction.')
         }
 
         if (this.isTransaction()) {
-            return Promise.reject(new Error('Connection is already in a transaction.'));
+            throw new Error('Connection is already in a transaction.');
         }
 
         this.$transaction = true;
 
-        return new Promise<void>((resolve, reject) => {
-            this.query(startTransactionQuery).then(() => {
-                resolve();
-            }).catch((ex) => {
-                this.$transaction = false;
-                getInstance().getLogger().error(TAG, ex);
-                reject(ex);
-            });
-        });
+        try {
+            if (isolationLevel) {
+                await new SetIsolationLevelQuery(isolationLevel).execute(this);
+            }
+            await startTransactionQuery.execute(this);
+        }
+        catch (ex) {
+            this.$transaction = false;
+            getInstance().getLogger().error(TAG, ex);
+            throw ex;
+        }
     }
 
     public endTransaction(requiresRollback: boolean = false): Promise<void> {

package/src/api.ts

@@ -35,6 +35,7 @@ export {IInsertQueryResult} from './IInsertQueryResult';
 export {IDeleteQueryResult} from './IDeleteQueryResult';
 export {IUpdateQueryResult} from './IUpdateQueryResult';
 export {SetSessionVariableQuery, ISetSessionVariableQueryInput} from './SetSessionVariableQuery';
+export {IsolationLevel} from './IsolationLevel';
 
 // Errors
 export {ErrorCode} from './ErrorCode';

package/src/private/SetIsolationLevelQuery.ts

@@ -0,0 +1,39 @@
+/*
+   Copyright 2017-2021 Norman Breau
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+import { IsolationLevel } from '../IsolationLevel';
+import {Query} from '../Query';
+
+export class SetIsolationLevelQuery extends Query<IsolationLevel, void> {
+    protected _getQuery(): string {
+        let level: string;
+        switch (this.getParameters()) {
+            case IsolationLevel.READ_COMMITTED:
+                level = 'READ COMMITTED';
+                break;
+            case IsolationLevel.READ_UNCOMMITTED:
+                level = 'READ UNCOMMITTED';
+                break;
+            case IsolationLevel.REPEATABLE_READ:
+                level = 'REPEATABLE READ';
+                break;
+            case IsolationLevel.SERIALIZABLE:
+                level = 'SERIALIZABLE';
+                break;
+        }
+        return `SET TRANSACTION ISOLATION LEVEL ${level}`;
+    }
+}