docs: 更新项目文档，完善需求和技术细节
This commit is contained in:
ylweng
165
scripts/backend/initDatabase.js
Normal file
165
scripts/backend/initDatabase.js
Normal file
@@ -0,0 +1,165 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
/**
|
||||
* 数据库初始化脚本
|
||||
* 用于验证数据库连接和创建基础表结构
|
||||
*/
|
||||
|
||||
const mysql = require('mysql2/promise');
|
||||
const databaseConfig = require('../config/database');
|
||||
const path = require('path');
|
||||
const fs = require('fs');
|
||||
|
||||
class DatabaseInitializer {
|
||||
constructor() {
|
||||
this.connection = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建数据库连接
|
||||
*/
|
||||
async createConnection() {
|
||||
try {
|
||||
this.connection = await mysql.createConnection({
|
||||
host: databaseConfig.host,
|
||||
port: databaseConfig.port,
|
||||
user: databaseConfig.username,
|
||||
password: databaseConfig.password,
|
||||
database: databaseConfig.database,
|
||||
charset: 'utf8mb4',
|
||||
timezone: '+08:00'
|
||||
});
|
||||
|
||||
console.log('✅ 数据库连接成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 数据库连接失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证数据库连接
|
||||
*/
|
||||
async validateConnection() {
|
||||
try {
|
||||
const [rows] = await this.connection.execute('SELECT NOW() as current_time, VERSION() as mysql_version');
|
||||
console.log('📊 数据库信息:');
|
||||
console.log(` 当前时间: ${rows[0].current_time}`);
|
||||
console.log(` MySQL版本: ${rows[0].mysql_version}`);
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 数据库验证失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查表是否存在
|
||||
* @param {string} tableName - 表名
|
||||
*/
|
||||
async checkTableExists(tableName) {
|
||||
try {
|
||||
const [rows] = await this.connection.execute(
|
||||
`SELECT COUNT(*) as count FROM information_schema.tables
|
||||
WHERE table_schema = ? AND table_name = ?`,
|
||||
[databaseConfig.database, tableName]
|
||||
);
|
||||
return rows[0].count > 0;
|
||||
} catch (error) {
|
||||
console.error(`❌ 检查表 ${tableName} 存在失败:`, error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行SQL文件
|
||||
* @param {string} filePath - SQL文件路径
|
||||
*/
|
||||
async executeSqlFile(filePath) {
|
||||
try {
|
||||
const sqlContent = fs.readFileSync(filePath, 'utf8');
|
||||
const statements = sqlContent.split(';').filter(stmt => stmt.trim());
|
||||
|
||||
for (const statement of statements) {
|
||||
if (statement.trim()) {
|
||||
await this.connection.execute(statement);
|
||||
}
|
||||
}
|
||||
|
||||
console.log(`✅ 成功执行SQL文件: ${path.basename(filePath)}`);
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error(`❌ 执行SQL文件失败:`, error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库连接
|
||||
*/
|
||||
async closeConnection() {
|
||||
if (this.connection) {
|
||||
await this.connection.end();
|
||||
console.log('🔌 数据库连接已关闭');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 主初始化方法
|
||||
*/
|
||||
async initialize() {
|
||||
console.log('🚀 开始数据库初始化...');
|
||||
console.log(`📋 环境: ${process.env.NODE_ENV || 'development'}`);
|
||||
console.log(`🗄️ 数据库: ${databaseConfig.database}`);
|
||||
console.log(`🌐 主机: ${databaseConfig.host}:${databaseConfig.port}`);
|
||||
console.log('─'.repeat(50));
|
||||
|
||||
// 创建连接
|
||||
const connected = await this.createConnection();
|
||||
if (!connected) {
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
// 验证连接
|
||||
const validated = await this.validateConnection();
|
||||
if (!validated) {
|
||||
await this.closeConnection();
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
console.log('✅ 数据库连接验证通过');
|
||||
console.log('─'.repeat(50));
|
||||
|
||||
// 这里可以添加具体的表创建逻辑
|
||||
console.log('📋 数据库初始化完成');
|
||||
console.log('✅ 所有检查通过，数据库连接正常');
|
||||
|
||||
await this.closeConnection();
|
||||
}
|
||||
}
|
||||
|
||||
// 执行初始化
|
||||
const initializer = new DatabaseInitializer();
|
||||
|
||||
// 处理命令行参数
|
||||
const args = process.argv.slice(2);
|
||||
if (args.includes('--help') || args.includes('-h')) {
|
||||
console.log(`
|
||||
使用方法: node scripts/initDatabase.js [选项]
|
||||
|
||||
选项:
|
||||
--help, -h 显示帮助信息
|
||||
--check 只检查连接，不执行初始化
|
||||
|
||||
示例:
|
||||
node scripts/initDatabase.js # 完整初始化
|
||||
node scripts/initDatabase.js --check # 只检查连接
|
||||
`);
|
||||
process.exit(0);
|
||||
}
|
||||
|
||||
initializer.initialize().catch(error => {
|
||||
console.error('❌ 初始化过程中发生错误:', error.message);
|
||||
process.exit(1);
|
||||
});
|
||||
260
scripts/backend/initSQLite.js
Normal file
260
scripts/backend/initSQLite.js
Normal file
@@ -0,0 +1,260 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
/**
|
||||
* SQLite数据库初始化脚本
|
||||
* 用于创建SQLite数据库表结构
|
||||
*/
|
||||
|
||||
const sqlite3 = require('sqlite3').verbose();
|
||||
const path = require('path');
|
||||
const databaseConfig = require('../config/database');
|
||||
|
||||
class SQLiteInitializer {
|
||||
constructor() {
|
||||
this.db = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建数据库连接
|
||||
*/
|
||||
createConnection() {
|
||||
try {
|
||||
this.db = new sqlite3.Database(databaseConfig.storage);
|
||||
console.log('✅ SQLite数据库连接成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ SQLite数据库连接失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行SQL语句
|
||||
*/
|
||||
run(sql, params = []) {
|
||||
return new Promise((resolve, reject) => {
|
||||
this.db.run(sql, params, function(err) {
|
||||
if (err) {
|
||||
reject(err);
|
||||
} else {
|
||||
resolve(this);
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建users表
|
||||
*/
|
||||
async createUsersTable() {
|
||||
const sql = `
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
username VARCHAR(50) UNIQUE NOT NULL,
|
||||
password VARCHAR(255) NOT NULL,
|
||||
phone VARCHAR(20) UNIQUE,
|
||||
email VARCHAR(100) UNIQUE,
|
||||
user_type VARCHAR(20) DEFAULT 'user',
|
||||
status INTEGER DEFAULT 1,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`;
|
||||
|
||||
try {
|
||||
await this.run(sql);
|
||||
console.log('✅ users表创建成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 创建users表失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建orders表
|
||||
*/
|
||||
async createOrdersTable() {
|
||||
const sql = `
|
||||
CREATE TABLE IF NOT EXISTS orders (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
order_number VARCHAR(50) UNIQUE NOT NULL,
|
||||
user_id INTEGER NOT NULL,
|
||||
total_amount DECIMAL(10,2) NOT NULL,
|
||||
payment_status INTEGER DEFAULT 0,
|
||||
shipping_status INTEGER DEFAULT 0,
|
||||
shipping_address TEXT,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
FOREIGN KEY (user_id) REFERENCES users (id)
|
||||
)
|
||||
`;
|
||||
|
||||
try {
|
||||
await this.run(sql);
|
||||
console.log('✅ orders表创建成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 创建orders表失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建identifications表
|
||||
*/
|
||||
async createIdentificationsTable() {
|
||||
const sql = `
|
||||
CREATE TABLE IF NOT EXISTS identifications (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
user_id INTEGER NOT NULL,
|
||||
image_url VARCHAR(255) NOT NULL,
|
||||
result TEXT NOT NULL,
|
||||
confidence DECIMAL(5,4) NOT NULL,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
FOREIGN KEY (user_id) REFERENCES users (id)
|
||||
)
|
||||
`;
|
||||
|
||||
try {
|
||||
await this.run(sql);
|
||||
console.log('✅ identifications表创建成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 创建identifications表失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建products表
|
||||
*/
|
||||
async createProductsTable() {
|
||||
const sql = `
|
||||
CREATE TABLE IF NOT EXISTS products (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
name VARCHAR(100) NOT NULL,
|
||||
category_id INTEGER,
|
||||
price DECIMAL(10,2) NOT NULL,
|
||||
stock INTEGER DEFAULT 0,
|
||||
image VARCHAR(255),
|
||||
description TEXT,
|
||||
status INTEGER DEFAULT 1,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`;
|
||||
|
||||
try {
|
||||
await this.run(sql);
|
||||
console.log('✅ products表创建成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 创建products表失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入初始管理员用户
|
||||
*/
|
||||
async insertAdminUser() {
|
||||
const sql = `
|
||||
INSERT OR IGNORE INTO users (username, password, phone, email, user_type, status)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
`;
|
||||
|
||||
// 密码: admin123 (bcrypt加密)
|
||||
const hashedPassword = '$2b$10$92IXUNpkjO0rOQ5byMi.Ye4oKoEa3Ro9llC/.og/at2.uheWG/igi';
|
||||
|
||||
try {
|
||||
await this.run(sql, [
|
||||
'admin',
|
||||
hashedPassword,
|
||||
'13800138000',
|
||||
'admin@example.com',
|
||||
'admin',
|
||||
1
|
||||
]);
|
||||
console.log('✅ 管理员用户创建成功');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('❌ 创建管理员用户失败:', error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库连接
|
||||
*/
|
||||
closeConnection() {
|
||||
if (this.db) {
|
||||
this.db.close();
|
||||
console.log('🔌 数据库连接已关闭');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 主初始化方法
|
||||
*/
|
||||
async initialize() {
|
||||
console.log('🚀 开始SQLite数据库初始化...');
|
||||
console.log(`📋 环境: ${process.env.NODE_ENV || 'development'}`);
|
||||
console.log(`🗄️ 数据库文件: ${databaseConfig.storage}`);
|
||||
console.log('─'.repeat(50));
|
||||
|
||||
// 创建连接
|
||||
const connected = this.createConnection();
|
||||
if (!connected) {
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
// 创建表
|
||||
const tablesCreated = await Promise.all([
|
||||
this.createUsersTable(),
|
||||
this.createOrdersTable(),
|
||||
this.createIdentificationsTable(),
|
||||
this.createProductsTable()
|
||||
]);
|
||||
|
||||
if (tablesCreated.every(result => result)) {
|
||||
console.log('✅ 所有表创建成功');
|
||||
} else {
|
||||
console.error('❌ 表创建过程中出现错误');
|
||||
this.closeConnection();
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
// 插入初始数据
|
||||
const adminCreated = await this.insertAdminUser();
|
||||
if (!adminCreated) {
|
||||
console.warn('⚠️ 管理员用户创建失败或已存在');
|
||||
}
|
||||
|
||||
console.log('✅ SQLite数据库初始化完成');
|
||||
this.closeConnection();
|
||||
}
|
||||
}
|
||||
|
||||
// 执行初始化
|
||||
const initializer = new SQLiteInitializer();
|
||||
|
||||
// 处理命令行参数
|
||||
const args = process.argv.slice(2);
|
||||
if (args.includes('--help') || args.includes('-h')) {
|
||||
console.log(`
|
||||
使用方法: node scripts/initSQLite.js [选项]
|
||||
|
||||
选项:
|
||||
--help, -h 显示帮助信息
|
||||
|
||||
示例:
|
||||
node scripts/initSQLite.js # 完整初始化
|
||||
`);
|
||||
process.exit(0);
|
||||
}
|
||||
|
||||
initializer.initialize().catch(error => {
|
||||
console.error('❌ 初始化过程中发生错误:', error.message);
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user