You've already forked AstralRinth
forked from didirus/AstralRinth
13187de97d
* typos :help_me: * (part 1/?) massive cleanup to make the code more Rust-ic and cut down heap allocations. * (part 2/?) massive cleanup to make the code more Rust-ic and cut down heap allocations. * (part 3/?) cut down some pretty major heap allocations here - more Bytes and BytesMuts, less Vec<u8>s also I don't really understand why you need to `to_vec` when you don't really use it again afterwards * (part 4/?) deduplicate error handling in backblaze logic * (part 5/?) fixes, cleanups, refactors, and reformatting * (part 6/?) cleanups and refactors * remove loads of `as_str` in types that already are `Display` * Revert "remove loads of `as_str` in types that already are `Display`" This reverts commit 4f974310cfb167ceba03001d81388db4f0fbb509. * reformat and move routes util to the util module * use streams * Run prepare + formatting issues Co-authored-by: Jai A <jaiagr+gpg@pm.me> Co-authored-by: Geometrically <18202329+Geometrically@users.noreply.github.com>
31 lines
1.0 KiB
Rust
31 lines
1.0 KiB
Rust
use super::{add_projects, IndexingError, UploadSearchProject};
|
|
use crate::search::SearchConfig;
|
|
use std::sync::Mutex;
|
|
|
|
pub struct CreationQueue {
|
|
// There's probably a better structure for this, but a mutex works
|
|
// and I don't think this can deadlock. This queue requires fast
|
|
// writes and then a single potentially slower read/write that
|
|
// empties the queue.
|
|
queue: Mutex<Vec<UploadSearchProject>>,
|
|
}
|
|
|
|
impl CreationQueue {
|
|
pub fn new() -> Self {
|
|
CreationQueue {
|
|
queue: Mutex::new(Vec::with_capacity(10)),
|
|
}
|
|
}
|
|
pub fn add(&self, search_project: UploadSearchProject) {
|
|
// Can only panic if mutex is poisoned
|
|
self.queue.lock().unwrap().push(search_project);
|
|
}
|
|
pub fn take(&self) -> Vec<UploadSearchProject> {
|
|
std::mem::replace(&mut *self.queue.lock().unwrap(), Vec::with_capacity(10))
|
|
}
|
|
pub async fn index(&self, config: &SearchConfig) -> Result<(), IndexingError> {
|
|
let queue = self.take();
|
|
add_projects(queue, config).await
|
|
}
|
|
}
|