Merge branch 'master' into add-product-upload-page

inventory-server/scripts/import/utils.js

@@ -1,5 +1,6 @@
 const mysql = require("mysql2/promise");
 const { Client } = require("ssh2");
+const { Pool } = require('pg');
 const dotenv = require("dotenv");
 const path = require("path");
 
@@ -41,23 +42,90 @@ async function setupSshTunnel(sshConfig) {
 async function setupConnections(sshConfig) {
   const tunnel = await setupSshTunnel(sshConfig);
   
+  // Setup MySQL connection for production
   const prodConnection = await mysql.createConnection({
     ...sshConfig.prodDbConfig,
     stream: tunnel.stream,
   });
 
-  const localConnection = await mysql.createPool({
-    ...sshConfig.localDbConfig,
-    waitForConnections: true,
-    connectionLimit: 10,
-    queueLimit: 0
-  });
+  // Setup PostgreSQL connection pool for local
+  const localPool = new Pool(sshConfig.localDbConfig);
 
-  return {
-    ssh: tunnel.ssh,
-    prodConnection,
-    localConnection
+  // Test the PostgreSQL connection
+  try {
+    const client = await localPool.connect();
+    await client.query('SELECT NOW()');
+    client.release();
+    console.log('PostgreSQL connection successful');
+  } catch (err) {
+    console.error('PostgreSQL connection error:', err);
+    throw err;
+  }
+
+  // Create a wrapper for the PostgreSQL pool to match MySQL interface
+  const localConnection = {
+    _client: null,
+    _transactionActive: false,
+
+    query: async (text, params) => {
+      // If we're not in a transaction, use the pool directly
+      if (!localConnection._transactionActive) {
+        const client = await localPool.connect();
+        try {
+          const result = await client.query(text, params);
+          return [result];
+        } finally {
+          client.release();
+        }
+      }
+      
+      // If we're in a transaction, use the dedicated client
+      if (!localConnection._client) {
+        throw new Error('No active transaction client');
+      }
+      const result = await localConnection._client.query(text, params);
+      return [result];
+    },
+
+    beginTransaction: async () => {
+      if (localConnection._transactionActive) {
+        throw new Error('Transaction already active');
+      }
+      localConnection._client = await localPool.connect();
+      await localConnection._client.query('BEGIN');
+      localConnection._transactionActive = true;
+    },
+
+    commit: async () => {
+      if (!localConnection._transactionActive) {
+        throw new Error('No active transaction to commit');
+      }
+      await localConnection._client.query('COMMIT');
+      localConnection._client.release();
+      localConnection._client = null;
+      localConnection._transactionActive = false;
+    },
+
+    rollback: async () => {
+      if (!localConnection._transactionActive) {
+        throw new Error('No active transaction to rollback');
+      }
+      await localConnection._client.query('ROLLBACK');
+      localConnection._client.release();
+      localConnection._client = null;
+      localConnection._transactionActive = false;
+    },
+
+    end: async () => {
+      if (localConnection._client) {
+        localConnection._client.release();
+        localConnection._client = null;
+      }
+      await localPool.end();
+    }
   };
+
+  return { prodConnection, localConnection, tunnel };
 }
 
 // Helper function to close connections