63 lines
2.4 KiB
JavaScript
63 lines
2.4 KiB
JavaScript
const pool = require("./db/pool.js");
|
|
|
|
async function checkAndAddColumn(tableName, columnName, columnDefinition) {
|
|
const client = await pool.connect();
|
|
try {
|
|
// 컬럼 존재 여부 확인
|
|
const checkQuery = `
|
|
SELECT COUNT(*)
|
|
FROM information_schema.columns
|
|
WHERE table_schema = 'ver4'
|
|
AND table_name = $1
|
|
AND column_name = $2;
|
|
`;
|
|
const res = await client.query(checkQuery, [tableName, columnName]);
|
|
const exists = parseInt(res.rows[0].count) > 0;
|
|
|
|
if (!exists) {
|
|
console.log(`Adding column [${columnName}] to [ver4.${tableName}]...`);
|
|
const alterQuery = `ALTER TABLE ver4.${tableName} ADD COLUMN ${columnName} ${columnDefinition};`;
|
|
await client.query(alterQuery);
|
|
console.log(`Successfully added [${columnName}] to [ver4.${tableName}].`);
|
|
} else {
|
|
console.log(`Column [${columnName}] already exists in [ver4.${tableName}].`);
|
|
}
|
|
} catch (err) {
|
|
console.error(`Error processing table ${tableName}, column ${columnName}:`, err);
|
|
} finally {
|
|
client.release();
|
|
}
|
|
}
|
|
|
|
async function run() {
|
|
try {
|
|
console.log("Starting DB patch processing...");
|
|
|
|
// 1. tb_data
|
|
await checkAndAddColumn("tb_data", "popup_size", "BIGINT DEFAULT 0");
|
|
await checkAndAddColumn("tb_data", "preview_size", "BIGINT DEFAULT 0");
|
|
|
|
// 2. _test_tb_data
|
|
await checkAndAddColumn("_test_tb_data", "popup_size", "BIGINT DEFAULT 0");
|
|
await checkAndAddColumn("_test_tb_data", "preview_size", "BIGINT DEFAULT 0");
|
|
|
|
// 3. tb_official_doc_file
|
|
await checkAndAddColumn("tb_official_doc_file", "popup_size", "BIGINT DEFAULT 0");
|
|
await checkAndAddColumn("tb_official_doc_file", "preview_size", "BIGINT DEFAULT 0");
|
|
|
|
// 4. tb_download_folder
|
|
await checkAndAddColumn("tb_download_folder", "expire_date", "TIMESTAMP");
|
|
await checkAndAddColumn("tb_download_folder", "made", "BOOLEAN DEFAULT FALSE");
|
|
await checkAndAddColumn("tb_download_folder", "path", "TEXT");
|
|
await checkAndAddColumn("tb_download_folder", "name", "TEXT");
|
|
|
|
console.log("DB patch processing finished.");
|
|
} catch (err) {
|
|
console.error("Migration script failed:", err);
|
|
} finally {
|
|
await pool.end();
|
|
}
|
|
}
|
|
|
|
run();
|