forked from imbytecat/fullstack-starter
refactor: 优化服务器启动逻辑并移除冗余端口检测
- 移除冗余的端口占用检测函数并优化开发与生产模式下的服务器启动逻辑,提升代码可读性与启动可靠性。
This commit is contained in:
@@ -13,9 +13,6 @@ const STARTUP_TIMEOUT_SECS: u64 = 5;
|
|||||||
/// 默认起始端口
|
/// 默认起始端口
|
||||||
const DEFAULT_PORT: u16 = 3000;
|
const DEFAULT_PORT: u16 = 3000;
|
||||||
|
|
||||||
/// 开发模式使用的端口
|
|
||||||
const DEV_PORT: u16 = 3000;
|
|
||||||
|
|
||||||
/// 端口扫描范围(从起始端口开始扫描的端口数量)
|
/// 端口扫描范围(从起始端口开始扫描的端口数量)
|
||||||
const PORT_SCAN_RANGE: u16 = 100;
|
const PORT_SCAN_RANGE: u16 = 100;
|
||||||
|
|
||||||
@@ -40,19 +37,6 @@ async fn is_port_available(port: u16) -> bool {
|
|||||||
.is_ok()
|
.is_ok()
|
||||||
}
|
}
|
||||||
|
|
||||||
// 检查端口是否被占用(服务器正在监听)
|
|
||||||
async fn is_port_in_use(port: u16) -> bool {
|
|
||||||
use tokio::io::AsyncWriteExt;
|
|
||||||
|
|
||||||
match tokio::net::TcpStream::connect(format!("127.0.0.1:{}", port)).await {
|
|
||||||
Ok(mut stream) => {
|
|
||||||
let _ = stream.shutdown().await;
|
|
||||||
true
|
|
||||||
}
|
|
||||||
Err(_) => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查找可用端口
|
// 查找可用端口
|
||||||
async fn find_available_port(start: u16) -> u16 {
|
async fn find_available_port(start: u16) -> u16 {
|
||||||
for port in start..start + PORT_SCAN_RANGE {
|
for port in start..start + PORT_SCAN_RANGE {
|
||||||
@@ -65,117 +49,95 @@ async fn find_available_port(start: u16) -> u16 {
|
|||||||
|
|
||||||
/// 启动 Sidecar 进程并创建主窗口
|
/// 启动 Sidecar 进程并创建主窗口
|
||||||
pub fn spawn_sidecar(app_handle: tauri::AppHandle) {
|
pub fn spawn_sidecar(app_handle: tauri::AppHandle) {
|
||||||
|
// 检测是否为开发模式
|
||||||
|
let is_dev = cfg!(debug_assertions);
|
||||||
|
|
||||||
|
if is_dev {
|
||||||
|
// 开发模式:直接创建窗口连接到 Vite 开发服务器
|
||||||
|
println!("🔧 开发模式:连接到 Vite 开发服务器 (localhost:3000)");
|
||||||
|
|
||||||
|
let url = "http://localhost:3000";
|
||||||
|
tauri::WebviewWindowBuilder::new(
|
||||||
|
&app_handle,
|
||||||
|
"main",
|
||||||
|
tauri::WebviewUrl::External(url.parse().unwrap()),
|
||||||
|
)
|
||||||
|
.title(WINDOW_TITLE)
|
||||||
|
.inner_size(DEFAULT_WINDOW_WIDTH, DEFAULT_WINDOW_HEIGHT)
|
||||||
|
.center()
|
||||||
|
.build()
|
||||||
|
.expect("创建窗口失败");
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生产模式:启动 sidecar 二进制
|
||||||
tauri::async_runtime::spawn(async move {
|
tauri::async_runtime::spawn(async move {
|
||||||
// 检测是否为开发模式
|
println!("🚀 生产模式:启动 Sidecar Server");
|
||||||
let is_dev = cfg!(debug_assertions);
|
|
||||||
|
|
||||||
if is_dev {
|
// 查找可用端口
|
||||||
// 开发模式:直接连接到 localhost:3000
|
let port = find_available_port(DEFAULT_PORT).await;
|
||||||
println!("🔧 开发模式:使用本地开发服务器 (localhost:{})", DEV_PORT);
|
println!("使用端口: {}", port);
|
||||||
|
|
||||||
// 等待开发服务器就绪(可选:添加重试逻辑)
|
// 启动 sidecar
|
||||||
let max_retries = 10;
|
let sidecar = app_handle
|
||||||
let retry_delay = Duration::from_millis(500);
|
.shell()
|
||||||
let mut server_ready = false;
|
.sidecar("server")
|
||||||
|
.expect("无法找到 server")
|
||||||
|
.env("NITRO_PORT", port.to_string());
|
||||||
|
|
||||||
for attempt in 1..=max_retries {
|
let (mut rx, child) = sidecar.spawn().expect("启动 sidecar 失败");
|
||||||
if is_port_in_use(DEV_PORT).await {
|
|
||||||
|
// 保存进程句柄到全局状态
|
||||||
|
if let Some(state) = app_handle.try_state::<SidecarProcess>() {
|
||||||
|
*state.0.lock().unwrap() = Some(child);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 监听 stdout,等待服务器就绪信号
|
||||||
|
let start_time = std::time::Instant::now();
|
||||||
|
let timeout = Duration::from_secs(STARTUP_TIMEOUT_SECS);
|
||||||
|
let mut server_ready = false;
|
||||||
|
|
||||||
|
while let Some(event) = rx.recv().await {
|
||||||
|
if let CommandEvent::Stdout(line) = event {
|
||||||
|
let output = String::from_utf8_lossy(&line);
|
||||||
|
println!("Server: {}", output);
|
||||||
|
|
||||||
|
// 检测服务器启动成功的标志
|
||||||
|
if output.contains("Listening on:") || output.contains("localhost") {
|
||||||
server_ready = true;
|
server_ready = true;
|
||||||
println!("✓ 开发服务器已就绪 (端口 {})", DEV_PORT);
|
println!("✓ Server 启动成功!");
|
||||||
|
|
||||||
|
// 创建主窗口
|
||||||
|
let url = format!("http://localhost:{}", port);
|
||||||
|
tauri::WebviewWindowBuilder::new(
|
||||||
|
&app_handle,
|
||||||
|
"main",
|
||||||
|
tauri::WebviewUrl::External(url.parse().unwrap()),
|
||||||
|
)
|
||||||
|
.title(WINDOW_TITLE)
|
||||||
|
.inner_size(DEFAULT_WINDOW_WIDTH, DEFAULT_WINDOW_HEIGHT)
|
||||||
|
.center()
|
||||||
|
.build()
|
||||||
|
.expect("创建窗口失败");
|
||||||
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
println!(
|
}
|
||||||
"⏳ 等待开发服务器启动... (尝试 {}/{})",
|
|
||||||
attempt, max_retries
|
// 超时检查
|
||||||
|
if start_time.elapsed() > timeout {
|
||||||
|
eprintln!(
|
||||||
|
"✗ 启动超时: Server 未能在 {} 秒内启动",
|
||||||
|
STARTUP_TIMEOUT_SECS
|
||||||
);
|
);
|
||||||
tokio::time::sleep(retry_delay).await;
|
break;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if !server_ready {
|
if !server_ready {
|
||||||
eprintln!("✗ 开发服务器未就绪,请确保运行了 `bun run dev`");
|
eprintln!("✗ Server 启动失败");
|
||||||
std::process::exit(1);
|
std::process::exit(1);
|
||||||
}
|
|
||||||
|
|
||||||
// 创建主窗口
|
|
||||||
let url = format!("http://localhost:{}", DEV_PORT);
|
|
||||||
tauri::WebviewWindowBuilder::new(
|
|
||||||
&app_handle,
|
|
||||||
"main",
|
|
||||||
tauri::WebviewUrl::External(url.parse().unwrap()),
|
|
||||||
)
|
|
||||||
.title(WINDOW_TITLE)
|
|
||||||
.inner_size(DEFAULT_WINDOW_WIDTH, DEFAULT_WINDOW_HEIGHT)
|
|
||||||
.center()
|
|
||||||
.build()
|
|
||||||
.expect("创建窗口失败");
|
|
||||||
} else {
|
|
||||||
// 生产模式:启动 sidecar 二进制
|
|
||||||
println!("🚀 生产模式:启动 Sidecar Server");
|
|
||||||
|
|
||||||
// 查找可用端口
|
|
||||||
let port = find_available_port(DEFAULT_PORT).await;
|
|
||||||
println!("使用端口: {}", port);
|
|
||||||
|
|
||||||
// 启动 sidecar
|
|
||||||
let sidecar = app_handle
|
|
||||||
.shell()
|
|
||||||
.sidecar("server")
|
|
||||||
.expect("无法找到 server")
|
|
||||||
.env("NITRO_PORT", port.to_string());
|
|
||||||
|
|
||||||
let (mut rx, child) = sidecar.spawn().expect("启动 sidecar 失败");
|
|
||||||
|
|
||||||
// 保存进程句柄到全局状态
|
|
||||||
if let Some(state) = app_handle.try_state::<SidecarProcess>() {
|
|
||||||
*state.0.lock().unwrap() = Some(child);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 监听 stdout,等待服务器就绪信号
|
|
||||||
let start_time = std::time::Instant::now();
|
|
||||||
let timeout = Duration::from_secs(STARTUP_TIMEOUT_SECS);
|
|
||||||
let mut server_ready = false;
|
|
||||||
|
|
||||||
while let Some(event) = rx.recv().await {
|
|
||||||
if let CommandEvent::Stdout(line) = event {
|
|
||||||
let output = String::from_utf8_lossy(&line);
|
|
||||||
println!("Server: {}", output);
|
|
||||||
|
|
||||||
// 检测服务器启动成功的标志
|
|
||||||
if output.contains("Listening on:") || output.contains("localhost") {
|
|
||||||
server_ready = true;
|
|
||||||
println!("✓ Server 启动成功!");
|
|
||||||
|
|
||||||
// 创建主窗口
|
|
||||||
let url = format!("http://localhost:{}", port);
|
|
||||||
tauri::WebviewWindowBuilder::new(
|
|
||||||
&app_handle,
|
|
||||||
"main",
|
|
||||||
tauri::WebviewUrl::External(url.parse().unwrap()),
|
|
||||||
)
|
|
||||||
.title(WINDOW_TITLE)
|
|
||||||
.inner_size(DEFAULT_WINDOW_WIDTH, DEFAULT_WINDOW_HEIGHT)
|
|
||||||
.center()
|
|
||||||
.build()
|
|
||||||
.expect("创建窗口失败");
|
|
||||||
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 超时检查
|
|
||||||
if start_time.elapsed() > timeout {
|
|
||||||
eprintln!(
|
|
||||||
"✗ 启动超时: Server 未能在 {} 秒内启动",
|
|
||||||
STARTUP_TIMEOUT_SECS
|
|
||||||
);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !server_ready {
|
|
||||||
eprintln!("✗ Server 启动失败");
|
|
||||||
std::process::exit(1);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user