Skip to content

Commit 2b54e48

Browse files
author
kamillecao
committed
fix sender drop.
1 parent bbcb988 commit 2b54e48

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

src/insert.rs

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,16 @@ impl RowsBuilder {
6464
self.take_and_prepare_chunk()
6565
}
6666

67+
#[inline]
68+
pub fn len(&self) -> usize {
69+
self.buffer.len()
70+
}
71+
72+
#[inline]
73+
pub fn is_empty(&self) -> bool {
74+
self.buffer.is_empty()
75+
}
76+
6777
#[cfg(feature = "lz4")]
6878
fn take_and_prepare_chunk(&mut self) -> Result<Bytes> {
6979
Ok(if self.compression.is_lz4() {
@@ -79,16 +89,6 @@ impl RowsBuilder {
7989
fn take_and_prepare_chunk(mut self) -> Result<Bytes> {
8090
Ok(mem::replace(&mut self.buffer, BytesMut::with_capacity(BUFFER_SIZE)).freeze())
8191
}
82-
83-
#[inline]
84-
fn len(&self) -> usize {
85-
self.buffer.len()
86-
}
87-
88-
#[inline]
89-
fn is_empty(&self) -> bool {
90-
self.buffer.is_empty()
91-
}
9292
}
9393

9494
/// inserted rows sender
@@ -303,6 +303,12 @@ impl<T> RowsSender<T> {
303303
}
304304
}
305305

306+
impl<T> Drop for RowsSender<T> {
307+
fn drop(&mut self) {
308+
self.abort();
309+
}
310+
}
311+
306312
pub(crate) enum RowsSenderState {
307313
NotStarted {
308314
client: Box<Client>,
@@ -494,9 +500,3 @@ impl<T> Insert<T> {
494500
self.sender.terminate().await
495501
}
496502
}
497-
498-
impl<T> Drop for Insert<T> {
499-
fn drop(&mut self) {
500-
self.sender.abort();
501-
}
502-
}

0 commit comments

Comments
 (0)