1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
//! Define an adapter to implement `std::io::Write` on `Sender<Bytes>`.
use actix_web::web::{Bytes, BytesMut};
use futures::channel::mpsc::Sender;
use futures::executor::block_on;
use futures::sink::SinkExt;
use std::io::{Error, ErrorKind, Result, Write};
/// Adapter to implement the `std::io::Write` trait on a `Sender<Bytes>` from a futures channel.
///
/// It uses an intermediate buffer to transfer packets.
pub struct Pipe {
dest: Sender<std::result::Result<Bytes, ()>>,
bytes: BytesMut,
}
impl Pipe {
/// Wrap the given sender in a `Pipe`.
pub fn new(destination: Sender<std::result::Result<Bytes, ()>>) -> Self {
Pipe {
dest: destination,
bytes: BytesMut::new(),
}
}
}
impl Drop for Pipe {
fn drop(&mut self) {
let _ = block_on(self.dest.close());
}
}
impl Write for Pipe {
fn write(&mut self, buf: &[u8]) -> Result<usize> {
// We are given a slice of bytes we do not own, so we must start by copying it.
self.bytes.extend_from_slice(buf);
// Then, take the buffer and send it in the channel.
block_on(self.dest.send(Ok(self.bytes.split().into())))
.map_err(|e| Error::new(ErrorKind::UnexpectedEof, e))?;
// Return how much we sent - all of it.
Ok(buf.len())
}
fn flush(&mut self) -> Result<()> {
block_on(self.dest.flush()).map_err(|e| Error::new(ErrorKind::UnexpectedEof, e))
}
}
|
