@pi-r/mysql 0.3.0 → 0.6.0

package/client/index.js

@@ -8,10 +8,10 @@ const types_1 = require("@e-mc/types");
 const Db = require("@e-mc/db");
 const DbPool = require('@e-mc/db/pool');
 class MySQLPool extends DbPool {
-    getConnection() {
+    async getConnection() {
         return this.client.getConnection();
     }
-    close() {
+    async close() {
         return this.client.end();
     }
     isEmpty() {
@@ -95,36 +95,39 @@ function setCredential(item) {
         }
     }
     const usePool = item.usePool;
-    if (usePool) {
-        let username, password, pool;
-        if (typeof usePool === 'string' && (username = credential.user || credential.uri && (0, util_1.parseConnectionString)(credential.uri)?.username)) {
-            [password, pool] = DbPool.validateKey(POOL_STATE, username, usePool);
-            if (pool) {
-                pool.add(item, password);
-                return;
-            }
+    if (!usePool) {
+        return;
+    }
+    let username, password, pool;
+    if (typeof usePool === 'string' && (username = credential.user || credential.uri && (0, util_1.parseConnectionString)(credential.uri)?.username)) {
+        [password, pool] = DbPool.validateKey(POOL_STATE, username, usePool);
+        if (pool) {
+            pool.add(item, password);
+            return;
         }
-        const poolKey = getPoolKey(credential);
-        if (!(pool = POOL_STATE[poolKey]) || pool.closed) {
-            const config = this.getPoolConfig("mysql" /* STRINGS.MODULE_NAME */, password);
-            if (config) {
-                const { max, idle, queue_max } = config;
-                if (max > 0) {
-                    credential.connectionLimit ?? (credential.connectionLimit = max);
-                }
-                if (idle >= 0) {
-                    credential.idleTimeout ?? (credential.idleTimeout = idle);
-                }
-                if (queue_max >= 0) {
-                    credential.queueLimit ?? (credential.queueLimit = queue_max);
-                }
-            }
-            new MySQLPool(mysql.createPool(removeUUIDKey(credential)), poolKey, username && password ? { username, password } : undefined).add(item).parent = POOL_STATE;
+    }
+    const poolKey = getPoolKey(credential);
+    if ((pool = POOL_STATE[poolKey]) && !pool.closed) {
+        pool.add(item);
+        return;
+    }
+    const config = this.getPoolConfig("mysql" /* STRINGS.MODULE_NAME */, password);
+    if (config) {
+        const { max, idle, queue_max, timeout } = config;
+        if (max > 0) {
+            credential.connectionLimit ?? (credential.connectionLimit = max);
         }
-        else {
-            pool.add(item);
+        if (idle >= 0) {
+            credential.idleTimeout ?? (credential.idleTimeout = idle);
+        }
+        if (queue_max >= 0) {
+            credential.queueLimit ?? (credential.queueLimit = queue_max);
+        }
+        if (timeout > 0) {
+            credential.connectTimeout ?? (credential.connectTimeout = timeout);
         }
     }
+    new MySQLPool(mysql.createPool(removeUUIDKey(credential)), poolKey, username && password ? { username, password } : undefined).add(item).parent = POOL_STATE;
 }
 exports.setCredential = setCredential;
 async function executeQuery(item, options) {
@@ -212,6 +215,9 @@ async function executeBatchQuery(batch, options = '', outResult) {
         item.transactionState &= ~64 /* DB_TRANSACTION.AUTH */;
         return client;
     };
+    if (this.host?.username) {
+        this.applyCommand(...batch);
+    }
     for (let i = 0, mysql2Client, mysql2Credential; i < length; ++i) {
         const item = batch[i];
         const { source, query, params, ignoreCache } = item;
@@ -359,14 +365,14 @@ async function executeBatchQuery(batch, options = '', outResult) {
     }
     return this.processRows(batch, tasks, {
         disconnect: () => {
-            clients.forEach(item => item.end(err => err && this.writeFail(["Unable to close connnection" /* ERR_DB.CONNECTION_CLOSE */, "mysql" /* STRINGS.MODULE_NAME */], err, { type: 65536 /* LOG_TYPE.DB */, fatal: false })));
+            clients.forEach(async (item) => item.end(err => err && this.writeFail(["Unable to close connnection" /* ERR_DB.CONNECTION_CLOSE */, "mysql" /* STRINGS.MODULE_NAME */], err, { type: 65536 /* LOG_TYPE.DB */, fatal: false })));
             pools.forEach(item => item.release());
         },
         parallel
     }, outResult);
 }
 exports.executeBatchQuery = executeBatchQuery;
-function checkTimeout(value, limit = 0) {
+async function checkTimeout(value, limit = 0) {
     return DbPool.checkTimeout(POOL_STATE, value, limit);
 }
 exports.checkTimeout = checkTimeout;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-r/mysql",
-  "version": "0.3.0",
+  "version": "0.6.0",
   "description": "MySQL client driver for E-mc.",
   "main": "client/index.js",
   "types": "client/index.d.ts",
@@ -21,8 +21,8 @@
   "license": "MIT",
   "homepage": "https://github.com/anpham6/pi-r#readme",
   "dependencies": {
-    "@e-mc/db": "^0.6.0",
-    "@e-mc/types": "^0.6.0",
-    "mysql2": "^3.6.0"
+    "@e-mc/db": "^0.8.0",
+    "@e-mc/types": "^0.8.0",
+    "mysql2": "^3.6.5"
   }
 }