87 lines
3.4 KiB
JavaScript
87 lines
3.4 KiB
JavaScript
const mysql = require('mysql2/promise');
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
require('dotenv').config();
|
|
|
|
async function runMigration() {
|
|
let connection;
|
|
|
|
try {
|
|
// 创建数据库连接
|
|
connection = await mysql.createConnection({
|
|
host: process.env.DB_HOST,
|
|
user: process.env.DB_USER,
|
|
password: process.env.DB_PASSWORD,
|
|
database: process.env.DB_NAME,
|
|
charset: 'utf8mb4'
|
|
});
|
|
|
|
console.log('数据库连接成功');
|
|
|
|
// 直接定义SQL语句
|
|
const sqlStatements = [
|
|
`CREATE TABLE IF NOT EXISTS \`payment_orders\` (
|
|
\`id\` int(11) NOT NULL AUTO_INCREMENT COMMENT '订单ID',
|
|
\`user_id\` int(11) NOT NULL COMMENT '用户ID',
|
|
\`out_trade_no\` varchar(64) NOT NULL COMMENT '商户订单号',
|
|
\`transaction_id\` varchar(64) DEFAULT NULL COMMENT '微信支付订单号',
|
|
\`total_fee\` int(11) NOT NULL COMMENT '订单金额（分）',
|
|
\`body\` varchar(128) NOT NULL COMMENT '商品描述',
|
|
\`trade_type\` varchar(16) NOT NULL COMMENT '交易类型',
|
|
\`prepay_id\` varchar(64) DEFAULT NULL COMMENT '预支付交易会话标识',
|
|
\`mweb_url\` text DEFAULT NULL COMMENT 'H5支付跳转链接',
|
|
\`status\` enum('pending','paid','failed','cancelled') NOT NULL DEFAULT 'pending' COMMENT '支付状态',
|
|
\`paid_at\` datetime DEFAULT NULL COMMENT '支付完成时间',
|
|
\`created_at\` datetime NOT NULL COMMENT '创建时间',
|
|
\`updated_at\` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
|
|
PRIMARY KEY (\`id\`),
|
|
UNIQUE KEY \`uk_out_trade_no\` (\`out_trade_no\`),
|
|
KEY \`idx_user_id\` (\`user_id\`),
|
|
KEY \`idx_status\` (\`status\`),
|
|
KEY \`idx_created_at\` (\`created_at\`)
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='支付订单表'`,
|
|
|
|
`ALTER TABLE \`users\` ADD COLUMN \`payment_status\` enum('unpaid','paid') NOT NULL DEFAULT 'unpaid' COMMENT '支付状态'`,
|
|
|
|
`ALTER TABLE \`users\` ADD KEY \`idx_payment_status\` (\`payment_status\`)`
|
|
];
|
|
|
|
console.log(`准备执行 ${sqlStatements.length} 条SQL语句`);
|
|
|
|
// 执行每条SQL语句
|
|
for (let i = 0; i < sqlStatements.length; i++) {
|
|
const statement = sqlStatements[i];
|
|
console.log(`执行第 ${i + 1} 条语句...`);
|
|
|
|
try {
|
|
await connection.execute(statement);
|
|
console.log(`✓ 第 ${i + 1} 条语句执行成功`);
|
|
} catch (error) {
|
|
if (error.code === 'ER_DUP_FIELDNAME') {
|
|
console.log(`⚠ 第 ${i + 1} 条语句跳过（字段已存在）: ${error.message}`);
|
|
} else if (error.code === 'ER_TABLE_EXISTS_ERROR') {
|
|
console.log(`⚠ 第 ${i + 1} 条语句跳过（表已存在）: ${error.message}`);
|
|
} else if (error.code === 'ER_DUP_KEYNAME') {
|
|
console.log(`⚠ 第 ${i + 1} 条语句跳过（索引已存在）: ${error.message}`);
|
|
} else {
|
|
console.error(`✗ 第 ${i + 1} 条语句执行失败:`, error.message);
|
|
throw error;
|
|
}
|
|
}
|
|
}
|
|
|
|
console.log('\n✅ 数据库迁移完成！');
|
|
|
|
} catch (error) {
|
|
console.error('❌ 数据库迁移失败:', error.message);
|
|
process.exit(1);
|
|
} finally {
|
|
if (connection) {
|
|
await connection.end();
|
|
console.log('数据库连接已关闭');
|
|
}
|
|
}
|
|
}
|
|
|
|
// 运行迁移
|
|
runMigration(); |