feat(PostgreSQL): manual commit mode

src/main/libs/clients/MySQLClient.js

@@ -104,6 +104,7 @@ export class MySQLClient extends AntaresCore {
    /**
     *
     * @returns dbConfig
+    * @memberof MySQLClient
     */
    async getDbConfig () {
       delete this._params.application_name;
@@ -159,7 +160,15 @@ export class MySQLClient extends AntaresCore {
 
    async getConnection () {
       const dbConfig = await this.getDbConfig();
-      const connection = await mysql.createConnection(dbConfig);
+      const connection = await mysql.createConnection({
+         ...dbConfig,
+         typeCast: (field, next) => {
+            if (field.type === 'DATETIME')
+               return field.string();
+            else
+               return next();
+         }
+      });
 
       // ANSI_QUOTES check
       const [res] = await connection.query('SHOW GLOBAL VARIABLES LIKE \'%sql_mode%\'');
@@ -196,12 +205,12 @@ export class MySQLClient extends AntaresCore {
       if (hasAnsiQuotes)
          await connection.query(`SET SESSION sql_mode = "${sqlMode.filter(m => m !== 'ANSI_QUOTES').join(',')}"`);
 
-      connection.on('connection', connection => {
+      connection.on('connection', conn => {
          if (this._params.readonly)
-            connection.query('SET SESSION TRANSACTION READ ONLY');
+            conn.query('SET SESSION TRANSACTION READ ONLY');
 
          if (hasAnsiQuotes)
-            connection.query(`SET SESSION sql_mode = "${sqlMode.filter(m => m !== 'ANSI_QUOTES').join(',')}"`);
+            conn.query(`SET SESSION sql_mode = "${sqlMode.filter(m => m !== 'ANSI_QUOTES').join(',')}"`);
       });
 
       return connection;

src/main/libs/clients/PostgreSQLClient.js

@@ -22,6 +22,7 @@ export class PostgreSQLClient extends AntaresCore {
 
       this._schema = null;
       this._runningConnections = new Map();
+      this._connectionsToCommit = new Map();
 
       this.types = {};
       for (const key in types.builtins)
@@ -71,9 +72,11 @@ export class PostgreSQLClient extends AntaresCore {
    }
 
    /**
+    *
+    * @returns dbConfig
     * @memberof PostgreSQLClient
     */
-   async connect () {
+   async getDbConfig () {
       const dbConfig = {
          host: this._params.host,
          port: this._params.port,
@@ -102,24 +105,43 @@ export class PostgreSQLClient extends AntaresCore {
          }
       }
 
-      if (!this._poolSize) {
+      return dbConfig;
-         const client = new Client(dbConfig);
+   }
-         await client.connect();
-         this._connection = client;
 
-         if (this._params.readonly)
+   /**
-            await this.raw('SET SESSION CHARACTERISTICS AS TRANSACTION READ ONLY');
+    * @memberof PostgreSQLClient
-      }
+    */
-      else {
+   async connect () {
-         const pool = new Pool({ ...dbConfig, max: this._poolSize });
+      if (!this._poolSize)
-         this._connection = pool;
+         this._connection = await this.getConnection();
+      else
+         this._connection = await this.getConnectionPool();
+   }
 
-         if (this._params.readonly) {
+   async getConnection () {
-            this._connection.on('connect', connection => {
+      const dbConfig = await this.getDbConfig();
-               connection.query('SET SESSION CHARACTERISTICS AS TRANSACTION READ ONLY');
+      const client = new Client(dbConfig);
-            });
+      await client.connect();
-         }
+      const connection = client;
+
+      if (this._params.readonly)
+         await connection.query('SET SESSION CHARACTERISTICS AS TRANSACTION READ ONLY');
+
+      return connection;
+   }
+
+   async getConnectionPool () {
+      const dbConfig = await this.getDbConfig();
+      const pool = new Pool({ ...dbConfig, max: this._poolSize });
+      const connection = pool;
+
+      if (this._params.readonly) {
+         connection.on('connect', conn => {
+            conn.query('SET SESSION CHARACTERISTICS AS TRANSACTION READ ONLY');
+         });
       }
+
+      return connection;
    }
 
    /**
@@ -1118,6 +1140,36 @@ export class PostgreSQLClient extends AntaresCore {
          return await this.raw(`SELECT pg_cancel_backend(${id})`);
    }
 
+   /**
+    *
+    * @param {string} tabUid
+    * @returns {Promise<null>}
+    */
+   async commitTab (tabUid) {
+      const connection = this._connectionsToCommit.get(tabUid);
+      if (connection)
+         return await connection.query('COMMIT');
+   }
+
+   /**
+    *
+    * @param {string} tabUid
+    * @returns {Promise<null>}
+    */
+   async rollbackTab (tabUid) {
+      const connection = this._connectionsToCommit.get(tabUid);
+      if (connection)
+         return await connection.query('ROLLBACK');
+   }
+
+   destroyConnectionToCommit (tabUid) {
+      const connection = this._connectionsToCommit.get(tabUid);
+      if (connection) {
+         connection.destroy();
+         this._connectionsToCommit.delete(tabUid);
+      }
+   }
+
    /**
     * CREATE TABLE
     *
@@ -1429,6 +1481,7 @@ export class PostgreSQLClient extends AntaresCore {
          details: false,
          split: true,
          comments: true,
+         autocommit: true,
          ...args
       };
 
@@ -1443,8 +1496,20 @@ export class PostgreSQLClient extends AntaresCore {
             .map(q => q.trim())
          : [sql];
 
+      let connection;
       const isPool = this._connection instanceof Pool;
-      const connection = isPool ? await this._connection.connect() : this._connection;
+
+      if (!args.autocommit && args.tabUid) { // autocommit OFF
+         if (this._connectionsToCommit.has(args.tabUid))
+            connection = this._connectionsToCommit.get(args.tabUid);
+         else {
+            connection = await this.getConnection();
+            await connection.query('START TRANSACTION');
+            this._connectionsToCommit.set(args.tabUid, connection);
+         }
+      }
+      else// autocommit ON
+         connection = isPool ? await this._connection.connect() : this._connection;
 
       if (args.tabUid && isPool)
          this._runningConnections.set(args.tabUid, connection.processID);
@@ -1554,7 +1619,7 @@ export class PostgreSQLClient extends AntaresCore {
                               });
                            }
                            catch (err) {
-                              if (isPool) {
+                              if (isPool && args.autocommit) {
                                  connection.release();
                                  this._runningConnections.delete(args.tabUid);
                               }
@@ -1566,7 +1631,7 @@ export class PostgreSQLClient extends AntaresCore {
                               keysArr = keysArr ? [...keysArr, ...response] : response;
                            }
                            catch (err) {
-                              if (isPool) {
+                              if (isPool && args.autocommit) {
                                  connection.release();
                                  this._runningConnections.delete(args.tabUid);
                               }
@@ -1585,7 +1650,7 @@ export class PostgreSQLClient extends AntaresCore {
                   });
                }
                catch (err) {
-                  if (isPool) {
+                  if (isPool && args.autocommit) {
                      connection.release();
                      this._runningConnections.delete(args.tabUid);
                   }
@@ -1597,7 +1662,7 @@ export class PostgreSQLClient extends AntaresCore {
          resultsArr.push({ rows, report, fields, keys, duration });
       }
 
-      if (isPool) {
+      if (isPool && args.autocommit) {
          connection.release();
          this._runningConnections.delete(args.tabUid);
       }