Comments (2)
You have bugs in the code you posted, please read https://docs.rs/mio/latest/mio/struct.Poll.html#draining-readiness.
// [dependencies] // mio = { version = "0.8", features = ["net", "os-ext", "os-poll"] } use mio::{ event::Event, net::{TcpListener, TcpStream}, Events, Interest, Poll, Token, }; use std::{cell::RefCell, collections::HashMap, io::Read, net::SocketAddr, rc::Rc}; const SERVER: Token = Token(0); struct MyServer { listener: TcpListener, connections: HashMap<Token, TcpStream>, event_loop: Rc<RefCell<Poll>>, last_token: usize, } impl MyServer { fn new(listener: TcpListener, event_loop: Rc<RefCell<Poll>>) -> MyServer { MyServer { listener, connections: HashMap::new(), event_loop, last_token: SERVER.0, } } fn new_token(&mut self) -> Token { self.last_token += 1; Token(self.last_token) } fn handle_event(&mut self, event: &Event) -> Result<(), Box<dyn std::error::Error>> { match event.token() { SERVER => { // 有新連接進來 match self.listener.accept() {
This needs a loop around accept
.
Ok((mut stream, _)) => { let token = self.new_token(); println!("New connection from: {}", stream.peer_addr().unwrap()); self.event_loop .borrow_mut() .registry() .register(&mut stream, token, Interest::READABLE) .unwrap(); self.connections.insert(token, stream); return Ok(()); } Err(ref e) if e.kind() == std::io::ErrorKind::WouldBlock => return Ok(()), Err(e) => return Err(e.into()), } } token => { // 有數據可讀 let mut buffer = [0; 1024]; match self.connections.get_mut(&token).unwrap().read(&mut buffer) {
This needs a loop around read
.
Ok(0) => { // 客戶端斷開連接 println!( "Connection closed: {:?}", self.connections[&token].peer_addr().unwrap() ); let mut stream = self.connections.remove(&token).unwrap(); self.event_loop .borrow_mut() .registry() .deregister(&mut stream) .unwrap(); return Ok(()); } Ok(n) => { // 輸出收到的數據 let data = String::from_utf8_lossy(&buffer[..n]); println!( "Received data from {}: {}", self.connections[&token].peer_addr().unwrap(), data ); return Ok(()); } Err(ref e) if e.kind() == std::io::ErrorKind::WouldBlock => return Ok(()), Err(e) => return Err(e.into()), } } } }
}
from mio.
Now it's work fine. Thanks.
// [dependencies]
// mio = { version = "0.8", features = ["net", "os-ext", "os-poll"] }
use mio::{
event::Event,
net::{TcpListener, TcpStream},
Events, Interest, Poll, Token,
};
use std::{cell::RefCell, collections::HashMap, io::Read, net::SocketAddr, rc::Rc};
const SERVER: Token = Token(0);
struct MyServer {
listener: TcpListener,
connections: HashMap<Token, TcpStream>,
event_loop: Rc<RefCell<Poll>>,
last_token: usize,
}
#[derive(Debug)]
pub struct SErr(pub &'static str);
impl std::fmt::Display for SErr {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
std::fmt::Display::fmt(self.0, f)
}
}
impl std::error::Error for SErr {}
impl MyServer {
fn new(listener: TcpListener, event_loop: Rc<RefCell<Poll>>) -> MyServer {
MyServer {
listener,
connections: HashMap::new(),
event_loop,
last_token: SERVER.0,
}
}
fn new_token(&mut self) -> Token {
self.last_token += 1;
Token(self.last_token)
}
fn handle_event(&mut self, event: &Event) -> Result<(), Box<dyn std::error::Error>> {
match event.token() {
SERVER => {
// 有新連接進來
loop {
match self.listener.accept() {
Ok((mut stream, _)) => {
let token = self.new_token();
println!("New connection from: {}", stream.peer_addr()?);
self.event_loop
.borrow_mut()
.registry()
.register(&mut stream, token, Interest::READABLE)?;
self.connections.insert(token, stream);
}
Err(ref e) if e.kind() == std::io::ErrorKind::WouldBlock => return Ok(()),
Err(e) => return Err(e.into()),
}
}
}
token => {
// 有數據可讀
let mut buffer = [0; 1024];
loop {
match self.connections.get_mut(&token).ok_or(SErr("error"))?.read(&mut buffer) {
Ok(0) => {
// 客戶端斷開連接
println!("Connection closed: {:?}", self.connections[&token].peer_addr()?);
let mut stream = self.connections.remove(&token).ok_or(SErr("Connection not found"))?;
self.event_loop.borrow_mut().registry().deregister(&mut stream)?;
return Ok(());
}
Ok(n) => {
// 輸出收到的數據
let data = String::from_utf8_lossy(&buffer[..n]);
println!("Received data from {}: {}", self.connections[&token].peer_addr()?, data);
}
Err(ref e) if e.kind() == std::io::ErrorKind::WouldBlock => return Ok(()),
Err(e) => return Err(e.into()),
}
}
}
}
}
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
// 建立事件循環
let event_loop = Rc::new(RefCell::new(Poll::new()?));
// 建立 TCP 監聽器
let address = "127.0.0.1:8080".parse::<SocketAddr>()?;
let mut listener = TcpListener::bind(address)?;
event_loop
.borrow_mut()
.registry()
.register(&mut listener, SERVER, Interest::READABLE)?;
let mut my_server = MyServer::new(listener, event_loop.clone());
println!("Server listening on {}", address);
let mut events = Events::with_capacity(1024);
loop {
if let Err(err) = event_loop.borrow_mut().poll(&mut events, None) {
if err.kind() == std::io::ErrorKind::Interrupted {
continue;
}
return Err(err.into());
}
println!("new event");
for event in &events {
my_server.handle_event(event)?;
}
}
}
from mio.
Related Issues (20)
- Windows named pipes hits unreachable code during read operation HOT 1
- Can TcpStream be read from a different thread? HOT 1
- Exposing `AsRawHandle` on `Registry` HOT 5
- build failed with wasm-pack HOT 1
- Wait for messages in the error queue HOT 6
- Setup WASM/WASI CI
- get any errors when i use `wasm-pack` command by `mio` HOT 4
- Need switch to not use eventfd on `target_os = "linux"` HOT 4
- Add ESP-IDF framework support
- IoRing support for windows HOT 3
- Has waker check doesn't work with try_cloned Registries HOT 1
- compile time error inside `poll.rs` HOT 3
- poll: Optimize locks and allocations in the critical path
- Poll(2) implementation follow-ups
- Official ESP-IDF framework support HOT 3
- Make `CompletionPort` public? HOT 7
- No data coming from Poll HOT 8
- How to wait for a socket to be writable? udp / Interest::READABLE | Interest::WRITABLE HOT 6
- could not compile mio due to 44 previous errors HOT 3
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from mio.