72 lines
1.8 KiB
Rust
72 lines
1.8 KiB
Rust
use lazy_static::lazy_static;
|
|
use nix::unistd;
|
|
use regex::Regex;
|
|
use std::boxed::Box;
|
|
use std::fs::File;
|
|
use std::io::*;
|
|
use std::path::Path;
|
|
use tokio::time::{sleep, Duration};
|
|
|
|
lazy_static! {
|
|
static ref R_FILE_GZIP: Regex = Regex::new(r".*\.gz.*").unwrap();
|
|
}
|
|
|
|
pub fn read_lines(filename: &String, offset: u64) -> Option<Box<dyn Read>> {
|
|
let mut file = match File::open(filename) {
|
|
Ok(f) => f,
|
|
Err(err) => {
|
|
println!("{err}");
|
|
return None;
|
|
}
|
|
};
|
|
file.seek(SeekFrom::Start(offset)).unwrap();
|
|
let lines: Box<dyn Read> = Box::new(BufReader::new(file));
|
|
Some(lines)
|
|
}
|
|
|
|
pub fn _dedup<T: Ord + PartialOrd>(list: &mut Vec<T>) -> usize {
|
|
// Begin with sorting entries
|
|
list.sort();
|
|
// Then deduplicate
|
|
list.dedup();
|
|
// Return the length
|
|
list.len()
|
|
}
|
|
|
|
pub async fn _sleep_ms(ms: u64) {
|
|
sleep(Duration::from_millis(ms)).await;
|
|
}
|
|
|
|
pub async fn sleep_s(s: u64) {
|
|
sleep(Duration::from_secs(s)).await;
|
|
}
|
|
|
|
pub fn gethostname(show_fqdn: bool) -> String {
|
|
let hostname_cstr = unistd::gethostname().expect("Failed getting hostname");
|
|
let fqdn = hostname_cstr
|
|
.to_str()
|
|
.expect("Hostname wasn't valid UTF-8")
|
|
.to_string();
|
|
let hostname: Vec<&str> = fqdn.split(".").collect();
|
|
if show_fqdn {
|
|
return fqdn;
|
|
}
|
|
hostname[0].to_string()
|
|
}
|
|
|
|
pub fn _search_subfolders(path: &Path) -> Vec<String> {
|
|
let dirs = std::fs::read_dir(path).unwrap();
|
|
let mut folders: Vec<String> = vec![];
|
|
for dir in dirs {
|
|
let dirpath = dir.unwrap().path();
|
|
let path = Path::new(dirpath.as_path());
|
|
if path.is_dir() {
|
|
folders.push(dirpath.to_str().unwrap().to_string());
|
|
for f in _search_subfolders(path) {
|
|
folders.push(f);
|
|
}
|
|
}
|
|
}
|
|
folders
|
|
}
|