Do simple reads from disk
This commit is contained in:
@@ -1,3 +1,5 @@
|
||||
#[allow(unused_imports)]
|
||||
use tracing::{error, warn, info, debug, trace};
|
||||
use err::Error;
|
||||
use std::task::{Context, Poll};
|
||||
use std::pin::Pin;
|
||||
@@ -11,28 +13,41 @@ pub async fn read_test_1() -> Result<netpod::BodyStream, Error> {
|
||||
.read(true)
|
||||
.open(path)
|
||||
.await?;
|
||||
let meta = fin.metadata().await;
|
||||
debug!("file meta {:?}", meta);
|
||||
let stream = netpod::BodyStream {
|
||||
inner: Box::new(FileReader { file: fin }),
|
||||
inner: Box::new(FileReader {
|
||||
file: fin,
|
||||
nreads: 0,
|
||||
}),
|
||||
};
|
||||
Ok(stream)
|
||||
}
|
||||
|
||||
struct FileReader {
|
||||
file: tokio::fs::File,
|
||||
nreads: u32,
|
||||
}
|
||||
|
||||
impl futures_core::Stream for FileReader {
|
||||
type Item = Result<bytes::Bytes, Error>;
|
||||
|
||||
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
|
||||
let mut buf2 = bytes::BytesMut::with_capacity(13);
|
||||
if buf2.as_mut().len() != 13 {
|
||||
if self.nreads >= 10 {
|
||||
return Poll::Ready(None);
|
||||
}
|
||||
let blen = 13;
|
||||
let mut buf2 = bytes::BytesMut::with_capacity(blen);
|
||||
buf2.resize(buf2.capacity(), 0);
|
||||
if buf2.as_mut().len() != blen {
|
||||
panic!("todo prepare slice");
|
||||
}
|
||||
let mut buf = tokio::io::ReadBuf::new(buf2.as_mut());
|
||||
let g = Pin::new(&mut self.file).poll_read(cx, &mut buf);
|
||||
match g {
|
||||
match Pin::new(&mut self.file).poll_read(cx, &mut buf) {
|
||||
Poll::Ready(Ok(_)) => {
|
||||
info!("read from disk: {} nreads {}", buf.filled().len(), self.nreads);
|
||||
info!("buf2 len: {}", buf2.len());
|
||||
self.nreads += 1;
|
||||
Poll::Ready(Some(Ok(buf2.freeze())))
|
||||
}
|
||||
Poll::Ready(Err(e)) => {
|
||||
|
||||
Reference in New Issue
Block a user