feat: 优化 Nitro sidecar 进程启动与监控逻辑
- 引入必要的同步与时间处理模块,优化 Nitro sidecar 进程的启动与监控逻辑,确保服务器就绪后正确创建主窗口并处理启动超时和失败情况。
This commit is contained in:
36
src/lib.rs
36
src/lib.rs
@@ -1,8 +1,8 @@
|
|||||||
use tauri::Manager;
|
|
||||||
use tauri_plugin_shell::ShellExt;
|
|
||||||
use tauri_plugin_shell::process::{CommandEvent, CommandChild};
|
|
||||||
use std::time::Duration;
|
|
||||||
use std::sync::Mutex;
|
use std::sync::Mutex;
|
||||||
|
use std::time::Duration;
|
||||||
|
use tauri::Manager;
|
||||||
|
use tauri_plugin_shell::process::{CommandChild, CommandEvent};
|
||||||
|
use tauri_plugin_shell::ShellExt;
|
||||||
|
|
||||||
// 全局状态:存储 Nitro 进程句柄
|
// 全局状态:存储 Nitro 进程句柄
|
||||||
struct NitroProcess(Mutex<Option<CommandChild>>);
|
struct NitroProcess(Mutex<Option<CommandChild>>);
|
||||||
@@ -38,74 +38,74 @@ pub fn run() {
|
|||||||
.setup(|app| {
|
.setup(|app| {
|
||||||
// 使用 Tauri 的状态管理存储进程句柄
|
// 使用 Tauri 的状态管理存储进程句柄
|
||||||
app.manage(NitroProcess(Mutex::new(None)));
|
app.manage(NitroProcess(Mutex::new(None)));
|
||||||
|
|
||||||
let app_handle = app.handle().clone();
|
let app_handle = app.handle().clone();
|
||||||
|
|
||||||
// 异步启动 Nitro sidecar
|
// 异步启动 Nitro sidecar
|
||||||
tauri::async_runtime::spawn(async move {
|
tauri::async_runtime::spawn(async move {
|
||||||
// 查找可用端口
|
// 查找可用端口
|
||||||
let port = find_available_port(3000).await;
|
let port = find_available_port(3000).await;
|
||||||
println!("使用端口: {}", port);
|
println!("使用端口: {}", port);
|
||||||
|
|
||||||
// 启动 sidecar
|
// 启动 sidecar
|
||||||
let sidecar = app_handle
|
let sidecar = app_handle
|
||||||
.shell()
|
.shell()
|
||||||
.sidecar("nitro-server")
|
.sidecar("nitro-server")
|
||||||
.expect("无法找到 nitro-server sidecar")
|
.expect("无法找到 nitro-server sidecar")
|
||||||
.env("NITRO_PORT", port.to_string());
|
.env("NITRO_PORT", port.to_string());
|
||||||
|
|
||||||
let (mut rx, child) = sidecar.spawn().expect("启动 sidecar 失败");
|
let (mut rx, child) = sidecar.spawn().expect("启动 sidecar 失败");
|
||||||
|
|
||||||
// 保存进程句柄到全局状态
|
// 保存进程句柄到全局状态
|
||||||
if let Some(state) = app_handle.try_state::<NitroProcess>() {
|
if let Some(state) = app_handle.try_state::<NitroProcess>() {
|
||||||
*state.0.lock().unwrap() = Some(child);
|
*state.0.lock().unwrap() = Some(child);
|
||||||
}
|
}
|
||||||
|
|
||||||
// 监听 stdout,等待服务器就绪信号
|
// 监听 stdout,等待服务器就绪信号
|
||||||
let start_time = std::time::Instant::now();
|
let start_time = std::time::Instant::now();
|
||||||
let timeout = Duration::from_secs(5);
|
let timeout = Duration::from_secs(5);
|
||||||
let mut server_ready = false;
|
let mut server_ready = false;
|
||||||
|
|
||||||
while let Some(event) = rx.recv().await {
|
while let Some(event) = rx.recv().await {
|
||||||
if let CommandEvent::Stdout(line) = event {
|
if let CommandEvent::Stdout(line) = event {
|
||||||
let output = String::from_utf8_lossy(&line);
|
let output = String::from_utf8_lossy(&line);
|
||||||
println!("Nitro: {}", output);
|
println!("Nitro: {}", output);
|
||||||
|
|
||||||
// 检测服务器启动成功的标志
|
// 检测服务器启动成功的标志
|
||||||
if output.contains("Listening on:") || output.contains("localhost") {
|
if output.contains("Listening on:") || output.contains("localhost") {
|
||||||
server_ready = true;
|
server_ready = true;
|
||||||
println!("✓ Nitro 服务器启动成功!");
|
println!("✓ Nitro 服务器启动成功!");
|
||||||
|
|
||||||
// 创建主窗口
|
// 创建主窗口
|
||||||
let url = format!("http://localhost:{}", port);
|
let url = format!("http://localhost:{}", port);
|
||||||
tauri::WebviewWindowBuilder::new(
|
tauri::WebviewWindowBuilder::new(
|
||||||
&app_handle,
|
&app_handle,
|
||||||
"main",
|
"main",
|
||||||
tauri::WebviewUrl::External(url.parse().unwrap())
|
tauri::WebviewUrl::External(url.parse().unwrap()),
|
||||||
)
|
)
|
||||||
.title("Nitro Application")
|
.title("Nitro Application")
|
||||||
.inner_size(1200.0, 800.0)
|
.inner_size(1200.0, 800.0)
|
||||||
.center()
|
.center()
|
||||||
.build()
|
.build()
|
||||||
.expect("创建窗口失败");
|
.expect("创建窗口失败");
|
||||||
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 超时检查
|
// 超时检查
|
||||||
if start_time.elapsed() > timeout {
|
if start_time.elapsed() > timeout {
|
||||||
eprintln!("✗ 启动超时:Nitro 服务器未能在 5 秒内启动");
|
eprintln!("✗ 启动超时:Nitro 服务器未能在 5 秒内启动");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if !server_ready {
|
if !server_ready {
|
||||||
eprintln!("✗ Nitro 服务器启动失败");
|
eprintln!("✗ Nitro 服务器启动失败");
|
||||||
std::process::exit(1);
|
std::process::exit(1);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
})
|
})
|
||||||
.invoke_handler(tauri::generate_handler![greet])
|
.invoke_handler(tauri::generate_handler![greet])
|
||||||
|
|||||||
Reference in New Issue
Block a user