imbytecat/fullstack-starter
1
1
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 优化服务器启动逻辑并移除冗余端口检测

Browse Source 
- 移除冗余的端口占用检测函数并优化开发与生产模式下的服务器启动逻辑，提升代码可读性与启动可靠性。
This commit is contained in:
imbytecat
2026-01-18 15:37:26 +08:00
parent a30d7c32fd
commit fc73243687
1 changed files with 79 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

196
src-tauri/src/sidecar.rs
View File

@@ -13,9 +13,6 @@ const STARTUP_TIMEOUT_SECS: u64 = 5;
/// 默认起始端口
const DEFAULT_PORT: u16 = 3000;
/// 开发模式使用的端口
const DEV_PORT: u16 = 3000;
/// 端口扫描范围（从起始端口开始扫描的端口数量）
const PORT_SCAN_RANGE: u16 = 100;
@@ -40,19 +37,6 @@ async fn is_port_available(port: u16) -> bool {
.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 {
for port in start..start + PORT_SCAN_RANGE {
@@ -65,117 +49,95 @@ async fn find_available_port(start: u16) -> u16 {
/// 启动 Sidecar 进程并创建主窗口
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 {
// 检测是否为开发模式
let is_dev = cfg!(debug_assertions);
println!("🚀 生产模式：启动 Sidecar Server");
if is_dev {
// 开发模式：直接连接到 localhost:3000
println!("🔧 开发模式：使用本地开发服务器 (localhost:{})", DEV_PORT);
// 查找可用端口
let port = find_available_port(DEFAULT_PORT).await;
println!("使用端口: {}", port);
// 等待开发服务器就绪（可选：添加重试逻辑）
let max_retries = 10;
let retry_delay = Duration::from_millis(500);
let mut server_ready = false;
// 启动 sidecar
let sidecar = app_handle
.shell()
.sidecar("server")
.expect("无法找到 server")
.env("NITRO_PORT", port.to_string());
for attempt in 1..=max_retries {
if is_port_in_use(DEV_PORT).await {
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!("✓ 开发服务器已就绪 (端口 {})", 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;
}
println!(
"⏳ 等待开发服务器启动... (尝试 {}/{})",
attempt, max_retries
}
// 超时检查
if start_time.elapsed() > timeout {
eprintln!(
"✗ 启动超时: Server 未能在 {} 秒内启动",
STARTUP_TIMEOUT_SECS
);
tokio::time::sleep(retry_delay).await;
break;
}
}
if !server_ready {
eprintln!("✗ 开发服务器未就绪，请确保运行了 `bun run dev`");
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);
}
if !server_ready {
eprintln!("✗ Server 启动失败");
std::process::exit(1);
}
});
}