Struct WebSocketRuntime 

pub struct WebSocketRuntime { /* private fields */ }

WebSocket 运行时

封装 sz_orm_websocket::WsServer，提供服务端 lifecycle 管理。

设计

• 默认使用 DefaultWebSocketHandler（echo 模式），可替换为自定义 handler
• start 方法 spawn 后台任务，返回 JoinHandle<()>
• stop 方法调用 WsServer::stop()，触发 oneshot shutdown
• 监听 CancellationToken：收到信号后自动调用 WsServer::stop()

用法

use sz_rust_core::runtime::websocket::{WebSocketRuntime, WebSocketRuntimeConfig};
use tokio_util::sync::CancellationToken;

let runtime = WebSocketRuntime::new(WebSocketRuntimeConfig::new("0.0.0.0:2346"));
let token = CancellationToken::new();
let handle = runtime.start(token.clone());
// ... 业务运行 ...
token.cancel();
let _ = handle.await;

Implementations

impl WebSocketRuntime

pub fn new(config: WebSocketRuntimeConfig) -> Self

创建 WebSocket 运行时，使用默认 handler

pub fn start(&self, token: CancellationToken) -> JoinHandle<Result<(), WsError>>

启动 WebSocket 服务（返回 JoinHandle，调用方持有）

• 使用 DefaultWebSocketHandler 处理连接
• 监听 token.cancelled()，收到信号后调用 server.stop()

pub fn start_with_handler( &self, handler: Arc<dyn WebSocketHandler>, token: CancellationToken, ) -> JoinHandle<Result<(), WsError>>

使用自定义 handler 启动 WebSocket 服务

pub async fn stop(&self) -> Result<(), WsError>

手动停止服务

pub async fn connection_count(&self) -> usize

获取当前连接数

pub async fn broadcast_to_all(&self, data: Vec<u8>) -> Result<usize, WsError>

广播消息到所有连接

pub async fn is_running(&self) -> bool

是否仍在运行

pub fn config(&self) -> &WebSocketRuntimeConfig

获取配置