mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2024-12-24 20:24:05 -08:00
Code Issues Actions 1 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
Show stats 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);
} }
/** /**