aboutsummaryrefslogtreecommitdiff
path: root/src/api/s3_put.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/api/s3_put.rs')
-rw-r--r--src/api/s3_put.rs24
1 files changed, 14 insertions, 10 deletions
diff --git a/src/api/s3_put.rs b/src/api/s3_put.rs
index e2a1b54d..f5607c9f 100644
--- a/src/api/s3_put.rs
+++ b/src/api/s3_put.rs
@@ -2,8 +2,8 @@ use std::collections::{BTreeMap, VecDeque};
use std::fmt::Write;
use std::sync::Arc;
+use fastcdc::{Chunk, FastCDC};
use futures::stream::*;
-use hash_roll::{ChunkIncr, fastcdc::{FastCdc, FastCdcIncr}, gear_table::GEAR_64};
use hyper::{Body, Request, Response};
use md5::{digest::generic_array::*, Digest as Md5Digest, Md5};
use sha2::Sha256;
@@ -269,22 +269,24 @@ async fn put_block_meta(
struct BodyChunker {
body: Body,
read_all: bool,
+ min_block_size: usize,
+ avg_block_size: usize,
max_block_size: usize,
buf: VecDeque<u8>,
- chunker: FastCdcIncr<'static>,
}
impl BodyChunker {
fn new(body: Body, block_size: usize) -> Self {
+ let min_block_size = block_size / 4 * 3;
+ let avg_block_size = block_size;
let max_block_size = block_size * 2;
- let chunker = FastCdc::new(&GEAR_64, block_size as u64 / 2, block_size as u64, max_block_size as u64);
- let chunker = (&chunker).into();
Self {
body,
read_all: false,
+ min_block_size,
+ avg_block_size,
max_block_size,
buf: VecDeque::with_capacity(2 * max_block_size),
- chunker,
}
}
async fn next(&mut self) -> Result<Option<Vec<u8>>, GarageError> {
@@ -299,12 +301,14 @@ impl BodyChunker {
}
if self.buf.len() == 0 {
Ok(None)
- } else if let Some(index) = self.chunker.push(self.buf.make_contiguous()) {
- let block = self.buf.drain(..index).collect::<Vec<u8>>();
- Ok(Some(block))
} else {
- let block = self.buf.drain(..).collect::<Vec<u8>>();
- Ok(Some(block))
+ let mut iter = FastCDC::with_eof(self.buf.make_contiguous(), self.min_block_size, self.avg_block_size, self.max_block_size, self.read_all);
+ if let Some(Chunk {length, ..}) = iter.next() {
+ let block = self.buf.drain(..length).collect::<Vec<u8>>();
+ Ok(Some(block))
+ } else {
+ Ok(None)
+ }
}
}
}