Clean accounts path on validator startup (#10929)
This commit is contained in:
@ -199,6 +199,10 @@ impl Validator {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for accounts_path in &config.account_paths {
|
||||||
|
cleanup_accounts_path(accounts_path);
|
||||||
|
}
|
||||||
|
|
||||||
let (
|
let (
|
||||||
genesis_config,
|
genesis_config,
|
||||||
bank_forks,
|
bank_forks,
|
||||||
@ -959,6 +963,33 @@ fn get_stake_percent_in_gossip(bank: &Bank, cluster_info: &ClusterInfo, log: boo
|
|||||||
online_stake * 100 / total_activated_stake
|
online_stake * 100 / total_activated_stake
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn cleanup_accounts_dir_entry(path: std::fs::DirEntry, accounts_file_regex: ®ex::Regex) {
|
||||||
|
if let Ok(file_type) = path.file_type() {
|
||||||
|
if file_type.is_file() {
|
||||||
|
if let Ok(file_name) = path.file_name().into_string() {
|
||||||
|
if accounts_file_regex.is_match(&file_name) {
|
||||||
|
if let Err(e) = std::fs::remove_file(path.path()) {
|
||||||
|
info!("Couldn't delete file: {:?} error: {:?}", path, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Cleanup anything that looks like an accounts append-vec
|
||||||
|
fn cleanup_accounts_path(account_path: &std::path::Path) {
|
||||||
|
use regex::Regex;
|
||||||
|
let accounts_file_regex = Regex::new(r"(\d+).(\d+)").unwrap();
|
||||||
|
if let Ok(dir_entries) = std::fs::read_dir(&account_path) {
|
||||||
|
for entry in dir_entries {
|
||||||
|
if let Ok(path) = entry {
|
||||||
|
cleanup_accounts_dir_entry(path, &accounts_file_regex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
@ -1137,4 +1168,20 @@ mod tests {
|
|||||||
rpc_override_health_check
|
rpc_override_health_check
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn accounts_clean() {
|
||||||
|
use std::fs::File;
|
||||||
|
let temp_dir = tempfile::tempdir_in("farf").unwrap();
|
||||||
|
let temp_path = temp_dir.path();
|
||||||
|
{
|
||||||
|
let _file1 = File::create(temp_path.join("foo.txt")).unwrap();
|
||||||
|
let _file2 = File::create(temp_path.join("123.2222")).unwrap();
|
||||||
|
}
|
||||||
|
std::fs::create_dir(temp_path.join("12.088")).unwrap();
|
||||||
|
cleanup_accounts_path(temp_dir.path());
|
||||||
|
assert!(File::open(temp_path.join("foo.txt")).is_ok());
|
||||||
|
assert!(File::open(temp_path.join("123.2222")).is_err());
|
||||||
|
assert!(std::fs::read_dir(temp_path.join("12.088")).is_ok());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user