thm/upend
thm/upend
1
0
Fork 0
Code Issues 75 Pull requests Projects Releases 6 Packages Wiki Activity

chore(backend): throttle job updates to 200ms
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Details

Browse source
This commit is contained in:
Tomáš Mládek 2024-08-18 21:16:12 +02:00
parent e1a22b8b86
commit 9fd07242f0
3 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Cargo.lock generated
View file

@ -3900,6 +3900,7 @@ dependencies = [
"tempfile",
"thiserror",
"tokio",
"tokio-stream",
"tracing",
"tracing-subscriber",
"tree_magic_mini",

1
cli/Cargo.toml
View file

@ -18,6 +18,7 @@ futures = "0.3.24"
futures-util = "~0.3.12"
tokio = { version = "1.38.1", features = ["sync", "fs", "process", "rt"] }
async-trait = "0.1.81"
tokio-stream = { version = "0.1", features = ["time"] }
clap = { version = "4.2.4", features = ["derive", "env", "color", "string", "cargo"] }
log = "0.4"

6
cli/src/routes.rs
View file

@ -27,7 +27,7 @@ use std::collections::HashMap;
use std::convert::{TryFrom, TryInto};
use std::io::Write;
use std::sync::{Arc, Mutex};
use std::time::{SystemTime, UNIX_EPOCH};
use std::time::{Duration, SystemTime, UNIX_EPOCH};
use tempfile::NamedTempFile;
use tracing::{debug, error, info, trace};
use upend_base::addressing::AddressComponents;
@ -1103,7 +1103,9 @@ pub async fn get_jobs_stream(
.stream_jobs()
.map_err(ErrorInternalServerError)?;
Ok(sse::Sse::from_stream(jobs.map(|jobs_msg| {
let throttled_jobs = tokio_stream::StreamExt::throttle(jobs, Duration::from_millis(200));
Ok(sse::Sse::from_stream(throttled_jobs.map(|jobs_msg| {
let jobs = jobs_msg.map_err(|err| anyhow!(err))?;
let data = sse::Data::new_json(jobs).map_err(|err| anyhow!(err))?;
Ok::<sse::Event, anyhow::Error>(sse::Event::Data(data))