use std::io;
use futures::{Async, Poll};
use mio_crate as mio;
use tokio_reactor::PollEvented;
use super::sys;
use super::Protocol;
pub struct TokioSocket(PollEvented<sys::Socket>);
impl TokioSocket {
pub fn bind(&mut self, addr: &sys::SocketAddr) -> io::Result<()> {
self.0.get_mut().bind(addr)
}
pub fn bind_auto(&mut self) -> io::Result<sys::SocketAddr> {
self.0.get_mut().bind_auto()
}
pub fn new(protocol: Protocol) -> io::Result<Self> {
let socket = sys::Socket::new(protocol)?;
socket.set_non_blocking(true)?;
Ok(TokioSocket(PollEvented::new(socket)))
}
pub fn connect(&self, addr: &sys::SocketAddr) -> io::Result<()> {
self.0.get_ref().connect(addr)
}
pub fn poll_send(&mut self, buf: &[u8]) -> Poll<usize, io::Error> {
try_ready!(self.0.poll_write_ready());
match self.0.get_ref().send(buf, 0) {
Ok(n) => Ok(Async::Ready(n)),
Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
self.0.clear_write_ready()?;
Ok(Async::NotReady)
}
Err(e) => Err(e),
}
}
pub fn poll_recv(&mut self, buf: &mut [u8]) -> Poll<usize, io::Error> {
try_ready!(self.0.poll_read_ready(mio::Ready::readable()));
match self.0.get_ref().recv(buf, 0) {
Ok(n) => Ok(Async::Ready(n)),
Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
self.0.clear_read_ready(mio::Ready::readable())?;
Ok(Async::NotReady)
}
Err(e) => Err(e),
}
}
pub fn poll_send_to(&mut self, buf: &[u8], target: &sys::SocketAddr) -> Poll<usize, io::Error> {
try_ready!(self.0.poll_write_ready());
match self.0.get_ref().send_to(buf, target, 0) {
Ok(n) => Ok(Async::Ready(n)),
Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
self.0.clear_write_ready()?;
Ok(Async::NotReady)
}
Err(e) => Err(e),
}
}
pub fn poll_recv_from(&mut self, buf: &mut [u8]) -> Poll<(usize, sys::SocketAddr), io::Error> {
try_ready!(self.0.poll_read_ready(mio::Ready::readable()));
match self.0.get_ref().recv_from(buf, 0) {
Ok(n) => Ok(Async::Ready(n)),
Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
self.0.clear_read_ready(mio::Ready::readable())?;
Ok(Async::NotReady)
}
Err(e) => Err(e),
}
}
}