373 lines
12 KiB
JavaScript
373 lines
12 KiB
JavaScript
|
const { spawn, exec } = require('child_process');
|
|||
|
|
const path = require('path');
|
|||
|
|
const fs = require('fs');
|
|||
|
|
|
|||
|
|
class MySQLManager {
|
|||
|
|
constructor() {
|
|||
|
|
// 在开发与打包后均可解析到应用根目录下的 mysql 目录
|
|||
|
|
// 开发环境：项目根目录
|
|||
|
|
// 打包后：应用根目录（win-unpacked）
|
|||
|
|
const isDev = !process.resourcesPath;
|
|||
|
|
const baseDir = isDev
|
|||
|
|
? path.join(__dirname, '..')
|
|||
|
|
: path.dirname(process.resourcesPath);
|
|||
|
|
this.mysqlPath = path.join(baseDir, 'mysql');
|
|||
|
|
this.binPath = path.join(this.mysqlPath, 'bin');
|
|||
|
|
this.dataPath = path.join(this.mysqlPath, 'data');
|
|||
|
|
this.process = null;
|
|||
|
|
this.currentPort = null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查MySQL是否已初始化
|
|||
|
|
isInitialized() {
|
|||
|
|
return fs.existsSync(this.dataPath) && fs.readdirSync(this.dataPath).length > 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 初始化MySQL数据库
|
|||
|
|
async initialize() {
|
|||
|
|
if (this.isInitialized()) {
|
|||
|
|
console.log('MySQL already initialized');
|
|||
|
|
return Promise.resolve();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return new Promise(async (resolve, reject) => {
|
|||
|
|
const mysqld = path.join(this.binPath, 'mysqld.exe');
|
|||
|
|
|
|||
|
|
// 创建初始化SQL文件（授权127.0.0.1和所有主机）
|
|||
|
|
const initSqlPath = path.join(this.mysqlPath, 'init_grant.sql');
|
|||
|
|
const initSql = `
|
|||
|
|
CREATE USER IF NOT EXISTS 'root'@'127.0.0.1' IDENTIFIED BY '';
|
|||
|
|
GRANT ALL PRIVILEGES ON *.* TO 'root'@'127.0.0.1' WITH GRANT OPTION;
|
|||
|
|
CREATE USER IF NOT EXISTS 'root'@'%' IDENTIFIED BY '';
|
|||
|
|
GRANT ALL PRIVILEGES ON *.* TO 'root'@'%' WITH GRANT OPTION;
|
|||
|
|
FLUSH PRIVILEGES;
|
|||
|
|
`;
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
fs.writeFileSync(initSqlPath, initSql, 'utf-8');
|
|||
|
|
console.log('[MySQL] Created init SQL file for granting permissions');
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('[MySQL] Failed to create init SQL file:', error);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用 --init-file 参数初始化并授权
|
|||
|
|
const initProcess = spawn(mysqld, [
|
|||
|
|
'--initialize-insecure',
|
|||
|
|
`--init-file=${initSqlPath}`
|
|||
|
|
], {
|
|||
|
|
cwd: this.mysqlPath,
|
|||
|
|
stdio: 'inherit'
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
initProcess.on('close', (code) => {
|
|||
|
|
if (code === 0) {
|
|||
|
|
console.log('[MySQL] Initialized successfully with permissions granted');
|
|||
|
|
// 删除临时SQL文件
|
|||
|
|
try {
|
|||
|
|
if (fs.existsSync(initSqlPath)) {
|
|||
|
|
fs.unlinkSync(initSqlPath);
|
|||
|
|
}
|
|||
|
|
} catch (e) {
|
|||
|
|
// 忽略删除失败
|
|||
|
|
}
|
|||
|
|
resolve();
|
|||
|
|
} else {
|
|||
|
|
reject(new Error(`MySQL initialization failed with code ${code}`));
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 启动MySQL服务
|
|||
|
|
start(port = 3306) {
|
|||
|
|
return new Promise(async (resolve, reject) => {
|
|||
|
|
try {
|
|||
|
|
// 确保数据库已初始化
|
|||
|
|
await this.initialize();
|
|||
|
|
|
|||
|
|
const mysqld = path.join(this.binPath, 'mysqld.exe');
|
|||
|
|
|
|||
|
|
// 启动MySQL，指定端口
|
|||
|
|
this.process = spawn(mysqld, [
|
|||
|
|
'--console',
|
|||
|
|
`--port=${port}`
|
|||
|
|
], {
|
|||
|
|
cwd: this.mysqlPath,
|
|||
|
|
stdio: ['ignore', 'pipe', 'pipe']
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
this.currentPort = port;
|
|||
|
|
|
|||
|
|
// 将当前端口写入文件，供停止脚本使用
|
|||
|
|
try {
|
|||
|
|
const portFilePath = path.join(this.mysqlPath, '.running-port');
|
|||
|
|
fs.writeFileSync(portFilePath, port.toString(), 'utf-8');
|
|||
|
|
console.log(`[MySQL] Port ${port} recorded to ${portFilePath}`);
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] Failed to record port:', error);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let output = '';
|
|||
|
|
this.process.stdout.on('data', (data) => {
|
|||
|
|
output += data.toString();
|
|||
|
|
console.log('MySQL:', data.toString());
|
|||
|
|
|
|||
|
|
// MySQL启动完成的标志
|
|||
|
|
if (output.includes('ready for connections') || output.includes('MySQL Community Server')) {
|
|||
|
|
console.log(`MySQL started successfully on port ${port}`);
|
|||
|
|
|
|||
|
|
// 自动授权 root 用户从任何主机连接
|
|||
|
|
setTimeout(async () => {
|
|||
|
|
try {
|
|||
|
|
console.log('[MySQL] Waiting 3 seconds before granting permissions...');
|
|||
|
|
await this.grantRootAccess();
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] Failed to grant root access, but MySQL is running:', error.message);
|
|||
|
|
}
|
|||
|
|
resolve(port);
|
|||
|
|
}, 3000);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
this.process.stderr.on('data', (data) => {
|
|||
|
|
console.error('MySQL Error:', data.toString());
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
this.process.on('close', (code) => {
|
|||
|
|
console.log(`MySQL process exited with code ${code}`);
|
|||
|
|
this.process = null;
|
|||
|
|
this.currentPort = null;
|
|||
|
|
|
|||
|
|
// 删除端口记录文件
|
|||
|
|
try {
|
|||
|
|
const portFilePath = path.join(this.mysqlPath, '.running-port');
|
|||
|
|
if (fs.existsSync(portFilePath)) {
|
|||
|
|
fs.unlinkSync(portFilePath);
|
|||
|
|
console.log('[MySQL] Port record file removed');
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] Failed to remove port record:', error);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 超时处理
|
|||
|
|
setTimeout(async () => {
|
|||
|
|
if (this.process && !this.process.killed) {
|
|||
|
|
console.log(`MySQL started on port ${port} (timeout reached, assuming success)`);
|
|||
|
|
|
|||
|
|
// 自动授权 root 用户从任何主机连接
|
|||
|
|
try {
|
|||
|
|
console.log('[MySQL] Granting permissions...');
|
|||
|
|
await this.grantRootAccess();
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] Failed to grant root access, but MySQL is running:', error.message);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resolve(port);
|
|||
|
|
}
|
|||
|
|
}, 18000);
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
reject(error);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 授权 root 用户从 127.0.0.1 访问
|
|||
|
|
grantRootAccess() {
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
// 创建 SQL 文件
|
|||
|
|
const sqlFilePath = path.join(this.mysqlPath, 'grant_root.sql');
|
|||
|
|
const sqlContent = `
|
|||
|
|
CREATE USER IF NOT EXISTS 'root'@'127.0.0.1' IDENTIFIED BY '';
|
|||
|
|
GRANT ALL PRIVILEGES ON *.* TO 'root'@'127.0.0.1' WITH GRANT OPTION;
|
|||
|
|
CREATE USER IF NOT EXISTS 'root'@'%' IDENTIFIED BY '';
|
|||
|
|
GRANT ALL PRIVILEGES ON *.* TO 'root'@'%' WITH GRANT OPTION;
|
|||
|
|
FLUSH PRIVILEGES;
|
|||
|
|
`;
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
fs.writeFileSync(sqlFilePath, sqlContent, 'utf-8');
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('[MySQL] Failed to create grant SQL file:', error);
|
|||
|
|
return resolve(); // 继续启动
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const mysqlExe = path.join(this.binPath, 'mysql.exe');
|
|||
|
|
const grantProcess = spawn(mysqlExe, [
|
|||
|
|
'--host=localhost',
|
|||
|
|
`--port=${this.currentPort}`,
|
|||
|
|
'--user=root'
|
|||
|
|
], {
|
|||
|
|
cwd: this.mysqlPath,
|
|||
|
|
stdio: ['pipe', 'pipe', 'pipe']
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 通过 stdin 输入 SQL
|
|||
|
|
grantProcess.stdin.write(sqlContent);
|
|||
|
|
grantProcess.stdin.end();
|
|||
|
|
|
|||
|
|
let output = '';
|
|||
|
|
let errorOutput = '';
|
|||
|
|
|
|||
|
|
grantProcess.stdout.on('data', (data) => {
|
|||
|
|
output += data.toString();
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
grantProcess.stderr.on('data', (data) => {
|
|||
|
|
errorOutput += data.toString();
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
grantProcess.on('close', (code) => {
|
|||
|
|
if (code === 0) {
|
|||
|
|
console.log('[MySQL] Root user granted access from 127.0.0.1 and all hosts');
|
|||
|
|
resolve();
|
|||
|
|
} else {
|
|||
|
|
console.error('[MySQL] Grant access failed (code:', code, ')');
|
|||
|
|
console.error('[MySQL] Error output:', errorOutput);
|
|||
|
|
console.error('[MySQL] Standard output:', output);
|
|||
|
|
// 即使失败也 resolve，让应用继续启动
|
|||
|
|
resolve();
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取当前MySQL端口
|
|||
|
|
getCurrentPort() {
|
|||
|
|
return this.currentPort || 3306;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 停止MySQL服务
|
|||
|
|
stop() {
|
|||
|
|
return new Promise(async (resolve) => {
|
|||
|
|
if (this.process && !this.process.killed) {
|
|||
|
|
console.log('[MySQL] Stopping MySQL...');
|
|||
|
|
|
|||
|
|
// 方法1: 尝试使用 mysqladmin shutdown 优雅关闭
|
|||
|
|
try {
|
|||
|
|
console.log('[MySQL] Trying mysqladmin shutdown...');
|
|||
|
|
const mysqladmin = path.join(this.binPath, 'mysqladmin.exe');
|
|||
|
|
|
|||
|
|
if (fs.existsSync(mysqladmin)) {
|
|||
|
|
const shutdownProcess = spawn(mysqladmin, [
|
|||
|
|
'-u', 'root',
|
|||
|
|
'-pnjcnpqs',
|
|||
|
|
'--port=' + this.currentPort,
|
|||
|
|
'shutdown'
|
|||
|
|
], {
|
|||
|
|
cwd: this.mysqlPath,
|
|||
|
|
stdio: 'ignore'
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 等待 mysqladmin 执行完成（最多5秒）
|
|||
|
|
const shutdownPromise = new Promise((res) => {
|
|||
|
|
shutdownProcess.on('close', (code) => {
|
|||
|
|
console.log(`[MySQL] mysqladmin shutdown exited with code ${code}`);
|
|||
|
|
res(code === 0);
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const timeoutPromise = new Promise((res) => setTimeout(() => res(false), 5000));
|
|||
|
|
const shutdownSuccess = await Promise.race([shutdownPromise, timeoutPromise]);
|
|||
|
|
|
|||
|
|
if (shutdownSuccess) {
|
|||
|
|
console.log('[MySQL] Shutdown successful via mysqladmin');
|
|||
|
|
// 等待进程真正退出
|
|||
|
|
await new Promise((res) => {
|
|||
|
|
if (this.process && !this.process.killed) {
|
|||
|
|
this.process.on('close', res);
|
|||
|
|
setTimeout(res, 2000); // 最多等2秒
|
|||
|
|
} else {
|
|||
|
|
res();
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
this.cleanupPortFile();
|
|||
|
|
return resolve();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] mysqladmin shutdown failed:', error.message);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 方法2: 如果 mysqladmin 失败，尝试 SIGTERM
|
|||
|
|
console.log('[MySQL] Trying SIGTERM...');
|
|||
|
|
const killTimeout = setTimeout(() => {
|
|||
|
|
// 方法3: 5秒后强制 SIGKILL
|
|||
|
|
console.log('[MySQL] Force killing with SIGKILL');
|
|||
|
|
try {
|
|||
|
|
if (this.process && !this.process.killed) {
|
|||
|
|
this.process.kill('SIGKILL');
|
|||
|
|
}
|
|||
|
|
} catch (e) {
|
|||
|
|
console.error('[MySQL] Error force killing:', e);
|
|||
|
|
}
|
|||
|
|
this.cleanupPortFile();
|
|||
|
|
resolve();
|
|||
|
|
}, 5000);
|
|||
|
|
|
|||
|
|
this.process.on('close', () => {
|
|||
|
|
clearTimeout(killTimeout);
|
|||
|
|
console.log('[MySQL] Process closed');
|
|||
|
|
this.cleanupPortFile();
|
|||
|
|
resolve();
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
this.process.kill('SIGTERM');
|
|||
|
|
} catch (e) {
|
|||
|
|
console.error('[MySQL] Error sending SIGTERM:', e);
|
|||
|
|
clearTimeout(killTimeout);
|
|||
|
|
this.cleanupPortFile();
|
|||
|
|
resolve();
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 没有进程引用，说明MySQL已经停止或不在我们控制下
|
|||
|
|
console.log('[MySQL] No process reference, MySQL may already be stopped');
|
|||
|
|
console.log('[MySQL] If MySQL is still running, please use kill-running-port.bat to clean up');
|
|||
|
|
this.cleanupPortFile();
|
|||
|
|
resolve();
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 清理端口记录文件
|
|||
|
|
cleanupPortFile() {
|
|||
|
|
try {
|
|||
|
|
const portFilePath = path.join(this.mysqlPath, '.running-port');
|
|||
|
|
if (fs.existsSync(portFilePath)) {
|
|||
|
|
fs.unlinkSync(portFilePath);
|
|||
|
|
console.log('[MySQL] Port record file cleaned up');
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] Failed to cleanup port record:', error);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取记录的运行端口
|
|||
|
|
getRecordedPort() {
|
|||
|
|
try {
|
|||
|
|
const portFilePath = path.join(this.mysqlPath, '.running-port');
|
|||
|
|
if (fs.existsSync(portFilePath)) {
|
|||
|
|
const port = fs.readFileSync(portFilePath, 'utf-8').trim();
|
|||
|
|
return parseInt(port);
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn('[MySQL] Failed to read port record:', error);
|
|||
|
|
}
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取MySQL连接配置
|
|||
|
|
getConnectionConfig() {
|
|||
|
|
return {
|
|||
|
|
host: 'localhost',
|
|||
|
|
port: 3306,
|
|||
|
|
user: 'root',
|
|||
|
|
password: '',
|
|||
|
|
database: 'app_db'
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = MySQLManager;
|