我编写了一个小型可执行文件来为 Node 开发环境启动 WebSocket 服务器。
在一个单独的 NodeJS 应用程序(在 Electron 中运行)中,我使用
child_process.spawn()
运行二进制文件并在应用程序启动时启动服务器。
此服务器子进程劫持了节点
child_process
生成的命令并且不退出,因此我无法捕获退出代码或处理故障。我只需要知道它是否开始了。
如何在 Rust 中启动这个进程,以便我可以退出(成功、失败等)并且子进程继续运行?
我的
main.rs
的相关部分:
extern crate ws;
use std::env;
use std::process::Command;
use ws::{connect, CloseCode};
fn main() {
let args: Vec<String> = env::args().collect();
let command = &args[1];
if command == "activate" {
println!("Activating");
stop_server_process();
start_server_process();
}
}
fn start_server_process() {
let mut command;
if cfg!(windows) {
command = Command::new(".\\node_modules\\.bin\\ws.cmd");
} else {
command = Command::new("./node_modules/.bin/ws");
}
command.arg("--websocket");
command.arg("test/mock-api/ws-server.js");
command.arg("-p");
command.arg("9401");
match command.spawn() {
Ok(child) => println!("Server process id is {}", child.id()),
Err(e) => {
println!("Server didn't start: {}", e);
std::process::exit(1);
},
}
std::process::exit(0);
}
fn stop_server_process() {
connect("ws://localhost:9401", |out| {
out.send("server-stop").unwrap();
move |msg| {
println!("Got message: {}", msg);
out.close(CloseCode::Normal)
}
}).unwrap()
}
我正在从我的 Electron 应用程序运行二进制文件,如下所示:
/**
* @return {Promise<Boolean>}
*/
static activate() {
return new Promise((resolve, reject) => {
const command = spawn(process.env.SERVER_LOCAL_PATH, ['activate']);
command.stdout.on('data', (data) => {
console.log(`stdout: ${data}`);
});
command.stderr.on('data', (data) => {
console.log(`stderr: ${data}`);
});
command.on('close', code => resolve(code === 0));
command.on('error', error => reject(error));
});
}
这个问题永远不会解决,因为
ws
websocket 开始运行,并且生成的进程现在正在从中获取事件。我的控制台输出如下:
stdout: Activating <--- Rust binary output
Server process id is 78057 <--- Rust binary output
stderr: Serving at [various URL variations] <--- ws binary output
stdout: Client Connected <--- ws binary output
WebSocket 服务器基于此库中的示例和代码。这是它的可复制版本:
module.exports = SocketBase => class Socket extends SocketBase {
websocket(wss) {
wss.on('connection', (ws) => {
console.log('Client Connected');
ws.on('message', (data) => {
if (data === 'server-stop') {
console.log('Stopping...');
process.exit(0);
}
// do some faked JSON response data
});
ws.on('close', () => {
console.log('Client Disconnected');
});
ws.on('error', (error) => {
console.log(error.toString());
});
});
}
};
经过几个小时的搜索,如果有人想知道如何从 Windows 控制台分离进程,可以通过 win32console 调用控制台 api。
技巧是在子进程中调用 freeconsole() 方法:
// Start the process from parent with Command::new()
// Send markings. For instance arguments to that process via args() associated function, in this case "hidden"
use win32console::console::WinConsole;
if self.args[1] == "hidden" {
WinConsole::free_console().unwrap();
}