Skip to content

Commit c9ff7a7

Browse files
author
Devdutt Shenoi
committed
refactor: simplify in-mem threshold handling
1 parent 7b8dca1 commit c9ff7a7

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

src/parseable/staging/writer.rs

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -201,20 +201,15 @@ impl<const N: usize> MutableBuffer<N> {
201201
fn push(&mut self, rb: &RecordBatch) -> Option<Vec<RecordBatch>> {
202202
if self.inner.len() + rb.num_rows() >= N {
203203
let left = N - self.inner.len();
204-
let right = rb.num_rows() - left;
205204
let left_slice = rb.slice(0, left);
206-
let right_slice = if left < rb.num_rows() {
207-
Some(rb.slice(left, right))
208-
} else {
209-
None
210-
};
211205
self.inner.push(left_slice);
212206
// take all records
213207
let src = Vec::with_capacity(self.inner.len());
214208
let inner = std::mem::replace(&mut self.inner, src);
215209

216-
if let Some(right_slice) = right_slice {
217-
self.inner.push(right_slice);
210+
if left < rb.num_rows() {
211+
let right = rb.num_rows() - left;
212+
self.inner.push(rb.slice(left, right));
218213
}
219214

220215
Some(inner)

0 commit comments

Comments
 (0)