admin/n8n-enterprise-unlocked
1
0
Fork 0
mirror of https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git synced 2025-12-17 18:12:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

🧪 Make PG truncation sequential

Browse Source
This commit is contained in:
Iván Ovejero
2022-06-30 15:53:13 +02:00
parent 4ae8b54dc2
commit a2c599d98d
1 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
packages/cli/test/integration/shared/testDb.ts
View File

@@ -2,7 +2,7 @@ import { exec as callbackExec } from 'child_process';
import { promisify } from 'util'; import { promisify } from 'util';
import { createConnection, getConnection, ConnectionOptions, Connection } from 'typeorm'; import { createConnection, getConnection, ConnectionOptions, Connection } from 'typeorm';
import { Credentials, UserSettings } from 'n8n-core'; import { UserSettings } from 'n8n-core';
import config from '../../../config'; import config from '../../../config';
import { import {
@@ -163,13 +163,15 @@ async function truncateMappingTables(
} }
if (dbType === 'postgresdb') { if (dbType === 'postgresdb') {
const promises = mappingTables.map((tableName) => { const schema = config.getEnv('database.postgresdb.schema');
const schema = config.getEnv('database.postgresdb.schema');
const fullTableName = `${schema}.${tableName}`;
testDb.query(`TRUNCATE TABLE ${fullTableName} RESTART IDENTITY CASCADE;`);
});
return Promise.all(promises); // `TRUNCATE` in postgres cannot be parallelized
for (const tableName of mappingTables) {
const fullTableName = `${schema}.${tableName}`;
await testDb.query(`TRUNCATE TABLE ${fullTableName} RESTART IDENTITY CASCADE;`);
}
return Promise.resolve([]);
} }
// mysqldb, mariadb // mysqldb, mariadb
@@ -210,16 +212,16 @@ export async function truncate(collections: Array<CollectionName>, testDbName: s
} }
if (dbType === 'postgresdb') { if (dbType === 'postgresdb') {
const truncationPromises = collections.map((collection) => { const schema = config.getEnv('database.postgresdb.schema');
const schema = config.getEnv('database.postgresdb.schema');
// `TRUNCATE` in postgres cannot be parallelized
for (const collection of collections) {
const fullTableName = `${schema}.${toTableName(collection)}`; const fullTableName = `${schema}.${toTableName(collection)}`;
await testDb.query(`TRUNCATE TABLE ${fullTableName} RESTART IDENTITY CASCADE;`);
}
return testDb.query(`TRUNCATE TABLE ${fullTableName} RESTART IDENTITY CASCADE;`); return await truncateMappingTables(dbType, collections, testDb);
}); // return Promise.resolve([])
truncationPromises.push(truncateMappingTables(dbType, collections, testDb));
return Promise.all(truncationPromises);
} }
/** /**