address pr feedback

This commit is contained in:
Tommy Verrall
2023-10-26 10:46:09 +02:00
parent 8139fcfe74
commit 76d4036883
2 changed files with 21 additions and 8 deletions
+12 -6
View File
@@ -107,18 +107,24 @@ impl GeoIp {
Ok(ip)
} else {
debug!("Fail to resolve IP address from {}:{}", &address, p);
let mut failed_ips_guard = self.failed_addresses.failed_ips.lock().unwrap();
if failed_ips_guard.insert(address.to_string()) {
append_ip_to_file(address);
if let Ok(mut failed_ips_guard) = self.failed_addresses.failed_ips.lock() {
if failed_ips_guard.insert(address.to_string()) {
append_ip_to_file(address);
}
} else {
error!("Failed to acquire lock on failed_ips");
}
Err(GeoIpError::NoValidIP)
}
}
Err(_) => {
debug!("Fail to resolve IP address from {}:{}.", &address, p);
let mut failed_ips_guard = self.failed_addresses.failed_ips.lock().unwrap();
if failed_ips_guard.insert(address.to_string()) {
append_ip_to_file(address);
if let Ok(mut failed_ips_guard) = self.failed_addresses.failed_ips.lock() {
if failed_ips_guard.insert(address.to_string()) {
append_ip_to_file(address);
}
} else {
error!("Failed to acquire lock on failed_ips");
}
Err(GeoIpError::NoValidIP)
}
+9 -2
View File
@@ -18,11 +18,18 @@ pub fn failed_ips_filepath() -> String {
}
pub fn append_ip_to_file(address: &str) {
if let Ok(mut file) = OpenOptions::new()
match OpenOptions::new()
.append(true)
.create(true)
.open(failed_ips_filepath())
{
writeln!(file, "{}", address).expect("Failed to write to file");
Ok(mut file) => {
if let Err(e) = writeln!(file, "{}", address) {
error!("Failed to write to file: {}", e);
}
}
Err(e) => {
error!("Failed to open or create the file: {}", e);
}
}
}