feat: notify the client when server is ready to accept connections

vyfor · Dec 6, 2024 · b27d763 · b27d763
1 parent 6313f45
commit b27d763
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/ipc/pipe/platform/unix/server.rs b/src/ipc/pipe/platform/unix/server.rs
@@ -55,9 +55,15 @@ impl PipeServerImpl for PipeServer {
         let next_client_id = Arc::clone(&self.next_client_id);
         let running = Arc::clone(&self.running);
         let listener = self.listener.as_ref().unwrap().try_clone()?;
+        let notified = Arc::new(AtomicBool::new(false));
 
         self.thread_handle = Some(std::thread::spawn(move || {
             while running.load(Ordering::SeqCst) {
+                if !notified.load(Ordering::SeqCst) {
+                    println!("Ready");
+                    notified.store(true, Ordering::SeqCst);
+                }
+
                 match listener.accept() {
                     Ok((stream, _)) => {
                         let client_id = next_client_id.fetch_add(1, Ordering::SeqCst);

diff --git a/src/ipc/pipe/platform/windows/server.rs b/src/ipc/pipe/platform/windows/server.rs
@@ -54,10 +54,16 @@ impl PipeServerImpl for PipeServer {
         let next_client_id = Arc::clone(&self.next_client_id);
         let running = Arc::clone(&self.running);
         let tx = self.tx.clone();
+        let notified = Arc::new(AtomicBool::new(false));
 
         self.thread_handle = Some(std::thread::spawn(move || {
             while running.load(Ordering::SeqCst) {
                 if let Ok(handle) = PipeServer::create_pipe_instance(&pipe_name) {
+                    if !notified.load(Ordering::SeqCst) {
+                        println!("Ready");
+                        notified.store(true, Ordering::SeqCst);
+                    }
+
                     unsafe {
                         let h_event =
                             CreateEventW(std::ptr::null_mut(), 1, 0, std::ptr::null_mut());