初始化
This commit is contained in:
485
scripts/java-runner.js
Normal file
485
scripts/java-runner.js
Normal file
@@ -0,0 +1,485 @@
|
||||
const { spawn, exec, execFile } = require('child_process');
|
||||
const path = require('path');
|
||||
const fs = require('fs');
|
||||
|
||||
/**
|
||||
* Java 运行器 - 用于调用便携式 JRE 运行 Java 程序
|
||||
*/
|
||||
class JavaRunner {
|
||||
constructor() {
|
||||
// 在开发与打包后均可解析到应用根目录下的 jre 目录
|
||||
// 开发环境：项目根目录
|
||||
// 打包后：应用根目录（最终交付目录）
|
||||
const isDev = !process.resourcesPath;
|
||||
const baseDir = isDev
|
||||
? path.join(__dirname, '..')
|
||||
: path.dirname(process.resourcesPath);
|
||||
this.baseDir = baseDir;
|
||||
this.jrePath = path.join(baseDir, 'jre');
|
||||
this.binPath = path.join(this.jrePath, 'bin');
|
||||
this.javaExe = path.join(this.binPath, 'java.exe');
|
||||
this.javaRuntimeDir = path.join(this.baseDir, 'java');
|
||||
this.processRecordPath = path.join(this.javaRuntimeDir, '.running-process.json');
|
||||
}
|
||||
|
||||
normalizeComparablePath(target = '') {
|
||||
return String(target).replace(/"/g, '').replace(/\//g, '\\').toLowerCase();
|
||||
}
|
||||
|
||||
execFileCommand(command, args = []) {
|
||||
return new Promise((resolve, reject) => {
|
||||
execFile(command, args, { encoding: 'utf8', windowsHide: true }, (error, stdout, stderr) => {
|
||||
if (error) {
|
||||
const err = new Error(error.message || stderr || stdout || 'Command execution failed');
|
||||
err.code = error.code;
|
||||
err.stdout = stdout;
|
||||
err.stderr = stderr;
|
||||
reject(err);
|
||||
} else {
|
||||
resolve({ stdout, stderr });
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
getRecordedJavaProcess() {
|
||||
try {
|
||||
if (!fs.existsSync(this.processRecordPath)) {
|
||||
return null;
|
||||
}
|
||||
return JSON.parse(fs.readFileSync(this.processRecordPath, 'utf-8'));
|
||||
} catch (error) {
|
||||
console.warn('[Java] Failed to read process record:', error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
recordJavaProcess(record = {}) {
|
||||
try {
|
||||
fs.mkdirSync(this.javaRuntimeDir, { recursive: true });
|
||||
fs.writeFileSync(this.processRecordPath, JSON.stringify(record, null, 2), 'utf-8');
|
||||
} catch (error) {
|
||||
console.warn('[Java] Failed to record process metadata:', error);
|
||||
}
|
||||
}
|
||||
|
||||
cleanupJavaProcessRecord() {
|
||||
try {
|
||||
if (fs.existsSync(this.processRecordPath)) {
|
||||
fs.unlinkSync(this.processRecordPath);
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('[Java] Failed to cleanup process record:', error);
|
||||
}
|
||||
}
|
||||
|
||||
async getProcessInfoByPid(pid) {
|
||||
if (!pid) {
|
||||
return null;
|
||||
}
|
||||
|
||||
try {
|
||||
const script = `$proc = Get-CimInstance Win32_Process -Filter "ProcessId = ${pid}"; if ($proc) { [PSCustomObject]@{ executablePath = $proc.ExecutablePath; commandLine = $proc.CommandLine; name = $proc.Name } | ConvertTo-Json -Compress }`;
|
||||
const { stdout } = await this.execFileCommand('powershell.exe', ['-NoProfile', '-Command', script]);
|
||||
const raw = (stdout || '').trim();
|
||||
return raw ? JSON.parse(raw) : null;
|
||||
} catch (error) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
async isOwnJavaProcess(pid, record = this.getRecordedJavaProcess()) {
|
||||
const processInfo = await this.getProcessInfoByPid(pid);
|
||||
if (!processInfo) {
|
||||
return false;
|
||||
}
|
||||
|
||||
const executablePath = this.normalizeComparablePath(processInfo.executablePath);
|
||||
const commandLine = this.normalizeComparablePath(processInfo.commandLine);
|
||||
const expectedJavaExe = this.normalizeComparablePath(this.javaExe);
|
||||
const expectedJarPath = this.normalizeComparablePath((record && record.jarPath) || this.currentJarPath || '');
|
||||
|
||||
if (executablePath !== expectedJavaExe) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!expectedJarPath) {
|
||||
return true;
|
||||
}
|
||||
|
||||
// 关键校验：仅当命令行仍指向当前应用自己的 JAR 时，才允许清理。
|
||||
return commandLine.includes(expectedJarPath);
|
||||
}
|
||||
|
||||
async findPidsByPort(port) {
|
||||
return new Promise(resolve => {
|
||||
exec(`netstat -ano | findstr :${port}`, (error, stdout) => {
|
||||
if (error || !stdout) {
|
||||
resolve([]);
|
||||
return;
|
||||
}
|
||||
|
||||
const pids = new Set();
|
||||
stdout.trim().split('\n').forEach(line => {
|
||||
const parts = line.trim().split(/\s+/);
|
||||
const pid = parts[parts.length - 1];
|
||||
if (pid && pid !== '0') {
|
||||
pids.add(Number(pid));
|
||||
}
|
||||
});
|
||||
|
||||
resolve(Array.from(pids).filter(Boolean));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
async killProcessByPid(pid, label = 'Java') {
|
||||
return new Promise(resolve => {
|
||||
exec(`taskkill /F /T /PID ${pid}`, error => {
|
||||
if (error) {
|
||||
console.warn(`[${label}] Failed to kill PID ${pid}:`, error);
|
||||
} else {
|
||||
console.log(`[${label}] Killed PID ${pid}`);
|
||||
}
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查 JRE 是否存在
|
||||
*/
|
||||
isJREAvailable() {
|
||||
return fs.existsSync(this.javaExe);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取 Java 版本
|
||||
*/
|
||||
getVersion() {
|
||||
return new Promise((resolve, reject) => {
|
||||
if (!this.isJREAvailable()) {
|
||||
reject(new Error('JRE not found at: ' + this.javaExe));
|
||||
return;
|
||||
}
|
||||
|
||||
const versionProcess = spawn(this.javaExe, ['-version'], {
|
||||
stdio: ['ignore', 'pipe', 'pipe']
|
||||
});
|
||||
|
||||
let output = '';
|
||||
let errorOutput = '';
|
||||
|
||||
versionProcess.stdout.on('data', (data) => {
|
||||
output += data.toString();
|
||||
});
|
||||
|
||||
versionProcess.stderr.on('data', (data) => {
|
||||
errorOutput += data.toString();
|
||||
});
|
||||
|
||||
versionProcess.on('close', (code) => {
|
||||
if (code === 0 || errorOutput.includes('version')) {
|
||||
// Java -version 输出到 stderr
|
||||
const versionInfo = (output + errorOutput).trim();
|
||||
resolve(versionInfo);
|
||||
} else {
|
||||
reject(new Error('Failed to get Java version'));
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 运行 JAR 文件
|
||||
* @param {string} jarPath - JAR 文件的绝对路径
|
||||
* @param {Array<string>} args - Java 程序参数
|
||||
* @param {Object} options - spawn 选项
|
||||
* @returns {ChildProcess}
|
||||
*/
|
||||
runJar(jarPath, args = [], options = {}) {
|
||||
if (!this.isJREAvailable()) {
|
||||
throw new Error('JRE not found at: ' + this.javaExe);
|
||||
}
|
||||
|
||||
if (!fs.existsSync(jarPath)) {
|
||||
throw new Error('JAR file not found at: ' + jarPath);
|
||||
}
|
||||
|
||||
const javaArgs = ['-jar', jarPath, ...args];
|
||||
|
||||
const defaultOptions = {
|
||||
cwd: path.dirname(jarPath),
|
||||
stdio: 'inherit'
|
||||
};
|
||||
|
||||
const mergedOptions = { ...defaultOptions, ...options };
|
||||
|
||||
console.log('Running Java:', this.javaExe, javaArgs.join(' '));
|
||||
return spawn(this.javaExe, javaArgs, mergedOptions);
|
||||
}
|
||||
|
||||
/**
|
||||
* 运行 JAR 文件并等待完成
|
||||
* @param {string} jarPath - JAR 文件的绝对路径
|
||||
* @param {Array<string>} args - Java 程序参数
|
||||
* @param {Object} options - spawn 选项
|
||||
* @returns {Promise<number>} 退出代码
|
||||
*/
|
||||
runJarAsync(jarPath, args = [], options = {}) {
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
const process = this.runJar(jarPath, args, options);
|
||||
|
||||
process.on('close', (code) => {
|
||||
if (code === 0) {
|
||||
resolve(code);
|
||||
} else {
|
||||
reject(new Error(`Java process exited with code ${code}`));
|
||||
}
|
||||
});
|
||||
|
||||
process.on('error', (error) => {
|
||||
reject(error);
|
||||
});
|
||||
} catch (error) {
|
||||
reject(error);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 运行 Java 类
|
||||
* @param {string} className - Java 类名（包含包名）
|
||||
* @param {string} classPath - classpath 路径
|
||||
* @param {Array<string>} args - 程序参数
|
||||
* @param {Object} options - spawn 选项
|
||||
* @returns {ChildProcess}
|
||||
*/
|
||||
runClass(className, classPath, args = [], options = {}) {
|
||||
if (!this.isJREAvailable()) {
|
||||
throw new Error('JRE not found at: ' + this.javaExe);
|
||||
}
|
||||
|
||||
const javaArgs = ['-cp', classPath, className, ...args];
|
||||
|
||||
const defaultOptions = {
|
||||
stdio: 'inherit'
|
||||
};
|
||||
|
||||
const mergedOptions = { ...defaultOptions, ...options };
|
||||
|
||||
console.log('Running Java:', this.javaExe, javaArgs.join(' '));
|
||||
return spawn(this.javaExe, javaArgs, mergedOptions);
|
||||
}
|
||||
|
||||
/**
|
||||
* 运行 Spring Boot JAR 文件
|
||||
* @param {string} jarPath - JAR 文件的绝对路径
|
||||
* @param {string} configPath - 配置文件路径
|
||||
* @param {Object} options - 启动选项（需包含 javaPort）
|
||||
* @returns {ChildProcess}
|
||||
*/
|
||||
runSpringBoot(jarPath, configPath, options = {}) {
|
||||
if (!this.isJREAvailable()) {
|
||||
throw new Error('JRE not found at: ' + this.javaExe);
|
||||
}
|
||||
|
||||
if (!fs.existsSync(jarPath)) {
|
||||
throw new Error('JAR file not found at: ' + jarPath);
|
||||
}
|
||||
|
||||
const javaArgs = [
|
||||
'-Dfile.encoding=UTF-8', // 设置文件编码为UTF-8，解决中文乱码
|
||||
'-Duser.language=zh', // 设置语言为中文
|
||||
'-Duser.region=CN', // 设置地区为中国
|
||||
];
|
||||
|
||||
// 如果提供了 logPath，通过 JVM 系统属性传递给 logback
|
||||
if (options.logPath) {
|
||||
// Windows 路径使用正斜杠或保持原样，JVM 会自动处理
|
||||
// 方案1：转换为正斜杠（跨平台兼容）
|
||||
const normalizedLogPath = options.logPath.replace(/\\/g, '/');
|
||||
javaArgs.push(`-DlogHomeDir=${normalizedLogPath}`);
|
||||
console.log('[Java] Setting log path to:', normalizedLogPath);
|
||||
console.log('[Java] Original log path:', options.logPath);
|
||||
}
|
||||
|
||||
javaArgs.push('-jar', jarPath, `--spring.config.location=${configPath}`);
|
||||
|
||||
const defaultOptions = {
|
||||
cwd: path.dirname(jarPath),
|
||||
stdio: ['ignore', 'pipe', 'pipe'],
|
||||
env: {
|
||||
...process.env,
|
||||
JAVA_TOOL_OPTIONS: '-Dfile.encoding=UTF-8' // 额外确保UTF-8编码
|
||||
}
|
||||
};
|
||||
|
||||
const mergedOptions = { ...defaultOptions, ...options };
|
||||
|
||||
console.log('Running Spring Boot:', this.javaExe, javaArgs.join(' '));
|
||||
const javaProcess = spawn(this.javaExe, javaArgs, mergedOptions);
|
||||
|
||||
// 记录PID和端口用于后续停止
|
||||
this.springBootProcess = javaProcess;
|
||||
this.currentJavaPort = options.javaPort;
|
||||
this.currentJarPath = jarPath;
|
||||
this.recordJavaProcess({
|
||||
pid: javaProcess.pid,
|
||||
port: options.javaPort || null,
|
||||
jarPath,
|
||||
javaExe: this.javaExe,
|
||||
createdAt: new Date().toISOString()
|
||||
});
|
||||
|
||||
// 将Java端口记录到文件，供手动清理脚本使用
|
||||
if (options.javaPort) {
|
||||
this.recordJavaPort(options.javaPort);
|
||||
}
|
||||
|
||||
// 进程退出时清理端口记录
|
||||
javaProcess.on('close', () => {
|
||||
this.cleanupJavaProcessRecord();
|
||||
this.cleanupJavaPortFile();
|
||||
});
|
||||
|
||||
return javaProcess;
|
||||
}
|
||||
|
||||
/**
|
||||
* 停止 Spring Boot 应用
|
||||
*/
|
||||
async stopSpringBoot() {
|
||||
const recordedProcess = this.getRecordedJavaProcess();
|
||||
const candidateProcesses = [];
|
||||
|
||||
if (this.springBootProcess && !this.springBootProcess.killed) {
|
||||
candidateProcesses.push({
|
||||
pid: this.springBootProcess.pid,
|
||||
port: this.currentJavaPort || null,
|
||||
jarPath: this.currentJarPath || null
|
||||
});
|
||||
}
|
||||
|
||||
if (recordedProcess && recordedProcess.pid && !candidateProcesses.some(item => item.pid === recordedProcess.pid)) {
|
||||
candidateProcesses.push(recordedProcess);
|
||||
}
|
||||
|
||||
for (const candidate of candidateProcesses) {
|
||||
const isOwnProcess = await this.isOwnJavaProcess(candidate.pid, candidate);
|
||||
if (!isOwnProcess) {
|
||||
continue;
|
||||
}
|
||||
|
||||
console.log(`[Java] Stopping tracked Spring Boot process PID ${candidate.pid}`);
|
||||
await this.killProcessByPid(candidate.pid, 'Java');
|
||||
this.cleanupJavaProcessRecord();
|
||||
this.cleanupJavaPortFile();
|
||||
this.springBootProcess = null;
|
||||
this.currentJavaPort = null;
|
||||
this.currentJarPath = null;
|
||||
console.log('[Java] Spring Boot stop process completed');
|
||||
console.log('[Java] Only the current application JAR process was cleaned');
|
||||
return;
|
||||
}
|
||||
|
||||
const fallbackPort = this.currentJavaPort || recordedProcess?.port || this.getRecordedJavaPort();
|
||||
if (fallbackPort) {
|
||||
const pids = await this.findPidsByPort(fallbackPort);
|
||||
for (const pid of pids) {
|
||||
const isOwnProcess = await this.isOwnJavaProcess(pid, recordedProcess || { jarPath: this.currentJarPath });
|
||||
if (!isOwnProcess) {
|
||||
continue;
|
||||
}
|
||||
|
||||
console.log(`[Java] Stopping fallback JAR process on port ${fallbackPort}, PID ${pid}`);
|
||||
await this.killProcessByPid(pid, 'Java');
|
||||
this.cleanupJavaProcessRecord();
|
||||
this.cleanupJavaPortFile();
|
||||
this.springBootProcess = null;
|
||||
this.currentJavaPort = null;
|
||||
this.currentJarPath = null;
|
||||
console.log('[Java] Spring Boot stop process completed');
|
||||
console.log('[Java] Only the current application JAR process was cleaned');
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
this.cleanupJavaProcessRecord();
|
||||
this.cleanupJavaPortFile();
|
||||
this.springBootProcess = null;
|
||||
this.currentJavaPort = null;
|
||||
this.currentJarPath = null;
|
||||
console.log('[Java] No tracked Spring Boot process found, process record cleaned');
|
||||
}
|
||||
|
||||
/**
|
||||
* 记录Java端口到文件
|
||||
*/
|
||||
recordJavaPort(port) {
|
||||
try {
|
||||
const javaDir = this.javaRuntimeDir;
|
||||
const portFilePath = path.join(javaDir, '.running-port');
|
||||
|
||||
fs.mkdirSync(javaDir, { recursive: true });
|
||||
fs.writeFileSync(portFilePath, port.toString(), 'utf-8');
|
||||
console.log(`[Java] Port ${port} recorded to ${portFilePath}`);
|
||||
} catch (error) {
|
||||
console.warn('[Java] Failed to record port:', error);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 清理Java端口记录文件
|
||||
*/
|
||||
cleanupJavaPortFile() {
|
||||
try {
|
||||
const portFilePath = path.join(this.javaRuntimeDir, '.running-port');
|
||||
|
||||
if (fs.existsSync(portFilePath)) {
|
||||
fs.unlinkSync(portFilePath);
|
||||
console.log('[Java] Port record file cleaned up');
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('[Java] Failed to cleanup port record:', error);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取记录的Java运行端口
|
||||
*/
|
||||
getRecordedJavaPort() {
|
||||
try {
|
||||
const recordedProcess = this.getRecordedJavaProcess();
|
||||
if (recordedProcess && recordedProcess.port) {
|
||||
return parseInt(recordedProcess.port);
|
||||
}
|
||||
|
||||
const portFilePath = path.join(this.javaRuntimeDir, '.running-port');
|
||||
|
||||
if (fs.existsSync(portFilePath)) {
|
||||
const port = fs.readFileSync(portFilePath, 'utf-8').trim();
|
||||
return parseInt(port);
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('[Java] Failed to read port record:', error);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取 JRE 路径信息
|
||||
*/
|
||||
getPathInfo() {
|
||||
return {
|
||||
jrePath: this.jrePath,
|
||||
binPath: this.binPath,
|
||||
javaExe: this.javaExe,
|
||||
available: this.isJREAvailable()
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = JavaRunner;
|
||||
|
||||
Reference in New Issue
Block a user