59 lines
1.9 KiB
JavaScript
59 lines
1.9 KiB
JavaScript
|
const sqlite3 = require('sqlite3').verbose();
|
||
|
|
const path = require('path');
|
||
|
|
|
||
|
|
const dbPath = process.env.DB_PATH || path.join(__dirname, 'idc_management.db');
|
||
|
|
|
||
|
|
const db = new sqlite3.Database(dbPath, (err) => {
|
||
|
|
if (err) {
|
||
|
|
console.error('无法连接到数据库:', err);
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
console.log('已连接到数据库:', dbPath);
|
||
|
|
});
|
||
|
|
|
||
|
|
const columnsToAdd = [
|
||
|
|
{ table: 'pending_devices', column: 'height', type: 'INTEGER', defaultValue: 1 },
|
||
|
|
{ table: 'pending_devices', column: 'powerConsumption', type: 'FLOAT', defaultValue: 0 },
|
||
|
|
{ table: 'pending_devices', column: 'brand', type: 'VARCHAR(255)', defaultValue: null },
|
||
|
|
{ table: 'pending_devices', column: 'purchaseDate', type: 'DATE', defaultValue: null },
|
||
|
|
{ table: 'pending_devices', column: 'warrantyExpiry', type: 'DATE', defaultValue: null },
|
||
|
|
];
|
||
|
|
|
||
|
|
db.serialize(() => {
|
||
|
|
columnsToAdd.forEach(({ table, column, type, defaultValue }) => {
|
||
|
|
db.all(`PRAGMA table_info(${table})`, (err, rows) => {
|
||
|
|
if (err) {
|
||
|
|
console.error(`获取表 ${table} 信息失败:`, err);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
const columnExists = rows && rows.some(row => row.name === column);
|
||
|
|
|
||
|
|
if (!columnExists) {
|
||
|
|
const defaultClause = defaultValue !== null ? ` DEFAULT ${typeof defaultValue === 'string' ? `'${defaultValue}'` : defaultValue}` : '';
|
||
|
|
const sql = `ALTER TABLE ${table} ADD COLUMN ${column} ${type}${defaultClause}`;
|
||
|
|
|
||
|
|
db.run(sql, (err) => {
|
||
|
|
if (err) {
|
||
|
|
console.error(`添加列 ${table}.${column} 失败:`, err.message);
|
||
|
|
} else {
|
||
|
|
console.log(`成功添加列 ${table}.${column}`);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
} else {
|
||
|
|
console.log(`列 ${table}.${column} 已存在，跳过`);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
setTimeout(() => {
|
||
|
|
db.close((err) => {
|
||
|
|
if (err) {
|
||
|
|
console.error('关闭数据库失败:', err);
|
||
|
|
} else {
|
||
|
|
console.log('数据库迁移完成，连接已关闭');
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}, 2000);
|