ZengQingming/JoyD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

首次连接回调

Browse Source
This commit is contained in:
zqm
2026-04-07 15:44:30 +08:00
parent 6cbd9e1e30
commit 7a832698c2
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Windows/Rust/CubeLib/src/lib.rs
View File

@@ -30,4 +30,4 @@
pub mod websocket; pub mod websocket;
pub use websocket::{WebSocketClient, WebSocketConfig, WebSocketMessage}; pub use websocket::{WebSocketClient, WebSocketConfig, WebSocketMessage, OutgoingMessage, ConnectionStatus};

10
Windows/Rust/CubeLib/src/websocket/client.rs
View File

@@ -2,6 +2,7 @@
use crate::websocket::{WebSocketConfig, WebSocketMessage, WebSocketError}; use crate::websocket::{WebSocketConfig, WebSocketMessage, WebSocketError};
use futures_util::{SinkExt, StreamExt}; use futures_util::{SinkExt, StreamExt};
use std::pin::Pin;
use std::sync::Arc; use std::sync::Arc;
use tokio::sync::{mpsc, Mutex}; use tokio::sync::{mpsc, Mutex};
use tokio::time::Duration; use tokio::time::Duration;
@@ -60,7 +61,8 @@ pub type SentCallback = Arc<dyn Fn(String, Value) + Send + Sync>;
pub type ReconnectingCallback = Arc<dyn Fn(u32, Arc<tokio::sync::Mutex<String>>) + Send + Sync>; pub type ReconnectingCallback = Arc<dyn Fn(u32, Arc<tokio::sync::Mutex<String>>) + Send + Sync>;
/// Callback triggered on first successful connection (before any reconnect) /// Callback triggered on first successful connection (before any reconnect)
/// Arguments: (url, send_fn) - send_fn can be called to send messages /// Arguments: (url, send_fn) - send_fn can be called to send messages
pub type FirstConnectCallback = Arc<dyn Fn(String, Arc<Mutex<Option<mpsc::Sender<OutgoingMessage>>>>) + Send + Sync>; /// Note: This is an async callback - return a boxed Future
pub type FirstConnectCallback = Arc<dyn Fn(String, Arc<Mutex<Option<mpsc::Sender<OutgoingMessage>>>>) -> Pin<Box<dyn std::future::Future<Output = ()> + Send + Sync + 'static>> + Send + Sync>;
/// WebSocket client with event-driven architecture /// WebSocket client with event-driven architecture
pub struct WebSocketClient { pub struct WebSocketClient {
@@ -198,10 +200,10 @@ impl WebSocketClient {
} }
/// Set callback for first successful connection (before any reconnect) /// Set callback for first successful connection (before any reconnect)
/// This is called only on the first connection, allowing the app to send initial messages /// This is an async callback - the returned Future will be awaited
pub fn on_first_connect<F>(&mut self, callback: F) -> &mut Self pub fn on_first_connect<F>(&mut self, callback: F) -> &mut Self
where where
F: Fn(String, Arc<Mutex<Option<mpsc::Sender<OutgoingMessage>>>>) + Send + Sync + 'static, F: Fn(String, Arc<Mutex<Option<mpsc::Sender<OutgoingMessage>>>>) -> Pin<Box<dyn std::future::Future<Output = ()> + Send + Sync + 'static>> + Send + Sync + 'static,
{ {
self.on_first_connect = Some(Arc::new(callback)); self.on_first_connect = Some(Arc::new(callback));
self self
@@ -468,7 +470,7 @@ impl WebSocketClient {
// Call on_first_connect callback (only on first connection, not on reconnect) // Call on_first_connect callback (only on first connection, not on reconnect)
if is_first { if is_first {
if let Some(ref callback) = on_first_connect { if let Some(ref callback) = on_first_connect {
callback(url.to_string(), sender); callback(url.to_string(), sender).await;
} }
} }