我想做消息广播:当其中一个客户端发送消息时,服务器将它写入每个套接字。我的主要问题是我无法弄清楚如何发送Vec
到线程。我不能使用Mutex
,因为这会将其他线程的访问锁定到Vec
进行读取。我无法克隆和发送,因为无法克隆和发送TcpStream
。这里是我的尝试到现在为止只要其中一个读取新输入,我可以维护一个TcpStream的Vec来跨线程写入它们吗?
use std::net::{TcpStream, TcpListener};
use std::io::prelude::*;
use std::sync::{Arc, Mutex};
use std::thread;
use std::sync::mpsc::{channel, Receiver};
use std::cell::RefCell;
type StreamSet = Arc<RefCell<Vec<TcpStream>>>;
type StreamReceiver = Arc<Mutex<Receiver<StreamSet>>>;
fn main() {
let listener = TcpListener::bind("0.0.0.0:8000").unwrap();
let mut connection_set: StreamSet = Arc::new(RefCell::new(vec![]));
let mut id = 0;
let (tx, rx) = channel();
let rx = Arc::new(Mutex::new(rx));
for stream in listener.incoming() {
let receiver = rx.clone();
let stream = stream.unwrap();
(*connection_set).borrow_mut().push(stream);
println!("A connection established with client {}", id);
thread::spawn(move || handle_connection(receiver, id));
id += 1;
tx.send(connection_set.clone()).unwrap();
}
}
fn handle_connection(rx: StreamReceiver, id: usize) {
let streams;
{
streams = *(rx.lock().unwrap().recv().unwrap()).borrow();
}
let mut connection = &streams[id];
loop {
let mut buffer = [0; 512];
if let Err(_) = connection.read(&mut buffer) {
break;
};
println!("Request: {}", String::from_utf8_lossy(&buffer[..]));
if let Err(_) = connection.write(&buffer[..]) {
break;
};
if let Err(_) = connection.flush() {
break;
};
}
}
啊谢谢。我实际上对编码比较陌生,将在下个月完成一年。所以我试图在使用像tokio这样的抽象之前尽我所能地自己实现它,所以我明白了底下发生了什么。我将尝试实施控制器工作者模型。每个单独的线程是否有单独的通道,或者它们是否使用相同的接收器,发送器对? – codeNoob
我期望控制器线程为每个工作线程有一个唯一的通道向它们发送数据,然后是从每个工作人员返回到控制器的重用通道(因为它们是** m ** ultiple ** p ** roducer ,** s ** ingle ** c ** onsumer频道)。总共应该有N + 1个通道。 – Shepmaster
啊好的。说得通。谢谢。 – codeNoob