Skip to content

Commit 2cabaf4

Browse files
committed
remove some diff
1 parent 7e925b1 commit 2cabaf4

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

src/filters/fb_builder.rs

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -39,29 +39,27 @@ struct NetworkFilterListBuilder {
3939
#[derive(Default)]
4040
struct EngineFlatBuilder<'a> {
4141
fb_builder: flatbuffers::FlatBufferBuilder<'a>,
42-
unique_domains_hashes: HashMap<Hash, u32>,
43-
unique_domains_hashes_vec: Vec<Hash>,
42+
unique_domains_hashes: Vec<Hash>,
43+
unique_domains_hashes_map: HashMap<Hash, u32>,
4444
}
4545

4646
impl<'a> EngineFlatBuilder<'a> {
47-
pub fn get_or_insert_unique_domain_hash(&mut self, hash: &Hash) -> u32 {
48-
if let Some(&index) = self.unique_domains_hashes.get(hash) {
47+
pub fn get_or_insert_unique_domain_hash(&mut self, h: &Hash) -> u32 {
48+
if let Some(&index) = self.unique_domains_hashes_map.get(h) {
4949
return index;
5050
}
51-
let index = self.unique_domains_hashes_vec.len() as u32;
52-
self.unique_domains_hashes_vec.push(*hash);
53-
self.unique_domains_hashes.insert(*hash, index);
51+
let index = self.unique_domains_hashes.len() as u32;
52+
self.unique_domains_hashes.push(*h);
53+
self.unique_domains_hashes_map.insert(*h, index);
5454
index
5555
}
5656

5757
pub fn finish(
5858
&mut self,
5959
network_rules: WIPFlatVec<'a, NetworkFilterListBuilder, EngineFlatBuilder<'a>>,
6060
) -> VerifiedFlatbufferMemory {
61-
let unique_domains_hashes = Some(
62-
self.fb_builder
63-
.create_vector(&self.unique_domains_hashes_vec),
64-
);
61+
let unique_domains_hashes =
62+
Some(self.fb_builder.create_vector(&self.unique_domains_hashes));
6563
let network_rules = Some(network_rules);
6664
let engine = fb::Engine::create(
6765
self.raw_builder(),
@@ -177,15 +175,15 @@ impl NetworkFilterListBuilder {
177175
impl<'a> FlatSerialize<'a, EngineFlatBuilder<'a>> for NetworkFilterListBuilder {
178176
type Output = WIPOffset<fb::NetworkFilterList<'a>>;
179177
fn serialize(
180-
value: Self,
178+
rule_list: Self,
181179
builder: &mut EngineFlatBuilder<'a>,
182180
) -> WIPOffset<fb::NetworkFilterList<'a>> {
183181
let mut filter_map = HashMap::<ShortHash, Vec<WIPOffset<fb::NetworkFilter<'a>>>>::new();
184182

185183
let mut optimizable = HashMap::<ShortHash, Vec<NetworkFilter>>::new();
186184

187185
// Compute tokens for all filters
188-
let filter_tokens: Vec<_> = value
186+
let filter_tokens: Vec<_> = rule_list
189187
.filters
190188
.into_iter()
191189
.map(|filter| {
@@ -199,7 +197,7 @@ impl<'a> FlatSerialize<'a, EngineFlatBuilder<'a>> for NetworkFilterListBuilder {
199197

200198
{
201199
for (network_filter, multi_tokens) in filter_tokens.into_iter() {
202-
let flat_filter = if !value.optimize
200+
let flat_filter = if !rule_list.optimize
203201
|| !optimizer::is_filter_optimizable_by_patterns(&network_filter)
204202
{
205203
Some(FlatSerialize::serialize(&network_filter, builder))
@@ -237,7 +235,7 @@ impl<'a> FlatSerialize<'a, EngineFlatBuilder<'a>> for NetworkFilterListBuilder {
237235
}
238236
}
239237

240-
if value.optimize {
238+
if rule_list.optimize {
241239
// Sort the entries to ensure deterministic iteration order
242240
let mut optimizable_entries: Vec<_> = optimizable.drain().collect();
243241
optimizable_entries.sort_unstable_by_key(|(token, _)| *token);
@@ -284,9 +282,9 @@ impl NetworkRulesBuilder {
284282

285283
// Collect badfilter ids in advance.
286284
for filter in network_filters.iter() {
287-
if filter.is_badfilter() {
288-
badfilter_ids.insert(filter.get_id_without_badfilter());
289-
}
285+
if filter.is_badfilter() {
286+
badfilter_ids.insert(filter.get_id_without_badfilter());
287+
}
290288
}
291289

292290
for filter in network_filters.into_iter() {

0 commit comments

Comments
 (0)