admin/yunrui_asset
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(scripts): 在迁移脚本开头加载环境变量

Browse Source
This commit is contained in:
zhang1106
2026-03-05 10:07:52 +08:00
parent 03d69456c0
commit 0917b618be
1 changed files with 199 additions and 223 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/scripts/migrate-all.js
+199 -223
View File
@@ -2,11 +2,11 @@
* IDC管理系统 - 数据库迁移汇总脚本
* 按顺序执行所有数据库迁移
* 支持幂等执行（重复执行不会出错）
* 支持 SQLite 和 MySQL
*/
const { sequelize, DB_TYPE } = require('../db');
// 迁移配置列表
const migrations = [
{
name: 'v2.0 - 网卡和端口表',
@@ -72,7 +72,6 @@ async function runMigrations() {
} catch (error) {
results.push({ name: migration.name, status: '失败', error: error.message });
console.error(` ✗ 失败: ${error.message}`);
// 继续执行下一个迁移，不中断
}
}
@@ -99,6 +98,59 @@ async function runMigrations() {
process.exit(failCount > 0 ? 1 : 0);
}
// ==================== 工具函数 ====================
async function getTableColumns(tableName) {
const dialect = sequelize.getDialect();
if (dialect === 'sqlite') {
const tableInfo = await sequelize.query(
`PRAGMA table_info(${tableName})`,
{ type: sequelize.QueryTypes.SELECT }
);
return tableInfo.map(col => col.name);
} else {
const tableInfo = await sequelize.query(
`SHOW COLUMNS FROM ${tableName}`,
{ type: sequelize.QueryTypes.SELECT }
);
return tableInfo.map(col => col.Field);
}
}
async function tableExists(tableName) {
const dialect = sequelize.getDialect();
if (dialect === 'sqlite') {
const tables = await sequelize.query(
"SELECT name FROM sqlite_master WHERE type='table' AND name=?",
{ replacements: [tableName], type: sequelize.QueryTypes.SELECT }
);
return tables.length > 0;
} else {
const tables = await sequelize.query(
"SHOW TABLES LIKE ?",
{ replacements: [tableName], type: sequelize.QueryTypes.SELECT }
);
return tables.length > 0;
}
}
async function addColumnIfNotExists(tableName, columnName, columnDef) {
const columns = await getTableColumns(tableName);
if (!columns.includes(columnName)) {
const dialect = sequelize.getDialect();
const sql = dialect === 'sqlite'
? `ALTER TABLE ${tableName} ADD COLUMN ${columnName} ${columnDef}`
: `ALTER TABLE ${tableName} ADD COLUMN ${columnName} ${columnDef}`;
await sequelize.query(sql);
console.log(` ${tableName} 表添加 ${columnName} 字段成功`);
} else {
console.log(` ${tableName} 表 ${columnName} 字段已存在，跳过`);
}
}
// ==================== 迁移函数 ====================
async function migrateV2() {
@@ -106,8 +158,7 @@ async function migrateV2() {
const dialect = sequelize.getDialect();
// 1. 创建 network_cards 表
const tables = await queryInterface.showAllTables();
if (!tables.includes('network_cards')) {
if (!(await tableExists('network_cards'))) {
await queryInterface.createTable('network_cards', {
id: {
type: sequelize.Sequelize.INTEGER,
@@ -142,261 +193,186 @@ async function migrateV2() {
}
// 2. 为 device_ports 添加 nic_id 字段
if (dialect === 'sqlite') {
const tableInfo = await sequelize.query(
"PRAGMA table_info(device_ports)",
{ type: sequelize.QueryTypes.SELECT }
);
if (!tableInfo.some(col => col.name === 'nic_id')) {
await sequelize.query("ALTER TABLE device_ports ADD COLUMN nic_id INTEGER");
}
} else {
try {
await queryInterface.addColumn('device_ports', 'nic_id', {
type: sequelize.Sequelize.INTEGER
});
} catch (err) {
if (!err.message.includes('Duplicate column')) throw err;
}
if (await tableExists('device_ports')) {
await addColumnIfNotExists('device_ports', 'nic_id', 'INTEGER');
}
}
async function migratePendingStatus() {
const queryInterface = sequelize.getQueryInterface();
const dialect = sequelize.getDialect();
if (dialect === 'sqlite') {
// 检查是否已有 status 字段
const tableInfo = await sequelize.query(
"PRAGMA table_info(users)",
{ type: sequelize.QueryTypes.SELECT }
);
if (!tableInfo.some(col => col.name === 'status')) {
// SQLite 需要重建表
await sequelize.query(`
CREATE TABLE users_new (
userId VARCHAR(255) PRIMARY KEY,
username VARCHAR(255) NOT NULL UNIQUE,
password VARCHAR(255) NOT NULL,
email VARCHAR(255),
role VARCHAR(255) DEFAULT 'user',
status VARCHAR(255) DEFAULT 'active',
createdAt DATETIME,
updatedAt DATETIME
)
`);
await sequelize.query(`
INSERT INTO users_new SELECT *, 'active' as status FROM users
`);
await sequelize.query(`DROP TABLE users`);
await sequelize.query(`ALTER TABLE users_new RENAME TO users`);
}
} else {
try {
await queryInterface.addColumn('users', 'status', {
type: sequelize.Sequelize.STRING,
defaultValue: 'active'
});
} catch (err) {
if (!err.message.includes('Duplicate column')) throw err;
}
if (await tableExists('users')) {
await addColumnIfNotExists('users', 'status', "VARCHAR(255) DEFAULT 'active'");
}
}
async function migrateConsumableVersion() {
const tableInfo = await sequelize.query(
"PRAGMA table_info(consumables)",
{ type: sequelize.QueryTypes.SELECT }
);
if (!tableInfo.some(col => col.name === 'version')) {
await sequelize.query(
"ALTER TABLE consumables ADD COLUMN version INTEGER DEFAULT 0"
);
if (await tableExists('consumables')) {
await addColumnIfNotExists('consumables', 'version', 'INTEGER DEFAULT 0');
}
}
async function migrateConsumableLogs() {
const tableInfo = await sequelize.query(
"PRAGMA table_info(consumable_logs)",
{ type: sequelize.QueryTypes.SELECT }
);
const columns = tableInfo.map(col => col.name);
if (!columns.includes('isEditable')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN isEditable BOOLEAN DEFAULT 1
`);
}
if (!columns.includes('originalLogId')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN originalLogId INTEGER
`);
}
if (!columns.includes('modifiedBy')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN modifiedBy VARCHAR(255)
`);
}
if (!columns.includes('modifiedAt')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN modifiedAt DATETIME
`);
}
if (!columns.includes('modificationReason')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN modificationReason TEXT
`);
if (await tableExists('consumable_logs')) {
await addColumnIfNotExists('consumable_logs', 'isEditable', 'BOOLEAN DEFAULT 1');
await addColumnIfNotExists('consumable_logs', 'originalLogId', 'INTEGER');
await addColumnIfNotExists('consumable_logs', 'modifiedBy', 'VARCHAR(255)');
await addColumnIfNotExists('consumable_logs', 'modifiedAt', 'DATETIME');
await addColumnIfNotExists('consumable_logs', 'modificationReason', 'TEXT');
}
}
async function migrateConsumableLogDecouple() {
const tableInfo = await sequelize.query(
"PRAGMA table_info(consumable_logs)",
{ type: sequelize.QueryTypes.SELECT }
);
const columns = tableInfo.map(col => col.name);
if (!columns.includes('isConsumableDeleted')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN isConsumableDeleted BOOLEAN DEFAULT 0
`);
}
if (!columns.includes('consumableSnapshot')) {
await sequelize.query(`
ALTER TABLE consumable_logs ADD COLUMN consumableSnapshot TEXT
`);
if (await tableExists('consumable_logs')) {
await addColumnIfNotExists('consumable_logs', 'isConsumableDeleted', 'BOOLEAN DEFAULT 0');
await addColumnIfNotExists('consumable_logs', 'consumableSnapshot', 'TEXT');
}
}
async function removeConsumableLogFK() {
// SQLite 不支持直接删除外键，需要重建表
const fks = await sequelize.query(
`PRAGMA foreign_key_list(consumable_logs);`,
{ type: sequelize.QueryTypes.SELECT }
);
const dialect = sequelize.getDialect();
if (dialect === 'sqlite') {
const fks = await sequelize.query(
`PRAGMA foreign_key_list(consumable_logs);`,
{ type: sequelize.QueryTypes.SELECT }
);
if (!fks || fks.length === 0) {
return; // 没有外键约束
if (!fks || fks.length === 0) {
return;
}
await sequelize.query(`
CREATE TABLE consumable_logs_new (
id INTEGER PRIMARY KEY AUTOINCREMENT,
consumableId VARCHAR(255),
consumableName VARCHAR(255),
operationType VARCHAR(50),
quantity INTEGER,
previousStock INTEGER,
currentStock INTEGER,
operator VARCHAR(255),
reason TEXT,
notes TEXT,
isEditable BOOLEAN DEFAULT 1,
originalLogId INTEGER,
modifiedBy VARCHAR(255),
modifiedAt DATETIME,
modificationReason TEXT,
isConsumableDeleted BOOLEAN DEFAULT 0,
consumableSnapshot TEXT,
relatedId VARCHAR(255),
createdAt DATETIME,
updatedAt DATETIME
)
`);
await sequelize.query(`
INSERT INTO consumable_logs_new
SELECT id, consumableId, consumableName, operationType, quantity,
previousStock, currentStock, operator, reason, notes,
isEditable, originalLogId, modifiedBy, modifiedAt, modificationReason,
isConsumableDeleted, consumableSnapshot, relatedId, createdAt, updatedAt
FROM consumable_logs
`);
await sequelize.query(`DROP TABLE consumable_logs`);
await sequelize.query(`ALTER TABLE consumable_logs_new RENAME TO consumable_logs`);
await sequelize.query(`CREATE INDEX idx_logs_consumable_id ON consumable_logs(consumableId)`);
await sequelize.query(`CREATE INDEX idx_logs_operation_type ON consumable_logs(operationType)`);
await sequelize.query(`CREATE INDEX idx_logs_created_at ON consumable_logs(createdAt)`);
await sequelize.query(`CREATE INDEX idx_logs_is_consumable_deleted ON consumable_logs(isConsumableDeleted)`);
}
// 重建表（去掉外键约束）
await sequelize.query(`
CREATE TABLE consumable_logs_new (
id INTEGER PRIMARY KEY AUTOINCREMENT,
consumableId VARCHAR(255),
consumableName VARCHAR(255),
operationType VARCHAR(50),
quantity INTEGER,
previousStock INTEGER,
currentStock INTEGER,
operator VARCHAR(255),
reason TEXT,
notes TEXT,
isEditable BOOLEAN DEFAULT 1,
originalLogId INTEGER,
modifiedBy VARCHAR(255),
modifiedAt DATETIME,
modificationReason TEXT,
isConsumableDeleted BOOLEAN DEFAULT 0,
consumableSnapshot TEXT,
relatedId VARCHAR(255),
createdAt DATETIME,
updatedAt DATETIME
)
`);
// 复制数据
await sequelize.query(`
INSERT INTO consumable_logs_new
SELECT id, consumableId, consumableName, operationType, quantity,
previousStock, currentStock, operator, reason, notes,
isEditable, originalLogId, modifiedBy, modifiedAt, modificationReason,
isConsumableDeleted, consumableSnapshot, relatedId, createdAt, updatedAt
FROM consumable_logs
`);
// 删除旧表，重命名新表
await sequelize.query(`DROP TABLE consumable_logs`);
await sequelize.query(`ALTER TABLE consumable_logs_new RENAME TO consumable_logs`);
// 创建索引
await sequelize.query(`CREATE INDEX idx_logs_consumable_id ON consumable_logs(consumableId)`);
await sequelize.query(`CREATE INDEX idx_logs_operation_type ON consumable_logs(operationType)`);
await sequelize.query(`CREATE INDEX idx_logs_created_at ON consumable_logs(createdAt)`);
await sequelize.query(`CREATE INDEX idx_logs_is_consumable_deleted ON consumable_logs(isConsumableDeleted)`);
}
async function migrateConsumableLogArchive() {
const tables = await sequelize.getQueryInterface().showAllTables();
if (tables.includes('consumable_log_archives')) {
return; // 表已存在
if (await tableExists('consumable_log_archives')) {
return;
}
await sequelize.query(`
CREATE TABLE consumable_log_archives (
id INTEGER PRIMARY KEY AUTOINCREMENT,
archiveId VARCHAR(255) NOT NULL UNIQUE,
consumableId VARCHAR(255) NOT NULL,
consumableName VARCHAR(255) NOT NULL,
consumableSnapshot TEXT,
totalOperations INTEGER DEFAULT 0,
firstOperationAt DATETIME,
lastOperationAt DATETIME,
totalInQuantity INTEGER DEFAULT 0,
totalOutQuantity INTEGER DEFAULT 0,
finalStock INTEGER DEFAULT 0,
deletedBy VARCHAR(255),
deletedAt DATETIME,
deleteReason VARCHAR(255),
createdAt DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
updatedAt DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
)
`);
const queryInterface = sequelize.getQueryInterface();
await queryInterface.createTable('consumable_log_archives', {
id: {
type: sequelize.Sequelize.INTEGER,
primaryKey: true,
autoIncrement: true
},
archiveId: {
type: sequelize.Sequelize.STRING,
allowNull: false,
unique: true
},
consumableId: {
type: sequelize.Sequelize.STRING,
allowNull: false
},
consumableName: {
type: sequelize.Sequelize.STRING,
allowNull: false
},
consumableSnapshot: {
type: sequelize.Sequelize.TEXT
},
totalOperations: {
type: sequelize.Sequelize.INTEGER,
defaultValue: 0
},
firstOperationAt: {
type: sequelize.Sequelize.DATE
},
lastOperationAt: {
type: sequelize.Sequelize.DATE
},
totalInQuantity: {
type: sequelize.Sequelize.INTEGER,
defaultValue: 0
},
totalOutQuantity: {
type: sequelize.Sequelize.INTEGER,
defaultValue: 0
},
finalStock: {
type: sequelize.Sequelize.INTEGER,
defaultValue: 0
},
deletedBy: {
type: sequelize.Sequelize.STRING
},
deletedAt: {
type: sequelize.Sequelize.DATE
},
deleteReason: {
type: sequelize.Sequelize.STRING
},
createdAt: {
type: sequelize.Sequelize.DATE,
allowNull: false,
defaultValue: sequelize.Sequelize.literal('CURRENT_TIMESTAMP')
},
updatedAt: {
type: sequelize.Sequelize.DATE,
allowNull: false,
defaultValue: sequelize.Sequelize.literal('CURRENT_TIMESTAMP')
}
});
// 创建索引
await sequelize.query(`CREATE INDEX idx_archive_consumable_id ON consumable_log_archives(consumableId)`);
await sequelize.query(`CREATE INDEX idx_archive_archive_id ON consumable_log_archives(archiveId)`);
await sequelize.query(`CREATE INDEX idx_archive_deleted_at ON consumable_log_archives(deletedAt)`);
const dialect = sequelize.getDialect();
if (dialect === 'sqlite') {
await sequelize.query(`CREATE INDEX idx_archive_consumable_id ON consumable_log_archives(consumableId)`);
await sequelize.query(`CREATE INDEX idx_archive_archive_id ON consumable_log_archives(archiveId)`);
await sequelize.query(`CREATE INDEX idx_archive_deleted_at ON consumable_log_archives(deletedAt)`);
}
}
async function migrateSnList() {
const dialect = sequelize.getDialect();
const tables = ['consumables', 'consumable_records', 'consumable_logs'];
for (const table of tables) {
const tableInfo = await sequelize.query(
dialect === 'sqlite'
? `PRAGMA table_info(${table})`
: `SHOW COLUMNS FROM ${table}`,
{ type: sequelize.QueryTypes.SELECT }
);
const columns = dialect === 'sqlite'
? tableInfo.map(col => col.name)
: tableInfo.map(col => col.Field);
if (!columns.includes('snList')) {
if (dialect === 'sqlite') {
await sequelize.query(`ALTER TABLE ${table} ADD COLUMN snList TEXT DEFAULT '[]'`);
} else {
await sequelize.query(`ALTER TABLE ${table} ADD COLUMN snList JSON DEFAULT '[]'`);
}
console.log(` ${table} 表添加 snList 字段成功`);
if (await tableExists(table)) {
const columnDef = dialect === 'sqlite' ? "TEXT DEFAULT '[]'" : "JSON DEFAULT '[]'";
await addColumnIfNotExists(table, 'snList', columnDef);
} else {
console.log(` ${table} 表 snList 字段已存在，跳过`);
console.log(` ${table} 表不存在，跳过`);
}
}
}