58 lines
1.5 KiB
TypeScript
58 lines
1.5 KiB
TypeScript
import { QueryTypes } from "@sequelize/core"
|
|
|
|
import { isNil } from "lodash"
|
|
|
|
import db from "@/db/db-client"
|
|
|
|
async function getTableNames() {
|
|
const query = /* sql */ `
|
|
SELECT
|
|
table_name as "tableName"
|
|
FROM
|
|
information_schema.tables
|
|
WHERE
|
|
table_schema = 'public'
|
|
AND table_type = 'BASE TABLE'
|
|
AND table_name != 'SequelizeMeta'
|
|
AND table_name != 'knex_migrations'
|
|
AND table_name != 'knex_migrations_lock';
|
|
`
|
|
|
|
try {
|
|
const result = await db.query<{ tableName: string }>(query, { type: QueryTypes.SELECT })
|
|
const tableNames = result.map((row) => row.tableName)
|
|
return tableNames
|
|
} catch (error) {
|
|
console.error("Error fetching table names:", error)
|
|
throw error
|
|
}
|
|
}
|
|
|
|
async function buildCleanDatabaseQuery() {
|
|
const tableNames = await getTableNames()
|
|
const quotedTableNames = tableNames.map((name) => `"${name}"`)
|
|
return /* sql */ `
|
|
TRUNCATE TABLE ${quotedTableNames.join(",\n ")} RESTART IDENTITY CASCADE;
|
|
`
|
|
}
|
|
|
|
let cleanDatabaseQuery: string | null = null
|
|
|
|
export async function cleanDatabase() {
|
|
if (isNil(cleanDatabaseQuery)) {
|
|
cleanDatabaseQuery = await buildCleanDatabaseQuery()
|
|
}
|
|
|
|
try {
|
|
// TODO: once all tables are in Sequelize models, use this instead:
|
|
// await db.truncate({ cascade: true, restartIdentity: true })
|
|
await db.query(cleanDatabaseQuery, { raw: true })
|
|
return true
|
|
} catch (error) {
|
|
console.error(error)
|
|
return false
|
|
}
|
|
}
|
|
|
|
export default cleanDatabase
|