cec8393770
The relevant processes are so fast, mutexes and mutex locks are so
expensive, and iterators so efficient, that it's actually faster to run
single-threaded across all the data than to spin up a bunch of threads
and have them basically spinlock waiting for the global mutex involved
either directly or in a channel.
ivy_files(kubernetes) time: [10.209 ms 10.245 ms 10.286 ms]
change: [-36.781% -36.178% -35.601%] (p = 0.00 < 0.05)
Performance has improved.
ivy_match(file.lua) time: [1.1626 µs 1.1668 µs 1.1709 µs]
change: [+0.2131% +1.5409% +2.9109%] (p = 0.02 < 0.05)
Change within noise threshold.
87 lines
2.1 KiB
Rust
87 lines
2.1 KiB
Rust
mod finder;
|
|
mod matcher;
|
|
mod sorter;
|
|
mod thread_pool;
|
|
|
|
use std::collections::HashMap;
|
|
use std::ffi::CStr;
|
|
use std::ffi::CString;
|
|
use std::os::raw::{c_char, c_int};
|
|
use std::sync::Mutex;
|
|
|
|
#[macro_use]
|
|
extern crate lazy_static;
|
|
|
|
lazy_static! {
|
|
static ref GLOBAL_FILE_CACHE: Mutex<HashMap<String, Vec<String>>> = Mutex::new(HashMap::new());
|
|
}
|
|
|
|
fn to_string(input: *const c_char) -> String {
|
|
unsafe { CStr::from_ptr(input) }
|
|
.to_str()
|
|
.unwrap()
|
|
.to_string()
|
|
}
|
|
|
|
fn get_files(directory: &String) -> Vec<String> {
|
|
let mut cache = GLOBAL_FILE_CACHE.lock().unwrap();
|
|
if !cache.contains_key(directory) {
|
|
let finder_options = finder::Options {
|
|
directory: directory.clone(),
|
|
};
|
|
cache.insert(directory.clone(), finder::find_files(finder_options));
|
|
}
|
|
|
|
return cache.get(directory).unwrap().to_vec();
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_init(c_base_dir: *const c_char) {
|
|
let directory = to_string(c_base_dir);
|
|
get_files(&directory);
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_match(c_pattern: *const c_char, c_text: *const c_char) -> c_int {
|
|
let pattern = to_string(c_pattern);
|
|
let text = to_string(c_text);
|
|
|
|
inner_match(pattern, text)
|
|
}
|
|
|
|
pub fn inner_match(pattern: String, text: String) -> i32 {
|
|
let m = matcher::Matcher::new(pattern);
|
|
|
|
m.score(text.as_str()) as i32
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_files(c_pattern: *const c_char, c_base_dir: *const c_char) -> *const c_char {
|
|
let pattern = to_string(c_pattern);
|
|
let directory = to_string(c_base_dir);
|
|
|
|
let output = inner_files(pattern, directory);
|
|
|
|
CString::new(output).unwrap().into_raw()
|
|
}
|
|
|
|
pub fn inner_files(pattern: String, base_dir: String) -> String {
|
|
let mut output = String::new();
|
|
|
|
// Bail out early if the pattern is empty; it's never going to find anything
|
|
if pattern.is_empty() {
|
|
return output;
|
|
}
|
|
|
|
let files = get_files(&base_dir);
|
|
|
|
let sorter_options = sorter::Options::new(pattern);
|
|
|
|
let files = sorter::sort_strings(sorter_options, files);
|
|
for file in files.iter() {
|
|
output.push_str(&file.content);
|
|
output.push('\n');
|
|
}
|
|
|
|
output
|
|
}
|