euzu/tuliprox
1
0
Fork 0
mirror of https://github.com/euzu/tuliprox.git synced 2026-10-03 14:32:08 +02:00
Code Issues Packages Projects Releases 5 Wiki Activity

added unit to throttle

Browse Source
This commit is contained in:
euzu
2025-03-28 17:31:24 +01:00
parent 5f9576e2a0
commit 5842e00ed4
9 changed files with 73 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/api_utils.rs
+4 -9
View File
@@ -351,13 +351,8 @@ pub async fn stream_response(app_state: &AppState,
let throttle_kbps = get_stream_throttle(app_state);
let body_stream = if throttle_kbps > 0 {
match item_type {
PlaylistItemType::Video | PlaylistItemType::Series | PlaylistItemType::SeriesInfo => {
axum::body::Body::from_stream(ThrottledStream::new(stream.boxed(), throttle_kbps as usize))
}
_ => axum::body::Body::from_stream(stream)
}
let body_stream = if throttle_kbps > 0 && matches!(item_type, PlaylistItemType::Video | PlaylistItemType::Series | PlaylistItemType::SeriesInfo) {
axum::body::Body::from_stream(ThrottledStream::new(stream.boxed(), throttle_kbps as usize))
} else {
axum::body::Body::from_stream(stream)
};
@@ -372,12 +367,12 @@ pub async fn stream_response(app_state: &AppState,
axum::http::StatusCode::BAD_REQUEST.into_response()
}
fn get_stream_throttle(app_state: &AppState) -> u32 {
fn get_stream_throttle(app_state: &AppState) -> u64 {
app_state.config
.reverse_proxy
.as_ref()
.and_then(|reverse_proxy| reverse_proxy.stream.as_ref())
.map(|stream| stream.throttle_kbps).map_or(0, |t| t.unwrap_or_default())
.map(|stream| stream.throttle_kbps).unwrap_or_default()
}
async fn shared_stream_response(app_state: &AppState, stream_url: &str, user: &ProxyUserCredentials) -> Option<impl IntoResponse> {