Skip to content

Commit b572f7f

Browse files
committed
refactor(decompression-plz): renamed try_decompress_chain to try_decompress_chain_first and try_decompress_chain_remaining to try_decompress_chain_continue
1 parent 085aef0 commit b572f7f

File tree

4 files changed

+27
-19
lines changed

4 files changed

+27
-19
lines changed

decompression-plz/src/decompression/dstruct.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ impl<'a> DecompressionStruct<'a> {
120120
let chained =
121121
Cursor::new(self.main).chain(Cursor::new(self.extra.unwrap()));
122122
let len = self.len() as u64;
123-
let result = Self::try_decompress_chain(
123+
let result = Self::try_decompress_chain_first(
124124
chained,
125125
&mut self.writer,
126126
&last_encoding,
@@ -132,14 +132,14 @@ impl<'a> DecompressionStruct<'a> {
132132
}
133133
let output = self.writer.get_mut().split();
134134
if !self.is_encodings_empty() {
135-
self.try_decompress_chain_remaining(output, last_encoding)
135+
self.try_decompress_chain_continue(output, last_encoding)
136136
} else {
137137
self.push_last_encoding(last_encoding);
138138
Ok(output)
139139
}
140140
}
141141

142-
fn try_decompress_chain(
142+
fn try_decompress_chain_first(
143143
mut input: std::io::Chain<Cursor<&[u8]>, Cursor<&[u8]>>,
144144
mut writer: &mut Writer<&mut BytesMut>,
145145
content_encoding: &ContentEncoding,
@@ -162,7 +162,7 @@ impl<'a> DecompressionStruct<'a> {
162162
Ok(())
163163
}
164164

165-
fn try_decompress_chain_remaining(
165+
fn try_decompress_chain_continue(
166166
&mut self,
167167
input: BytesMut,
168168
last_encoding: ContentEncoding,
@@ -468,7 +468,7 @@ mod tests {
468468
}
469469
}
470470

471-
// try_decompress_remaining
471+
// try_decompress_continue
472472
#[track_caller]
473473
fn assert_dstruct_d_main_extra_partial_err(
474474
main: &[u8],

decompression-plz/src/decompression/mod.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
21
pub mod magic_bytes;
32

43
pub mod dstruct;

decompression-plz/src/decompression/multi/mod.rs

Lines changed: 21 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
21
use bytes::{BytesMut, buf::Writer};
32
use header_plz::body_headers::encoding_info::EncodingInfo;
43

@@ -85,11 +84,13 @@ mod tests {
8584
let input = all_compressed_data();
8685
let mut buf = BytesMut::new();
8786
let mut writer = (&mut buf).writer();
88-
let einfo_list = [EncodingInfo::new(0, vec![ContentEncoding::Brotli]),
87+
let einfo_list = [
88+
EncodingInfo::new(0, vec![ContentEncoding::Brotli]),
8989
EncodingInfo::new(1, vec![ContentEncoding::Deflate]),
9090
EncodingInfo::new(2, vec![ContentEncoding::Gzip]),
9191
EncodingInfo::new(3, vec![ContentEncoding::Zstd]),
92-
EncodingInfo::new(4, vec![ContentEncoding::Identity])];
92+
EncodingInfo::new(4, vec![ContentEncoding::Identity]),
93+
];
9394

9495
let result =
9596
decompress_multi(&input, &mut writer, &mut einfo_list.iter())
@@ -102,7 +103,8 @@ mod tests {
102103
let input = all_compressed_data();
103104
let mut buf = BytesMut::new();
104105
let mut writer = (&mut buf).writer();
105-
let einfo_list = [EncodingInfo::new(
106+
let einfo_list = [
107+
EncodingInfo::new(
106108
0,
107109
vec![ContentEncoding::Brotli, ContentEncoding::Deflate],
108110
),
@@ -113,7 +115,8 @@ mod tests {
113115
EncodingInfo::new(
114116
3,
115117
vec![ContentEncoding::Zstd, ContentEncoding::Identity],
116-
)];
118+
),
119+
];
117120

118121
let result =
119122
decompress_multi(&input, &mut writer, &mut einfo_list.iter())
@@ -181,8 +184,10 @@ mod tests {
181184
let input = compress_brotli(INPUT);
182185
let mut buf = BytesMut::new();
183186
let mut writer = (&mut buf).writer();
184-
let einfo_list = [EncodingInfo::new(0, vec![ContentEncoding::Zstd]),
185-
EncodingInfo::new(4, vec![ContentEncoding::Brotli])];
187+
let einfo_list = [
188+
EncodingInfo::new(0, vec![ContentEncoding::Zstd]),
189+
EncodingInfo::new(4, vec![ContentEncoding::Brotli]),
190+
];
186191
let result =
187192
decompress_multi(&input, &mut writer, &mut einfo_list.iter())
188193
.unwrap_err();
@@ -203,12 +208,14 @@ mod tests {
203208
let input = all_compressed_data();
204209
let mut buf = BytesMut::new();
205210
let mut writer = (&mut buf).writer();
206-
let einfo_list = [EncodingInfo::new(0, vec![ContentEncoding::Brotli]),
211+
let einfo_list = [
212+
EncodingInfo::new(0, vec![ContentEncoding::Brotli]),
207213
EncodingInfo::new(1, vec![ContentEncoding::Brotli]),
208214
EncodingInfo::new(2, vec![ContentEncoding::Deflate]),
209215
EncodingInfo::new(3, vec![ContentEncoding::Gzip]),
210216
EncodingInfo::new(4, vec![ContentEncoding::Zstd]),
211-
EncodingInfo::new(5, vec![ContentEncoding::Identity])];
217+
EncodingInfo::new(5, vec![ContentEncoding::Identity]),
218+
];
212219
let result =
213220
decompress_multi(&input, &mut writer, &mut einfo_list.iter())
214221
.unwrap_err();
@@ -228,8 +235,7 @@ mod tests {
228235
fn test_decompress_multi_error_corrupt_single_header() {
229236
let mut buf = BytesMut::new();
230237
let mut writer = (&mut buf).writer();
231-
let einfo_list =
232-
[EncodingInfo::new(0, vec![ContentEncoding::Zstd])];
238+
let einfo_list = [EncodingInfo::new(0, vec![ContentEncoding::Zstd])];
233239
let result =
234240
decompress_multi(INPUT, &mut writer, &mut einfo_list.iter())
235241
.unwrap_err();
@@ -240,8 +246,10 @@ mod tests {
240246
fn test_decompress_multi_error_corrupt_multi_header() {
241247
let mut buf = BytesMut::new();
242248
let mut writer = (&mut buf).writer();
243-
let einfo_list = [EncodingInfo::new(0, vec![ContentEncoding::Zstd]),
244-
EncodingInfo::new(4, vec![ContentEncoding::Brotli])];
249+
let einfo_list = [
250+
EncodingInfo::new(0, vec![ContentEncoding::Zstd]),
251+
EncodingInfo::new(4, vec![ContentEncoding::Brotli]),
252+
];
245253

246254
let result =
247255
decompress_multi(INPUT, &mut writer, &mut einfo_list.iter())

decompression-plz/src/decompression/state.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ impl<'a> DecompressionState<'a> {
144144
main_plus_extra_decompressed,
145145
)
146146
}
147+
// what if partially decompressed
147148
Err(_) => DecompressionState::ExtraRawMainTry(
148149
decompression_struct,
149150
),

0 commit comments

Comments
 (0)