typeorm 0.3.17 → 0.3.18-dev.0f11739

package/driver/cockroachdb/CockroachQueryRunner.js

@@ -65,7 +65,12 @@ class CockroachQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                 .then(([connection, release]) => {
                 this.driver.connectedQueryRunners.push(this);
                 this.databaseConnection = connection;
-                this.releaseCallback = release;
+                const onErrorCallback = (err) => this.releaseConnection(err);
+                this.releaseCallback = (err) => {
+                    this.databaseConnection.removeListener("error", onErrorCallback);
+                    release(err);
+                };
+                this.databaseConnection.on("error", onErrorCallback);
                 return this.databaseConnection;
             });
         }
@@ -76,27 +81,41 @@ class CockroachQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
                 .then(([connection, release]) => {
                 this.driver.connectedQueryRunners.push(this);
                 this.databaseConnection = connection;
-                this.releaseCallback = release;
+                const onErrorCallback = (err) => this.releaseConnection(err);
+                this.releaseCallback = (err) => {
+                    this.databaseConnection.removeListener("error", onErrorCallback);
+                    release(err);
+                };
+                this.databaseConnection.on("error", onErrorCallback);
                 return this.databaseConnection;
             });
         }
         return this.databaseConnectionPromise;
     }
     /**
-     * Releases used database connection.
-     * You cannot use query runner methods once its released.
+     * Release a connection back to the pool, optionally specifying an Error to release with.
+     * Per pg-pool documentation this will prevent the pool from re-using the broken connection.
      */
-    release() {
+    async releaseConnection(err) {
         if (this.isReleased) {
-            return Promise.resolve();
+            return;
         }
         this.isReleased = true;
-        if (this.releaseCallback)
-            this.releaseCallback();
+        if (this.releaseCallback) {
+            this.releaseCallback(err);
+            this.releaseCallback = undefined;
+        }
         const index = this.driver.connectedQueryRunners.indexOf(this);
-        if (index !== -1)
-            this.driver.connectedQueryRunners.splice(index);
-        return Promise.resolve();
+        if (index !== -1) {
+            this.driver.connectedQueryRunners.splice(index, 1);
+        }
+    }
+    /**
+     * Releases used database connection.
+     * You cannot use query runner methods once its released.
+     */
+    release() {
+        return this.releaseConnection();
     }
     /**
      * Starts transaction.
@@ -112,6 +131,7 @@ class CockroachQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
             throw err;
         }
         if (this.transactionDepth === 0) {
+            this.transactionDepth += 1;
             await this.query("START TRANSACTION");
             await this.query("SAVEPOINT cockroach_restart");
             if (isolationLevel) {
@@ -119,9 +139,9 @@ class CockroachQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
             }
         }
         else {
-            await this.query(`SAVEPOINT typeorm_${this.transactionDepth}`);
+            this.transactionDepth += 1;
+            await this.query(`SAVEPOINT typeorm_${this.transactionDepth - 1}`);
         }
-        this.transactionDepth += 1;
         this.storeQueries = true;
         await this.broadcaster.broadcast("AfterTransactionStart");
     }
@@ -134,17 +154,17 @@ class CockroachQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
             throw new TransactionNotStartedError_1.TransactionNotStartedError();
         await this.broadcaster.broadcast("BeforeTransactionCommit");
         if (this.transactionDepth > 1) {
-            await this.query(`RELEASE SAVEPOINT typeorm_${this.transactionDepth - 1}`);
             this.transactionDepth -= 1;
+            await this.query(`RELEASE SAVEPOINT typeorm_${this.transactionDepth}`);
         }
         else {
             this.storeQueries = false;
+            this.transactionDepth -= 1;
             await this.query("RELEASE SAVEPOINT cockroach_restart");
             await this.query("COMMIT");
             this.queries = [];
             this.isTransactionActive = false;
             this.transactionRetries = 0;
-            this.transactionDepth -= 1;
         }
         await this.broadcaster.broadcast("AfterTransactionCommit");
     }
@@ -157,16 +177,17 @@ class CockroachQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
             throw new TransactionNotStartedError_1.TransactionNotStartedError();
         await this.broadcaster.broadcast("BeforeTransactionRollback");
         if (this.transactionDepth > 1) {
-            await this.query(`ROLLBACK TO SAVEPOINT typeorm_${this.transactionDepth - 1}`);
+            this.transactionDepth -= 1;
+            await this.query(`ROLLBACK TO SAVEPOINT typeorm_${this.transactionDepth}`);
         }
         else {
             this.storeQueries = false;
+            this.transactionDepth -= 1;
             await this.query("ROLLBACK");
             this.queries = [];
             this.isTransactionActive = false;
             this.transactionRetries = 0;
         }
-        this.transactionDepth -= 1;
         await this.broadcaster.broadcast("AfterTransactionRollback");
     }
     /**