82 lines
3.0 KiB
JavaScript
82 lines
3.0 KiB
JavaScript
// Скрипт для очистки всех таблиц в базе данных
|
||
import { Pool } from 'pg';
|
||
import dotenv from 'dotenv';
|
||
|
||
// Загружаем переменные окружения из .env файла
|
||
dotenv.config();
|
||
|
||
const pool = new Pool({
|
||
host: process.env.DB_HOST,
|
||
port: parseInt(process.env.DB_PORT || '5432'),
|
||
database: process.env.DB_NAME,
|
||
user: process.env.DB_USERNAME,
|
||
password: process.env.DB_PASSWORD
|
||
});
|
||
|
||
async function clearDatabase() {
|
||
const client = await pool.connect();
|
||
try {
|
||
console.log('Начинаем очистку базы данных...');
|
||
|
||
// Начинаем транзакцию
|
||
await client.query('BEGIN');
|
||
|
||
// Получаем список существующих таблиц
|
||
const tablesResult = await client.query(`
|
||
SELECT table_name
|
||
FROM information_schema.tables
|
||
WHERE table_schema = 'public'
|
||
AND table_type = 'BASE TABLE'
|
||
`);
|
||
|
||
const tables = tablesResult.rows.map(row => row.table_name);
|
||
console.log('Найдены таблицы:', tables.join(', '));
|
||
|
||
// Очистка таблиц в порядке, учитывающем зависимости
|
||
if (tables.includes('messages')) {
|
||
console.log('Очистка таблицы messages...');
|
||
await client.query('TRUNCATE TABLE messages CASCADE');
|
||
}
|
||
|
||
if (tables.includes('swipes')) {
|
||
console.log('Очистка таблицы swipes...');
|
||
await client.query('TRUNCATE TABLE swipes CASCADE');
|
||
}
|
||
|
||
if (tables.includes('matches')) {
|
||
console.log('Очистка таблицы matches...');
|
||
await client.query('TRUNCATE TABLE matches CASCADE');
|
||
}
|
||
|
||
if (tables.includes('profiles')) {
|
||
console.log('Очистка таблицы profiles...');
|
||
await client.query('TRUNCATE TABLE profiles CASCADE');
|
||
}
|
||
|
||
if (tables.includes('users')) {
|
||
console.log('Очистка таблицы users...');
|
||
await client.query('TRUNCATE TABLE users CASCADE');
|
||
}
|
||
|
||
// Возвращаем внешние ключи (если они использовались)
|
||
// await client.query('SET session_replication_role = \'origin\'');
|
||
|
||
// Фиксируем транзакцию
|
||
await client.query('COMMIT');
|
||
|
||
console.log('Все таблицы успешно очищены!');
|
||
} catch (error) {
|
||
// В случае ошибки откатываем транзакцию
|
||
await client.query('ROLLBACK');
|
||
console.error('Произошла ошибка при очистке базы данных:', error);
|
||
} finally {
|
||
// Освобождаем клиента
|
||
client.release();
|
||
// Закрываем пул соединений
|
||
await pool.end();
|
||
}
|
||
}
|
||
|
||
// Запускаем функцию очистки
|
||
clearDatabase();
|