Start migrating and fixing calculate scripts

inventory-server/scripts/reset-metrics.js

@@ -100,6 +100,9 @@ async function resetMetrics() {
         client = new Client(dbConfig);
         await client.connect();
 
+        // Explicitly begin a transaction
+        await client.query('BEGIN');
+
         // First verify current state
         const initialTables = await client.query(`
             SELECT tablename as name
@@ -124,6 +127,7 @@ async function resetMetrics() {
 
         for (const table of [...METRICS_TABLES].reverse()) {
             try {
+                // Use NOWAIT to avoid hanging if there's a lock
                 await client.query(`DROP TABLE IF EXISTS "${table}" CASCADE`);
                 
                 // Verify the table was actually dropped
@@ -142,13 +146,23 @@ async function resetMetrics() {
                     operation: 'Table dropped',
                     message: `Successfully dropped table: ${table}`
                 });
+                
+                // Commit after each table drop to ensure locks are released
+                await client.query('COMMIT');
+                // Start a new transaction for the next table
+                await client.query('BEGIN');
+                // Re-disable foreign key constraints for the new transaction
+                await client.query('SET session_replication_role = \'replica\'');
             } catch (err) {
                 outputProgress({
                     status: 'error',
                     operation: 'Drop table error',
                     message: `Error dropping table ${table}: ${err.message}`
                 });
-                throw err;
+                await client.query('ROLLBACK');
+                // Re-start transaction for next table
+                await client.query('BEGIN');
+                await client.query('SET session_replication_role = \'replica\'');
             }
         }
 
@@ -164,6 +178,11 @@ async function resetMetrics() {
             throw new Error(`Failed to drop all tables. Remaining tables: ${afterDrop.rows.map(t => t.name).join(', ')}`);
         }
 
+        // Make sure we have a fresh transaction here
+        await client.query('COMMIT');
+        await client.query('BEGIN');
+        await client.query('SET session_replication_role = \'replica\'');
+
         // Read metrics schema
         outputProgress({
             operation: 'Reading schema',
@@ -220,6 +239,13 @@ async function resetMetrics() {
                         rowCount: result.rowCount
                     }
                 });
+                
+                // Commit every 10 statements to avoid long-running transactions
+                if (i > 0 && i % 10 === 0) {
+                    await client.query('COMMIT');
+                    await client.query('BEGIN');
+                    await client.query('SET session_replication_role = \'replica\'');
+                }
             } catch (sqlError) {
                 outputProgress({
                     status: 'error',
@@ -230,10 +256,17 @@ async function resetMetrics() {
                         statementNumber: i + 1
                     }
                 });
+                await client.query('ROLLBACK');
                 throw sqlError;
             }
         }
 
+        // Final commit for any pending statements
+        await client.query('COMMIT');
+        
+        // Start new transaction for final checks
+        await client.query('BEGIN');
+
         // Re-enable foreign key checks after all tables are created
         await client.query('SET session_replication_role = \'origin\'');
 
@@ -269,9 +302,11 @@ async function resetMetrics() {
                 operation: 'Final table check',
                 message: `All database tables: ${finalCheck.rows.map(t => t.name).join(', ')}`
             });
+            await client.query('ROLLBACK');
             throw new Error(`Failed to create metrics tables: ${missingMetricsTables.join(', ')}`);
         }
 
+        // Commit final transaction
         await client.query('COMMIT');
         
         outputProgress({
@@ -288,7 +323,11 @@ async function resetMetrics() {
         });
         
         if (client) {
-            await client.query('ROLLBACK');
+            try {
+                await client.query('ROLLBACK');
+            } catch (rollbackError) {
+                console.error('Error during rollback:', rollbackError);
+            }
             // Make sure to re-enable foreign key checks even if there's an error
             await client.query('SET session_replication_role = \'origin\'').catch(() => {});
         }