silas/mkv
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Format

Browse source
This commit is contained in:
Silas Brack 2026-03-08 17:55:37 +01:00
parent f19656486a
commit 07e1ab0796
3 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.gitignore vendored
View file

@ -2,3 +2,5 @@
*.db* *.db*
/result /result
*~ *~
#*
#*#

4
src/rebuild.rs
View file

@ -19,7 +19,9 @@ pub fn merge_volume_scans(
let mut index: HashMap<String, (Vec<String>, i64)> = HashMap::new(); let mut index: HashMap<String, (Vec<String>, i64)> = HashMap::new();
for (vol_url, keys) in scans { for (vol_url, keys) in scans {
for (key, size) in keys { for (key, size) in keys {
let entry = index.entry(key.clone()).or_insert_with(|| (Vec::new(), *size)); let entry = index
.entry(key.clone())
.or_insert_with(|| (Vec::new(), *size));
entry.0.push(vol_url.clone()); entry.0.push(vol_url.clone());
if *size > entry.1 { if *size > entry.1 {
entry.1 = *size; entry.1 = *size;

11
src/server.rs
View file

@ -33,8 +33,8 @@ pub fn first_healthy_volume(key: &str, volumes: &[String], results: &[bool]) ->
} }
pub fn shuffle_volumes(volumes: Vec<String>, seed: u64) -> Vec<String> { pub fn shuffle_volumes(volumes: Vec<String>, seed: u64) -> Vec<String> {
use rand::seq::SliceRandom;
use rand::SeedableRng; use rand::SeedableRng;
use rand::seq::SliceRandom;
let mut rng = rand::rngs::StdRng::seed_from_u64(seed); let mut rng = rand::rngs::StdRng::seed_from_u64(seed);
let mut vols = volumes; let mut vols = volumes;
vols.shuffle(&mut rng); vols.shuffle(&mut rng);
@ -77,11 +77,9 @@ pub async fn get_key(
} }
match first_healthy_volume(&key, &volumes, &results) { match first_healthy_volume(&key, &volumes, &results) {
ProbeResult::Found(url) => Ok(( ProbeResult::Found(url) => {
StatusCode::FOUND, Ok((StatusCode::FOUND, [(axum::http::header::LOCATION, url)]).into_response())
[(axum::http::header::LOCATION, url)], }
)
.into_response()),
ProbeResult::AllFailed => Err(AppError::AllVolumesUnreachable), ProbeResult::AllFailed => Err(AppError::AllVolumesUnreachable),
} }
} }
@ -233,4 +231,3 @@ pub async fn list_keys(
let keys = state.db.list_keys(&query.prefix).await?; let keys = state.db.list_keys(&query.prefix).await?;
Ok((StatusCode::OK, keys.join("\n")).into_response()) Ok((StatusCode::OK, keys.join("\n")).into_response())
} }