56 lines
1.7 KiB
JavaScript
56 lines
1.7 KiB
JavaScript
|
const mysql = require('mysql2/promise');
|
|||
|
|
const path = require('path');
|
|||
|
|
const fs = require('fs');
|
|||
|
|
|
|||
|
|
// 数据库配置（从.env读取或使用默认值）
|
|||
|
|
const config = {
|
|||
|
|
host: process.env.DB_HOST || '129.211.213.226',
|
|||
|
|
port: process.env.DB_PORT || 9527,
|
|||
|
|
user: process.env.DB_USER || 'root',
|
|||
|
|
password: process.env.DB_PASSWORD || 'aiotAiot123!',
|
|||
|
|
database: process.env.DB_NAME || 'ningxia_zhengfu',
|
|||
|
|
multipleStatements: true // 允许执行多条SQL语句
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
async function run() {
|
|||
|
|
let connection;
|
|||
|
|
try {
|
|||
|
|
// 1. 连接数据库
|
|||
|
|
connection = await mysql.createConnection(config);
|
|||
|
|
console.log('数据库连接成功');
|
|||
|
|
|
|||
|
|
// 2. 执行建表SQL
|
|||
|
|
const schemaSql = fs.readFileSync(
|
|||
|
|
path.join(__dirname, '../sql/schema.sql'),
|
|||
|
|
'utf8'
|
|||
|
|
);
|
|||
|
|
await connection.query(schemaSql);
|
|||
|
|
console.log('数据库表创建完成');
|
|||
|
|
|
|||
|
|
// 3. 插入测试数据
|
|||
|
|
const dataSql = fs.readFileSync(
|
|||
|
|
path.join(__dirname, '../sql/seed-data.sql'),
|
|||
|
|
'utf8'
|
|||
|
|
);
|
|||
|
|
await connection.query(dataSql);
|
|||
|
|
console.log('测试数据插入完成');
|
|||
|
|
|
|||
|
|
// 4. 验证数据
|
|||
|
|
const [entities] = await connection.query('SELECT COUNT(*) as count FROM supervision_entities');
|
|||
|
|
const [inspections] = await connection.query('SELECT COUNT(*) as count FROM inspections');
|
|||
|
|
console.log('\n数据验证结果:');
|
|||
|
|
console.log(`监管实体数量: ${entities[0].count}`);
|
|||
|
|
console.log(`检查记录数量: ${inspections[0].count}`);
|
|||
|
|
|
|||
|
|
} catch (err) {
|
|||
|
|
console.error('数据库操作失败:', err.message);
|
|||
|
|
} finally {
|
|||
|
|
if (connection) {
|
|||
|
|
await connection.end();
|
|||
|
|
console.log('数据库连接已关闭');
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 立即执行
|
|||
|
|
run();
|