@pi-r/postgres 0.6.3 → 0.6.5

package/LICENSE

@@ -1,4 +1,4 @@
-Copyright 2023 An Pham
+Copyright 2024 An Pham
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 

package/README.md

@@ -1,7 +1,10 @@
-### @pi-r/postgres
+# @pi-r/postgres
 
-https://e-mc.readthedocs.io/en/latest/db/postgres.html
+## Documentation
 
-### LICENSE
+- [E-mc](https://e-mc.readthedocs.io/en/latest/db/postgres.html)
+- [squared](https://squared.readthedocs.io)
+
+## LICENSE
 
 MIT

package/client/index.d.ts

@@ -1,8 +1,9 @@
 import type { IDb } from '@e-mc/types/lib';
-import type { PostgresCredential, PostgresDataSource } from '@e-mc/types/lib/db';
 
 import type { IDbSourceClient } from '@e-mc/db/types';
 
+import type { PostgresCredential, PostgresDataSource } from '../types';
+
 declare const Postgres: IDbSourceClient<PostgresDataSource> & {
     setAuthentication(this: IDb, credential: PostgresCredential): void;
 };

package/client/index.js

@@ -24,8 +24,9 @@ class PostgresPool extends DbPool {
 }
 const POOL_STATE = {};
 const POOL_UNUSED = ['ssl', 'keepAlive', 'statement_timeout', 'query_timeout', 'keepAliveInitialDelayMillis', 'idle_in_transaction_session_timeout', 'connectionTimeoutMillis', 'max', 'min', 'idleTimeoutMillis', 'log', 'allowExitOnIdle', 'maxUses'];
+const POOL_ACTIVE = new WeakSet();
 function removePoolProperties(credential) {
-    if (!credential.uuidKey) {
+    if (!credential.uuidKey && POOL_ACTIVE.has(credential)) {
         let result;
         for (let i = 0, length = POOL_UNUSED.length; i < length; ++i) {
             const attr = POOL_UNUSED[i];
@@ -103,7 +104,7 @@ function setCredential(item) {
             errors.push('user');
         }
         if (errors.length) {
-            throw (0, types_1.errorMessage)("postgres" /* STRINGS.MODULE_NAME */, 'Not defined - ' + errors.join(' | '));
+            throw (0, types_1.errorMessage)("postgres", 'Not defined - ' + errors.join(' | '));
         }
     }
     const usePool = item.usePool;
@@ -127,7 +128,7 @@ function setCredential(item) {
             return;
         }
     }
-    const config = this.getPoolConfig("postgres" /* STRINGS.MODULE_NAME */, password);
+    const config = this.getPoolConfig("postgres", password);
     if (config) {
         const { min, max, idle, timeout } = config;
         if (min >= 0) {
@@ -180,17 +181,18 @@ async function executeBatchQuery(batch, options = '', outResult) {
     if (!parallel) {
         outResult || (outResult = new Array(length));
     }
-    const caching = this.hasCache("postgres" /* STRINGS.MODULE_NAME */, sessionKey);
+    const caching = this.hasCache("postgres", sessionKey);
     const tasks = new Array(length);
     const clients = [];
     const pools = [];
     let postgresPool, postgresClient, postgresCredential, onceCredential = connectOnce ? batch[0].credential : undefined;
     const getConnection = async (item, credential) => {
-        item.transactionState = 64 /* DB_TRANSACTION.AUTH */;
+        item.transactionState = 64;
         let client;
         if (postgresPool) {
             pools.push(client = await postgresPool.getConnection());
-            item.transactionState &= ~64 /* DB_TRANSACTION.AUTH */;
+            item.transactionState &= ~64;
+            POOL_ACTIVE.add(credential);
             return client;
         }
         const pool = item.usePool && DbPool.findKey(POOL_STATE, item.usePool, getPoolKey(credential), ...connectOnce ? [item, batch[0]] : [item]);
@@ -201,6 +203,7 @@ async function executeBatchQuery(batch, options = '', outResult) {
                     postgresPool = pool;
                 }
                 pool.connected = true;
+                POOL_ACTIVE.add(credential);
             }
             catch (err) {
                 let close;
@@ -232,7 +235,7 @@ async function executeBatchQuery(batch, options = '', outResult) {
             }
             postgresCredential = credential;
         }
-        item.transactionState &= ~64 /* DB_TRANSACTION.AUTH */;
+        item.transactionState &= ~64;
         return client;
     };
     if (this.host?.username) {
@@ -242,7 +245,7 @@ async function executeBatchQuery(batch, options = '', outResult) {
         const item = batch[i];
         const { source, query, params, ignoreCache } = item;
         let credential = postgresCredential || onceCredential, error;
-        if (!credential && !(0, types_1.isPlainObject)(credential = item.credential) && (error = (0, types_1.errorMessage)(source, "Invalid credentials" /* ERR_DB.CREDENTIALS */)) || !query && (error = (0, types_1.errorMessage)(source, "Missing database query" /* ERR_DB.QUERY */))) {
+        if (!credential && !(0, types_1.isPlainObject)(credential = item.credential) && (error = (0, types_1.errorMessage)(source, "Invalid credentials")) || !query && (error = (0, types_1.errorMessage)(source, "Missing database query"))) {
             if (this.handleFail(error, item, { errorQuery })) {
                 if (!parallel) {
                     tasks.length = 0;
@@ -258,8 +261,8 @@ async function executeBatchQuery(batch, options = '', outResult) {
             }
             continue;
         }
-        item.transactionState = 1 /* DB_TRANSACTION.ACTIVE */;
-        const streamRow = typeof item.streamRow === 'string' ? this.hasCoerce("postgres" /* STRINGS.MODULE_NAME */, 'options', null, credential) && (0, types_1.asFunction)(item.streamRow) : item.streamRow;
+        item.transactionState = 1;
+        const streamRow = typeof item.streamRow === 'string' ? this.hasCoerce("postgres", 'options', null, credential) && (0, types_1.asFunction)(item.streamRow) : item.streamRow;
         const cacheValue = ignoreCache === undefined ? sessionKey : Array.isArray(ignoreCache) ? { sessionKey, exclusiveOf: ignoreCache } : { sessionKey, renewCache: ignoreCache === 0 };
         let queryString = '';
         if (caching && ignoreCache !== true) {
@@ -273,7 +276,7 @@ async function executeBatchQuery(batch, options = '', outResult) {
                     else {
                         outResult[i] = result;
                     }
-                    this.add(item, 4 /* DB_TRANSACTION.COMMIT */ | 128 /* DB_TRANSACTION.CACHE */);
+                    this.add(item, 4 | 128);
                     continue;
                 }
                 if (!ignoreCache && outCacheMiss) {
@@ -323,7 +326,7 @@ async function executeBatchQuery(batch, options = '', outResult) {
                             reject(error);
                         }
                         else {
-                            this.add(item, 4 /* DB_TRANSACTION.COMMIT */);
+                            this.add(item, 4);
                             resolve(!error ? this.setQueryResult(source, removePoolProperties(credential), queryString, rows, cacheValue) : rows);
                         }
                     })
@@ -331,7 +334,7 @@ async function executeBatchQuery(batch, options = '', outResult) {
                 }
                 else {
                     const { rows, command } = await client.query(query, params);
-                    this.add(item, 4 /* DB_TRANSACTION.COMMIT */);
+                    this.add(item, 4);
                     resolve(rows.length || command === 'SELECT' ? this.setQueryResult(source, removePoolProperties(credential), queryString, rows, cacheValue) : null);
                 }
             }
@@ -356,7 +359,7 @@ 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 */, "postgres" /* STRINGS.MODULE_NAME */], err, { type: 65536 /* LOG_TYPE.DB */, fatal: false })));
+            clients.forEach(item => item.end(err => err && this.writeFail(["Unable to close connnection", "postgres"], err, { type: 65536, fatal: false })));
             pools.forEach(item => item.release());
         },
         parallel

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-r/postgres",
-  "version": "0.6.3",
+  "version": "0.6.5",
   "description": "PostgreSQL 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.8.3",
-    "@e-mc/types": "^0.8.3",
+    "@e-mc/db": "^0.8.6",
+    "@e-mc/types": "^0.8.6",
     "pg": "^8.11.3",
     "pg-connection-string": "^2.6.2",
     "pg-query-stream": "^4.5.3"

package/types/index.d.ts

@@ -2,6 +2,7 @@ import type { DbDataSource } from '@e-mc/types/lib/squared';
 
 import type { ExecuteAction, ServerAuth } from '@e-mc/types/lib/db';
 
+// @ts-ignore
 import type { PoolConfig, QueryArrayConfig } from 'pg';
 
 export interface PostgresDataSource<T = unknown[]> extends DbDataSource<string | QueryArrayConfig, unknown, unknown, string | PostgresCredential, string>, ExecuteAction<T> {