2
0
mirror of https://github.com/jedisct1/encrypted-dns-server synced 2024-11-10 19:10:25 +00:00

Blacklist: use FxHashMap

This commit is contained in:
Frank Denis 2019-12-01 01:46:23 +01:00
parent df26dddb86
commit 5e0f4a6223
2 changed files with 5 additions and 4 deletions

View File

@ -30,6 +30,7 @@ net2 = "0.2.33"
parking_lot = "0.9.0"
privdrop = "0.3.3"
rand = "0.7.2"
rustc-hash = "1"
serde = "1.0.102"
serde_derive = "1.0.102"
serde-big-array = "0.2.0"

View File

@ -1,6 +1,6 @@
use crate::errors::*;
use std::collections::HashMap;
use rustc_hash::FxHashMap;
use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::Path;
@ -10,7 +10,7 @@ const MAX_ITERATIONS: usize = 5;
#[derive(Debug)]
struct BlackListInner {
map: HashMap<Vec<u8>, ()>,
map: FxHashMap<Vec<u8>, ()>,
}
#[derive(Clone, Debug)]
@ -20,7 +20,7 @@ pub struct BlackList {
}
impl BlackList {
pub fn new(map: HashMap<Vec<u8>, ()>, max_iterations: usize) -> Self {
pub fn new(map: FxHashMap<Vec<u8>, ()>, max_iterations: usize) -> Self {
let inner = Arc::new(BlackListInner { map });
BlackList {
inner,
@ -29,7 +29,7 @@ impl BlackList {
}
pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, Error> {
let mut map = HashMap::new();
let mut map = FxHashMap::default();
let fp = BufReader::new(File::open(path)?);
for (line_nb, line) in fp.lines().enumerate() {
let line = line?;