arc-reactor 0.2.4

Minimal, Asynchronous, Web Framework
Documentation 
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

use futures::prelude::*;
use hyper::Chunk;
use tokio::{
	fs::File,
	io::{AsyncRead, Error},
};
/// wraps a tokio::fs::File as a futures::Stream
/// will produce an error if this stream isn't polled in the context of a tokio
/// executor
struct FileStream {
	file: File,
	buf: [u8; 4096],
}

pub fn stream(file: File) -> impl Stream<Item = Chunk, Error = Error> {
	FileStream::new(file)
}

impl FileStream {
	fn new(file: File) -> Self {
		Self {
			file,
			buf: [0u8; 4096],
		}
	}
}

impl Stream for FileStream {
	type Item = Chunk;
	type Error = Error;

	fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
		let n_bytes = try_ready!(self.file.poll_read(&mut self.buf));
		if n_bytes > 0 {
			Ok(Async::Ready(Some(Chunk::from(self.buf.to_vec()))))
		} else {
			Ok(Async::Ready(None))
		}
	}
}