specify the address to send

mzaryabrafique · mzaryabrafique · commit 90fb3be592af · 2022-08-28T20:13:49.000+05:00
diff --git a/src/main.rs b/src/main.rs
@@ -8,10 +8,10 @@ use tokio::{
 async fn main() {
     let listener = TcpListener::bind("localhost:8080").await.unwrap();
 
-    let (tx, _rx) = broadcast::channel::<String>(10);
+    let (tx, _rx) = broadcast::channel(10);
 
     loop {
-        let (mut socket, _addr) = listener.accept().await.unwrap();
+        let (mut socket, addr) = listener.accept().await.unwrap();
 
         let tx = tx.clone();
         let mut rx = tx.subscribe();
@@ -29,14 +29,17 @@ async fn main() {
                             break;
                         }
 
-                        tx.send(line.clone()).unwrap();
+                        tx.send((line.clone(), addr)).unwrap();
                         line.clear();
                     }
                     result = rx.recv() => {
-                        let msg = result.unwrap();
+                        let (msg, other_addr) = result.unwrap();
+
+                        if addr != other_addr{
+                            // write every single byte in the buffer
+                            writer.write_all(msg.as_bytes()).await.unwrap();
+                        }
 
-                        // write every single byte in the buffer
-                        writer.write_all(msg.as_bytes()).await.unwrap();
                     }
                 }
             }

Original file line number	Diff line number	Diff line change
`@@ -8,10 +8,10 @@ use tokio::{`
`8`	`8`	`async fn main() {`
`9`	`9`	`let listener = TcpListener::bind("localhost:8080").await.unwrap();`
`10`	`10`
`11`		`- let (tx, _rx) = broadcast::channel::<String>(10);`
	`11`	`+ let (tx, _rx) = broadcast::channel(10);`
`12`	`12`
`13`	`13`	`loop {`
`14`		`- let (mut socket, _addr) = listener.accept().await.unwrap();`
	`14`	`+ let (mut socket, addr) = listener.accept().await.unwrap();`
`15`	`15`
`16`	`16`	`let tx = tx.clone();`
`17`	`17`	`let mut rx = tx.subscribe();`
`@@ -29,14 +29,17 @@ async fn main() {`
`29`	`29`	`break;`
`30`	`30`	`}`
`31`	`31`
`32`		`- tx.send(line.clone()).unwrap();`
	`32`	`+ tx.send((line.clone(), addr)).unwrap();`
`33`	`33`	`line.clear();`
`34`	`34`	`}`
`35`	`35`	`result = rx.recv() => {`
`36`		`- let msg = result.unwrap();`
	`36`	`+ let (msg, other_addr) = result.unwrap();`
	`37`	`+`
	`38`	`+ if addr != other_addr{`
	`39`	`+ // write every single byte in the buffer`
	`40`	`+ writer.write_all(msg.as_bytes()).await.unwrap();`
	`41`	`+ }`
`37`	`42`
`38`		`- // write every single byte in the buffer`
`39`		`- writer.write_all(msg.as_bytes()).await.unwrap();`
`40`	`43`	`}`
`41`	`44`	`}`
`42`	`45`	`}`