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

Drop Provider connection on custom stream response

Browse Source
This commit is contained in:
euzu
2025-07-23 19:11:44 +02:00
parent fed7b8051f
commit 213496cf90
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/src/api/model/streams/active_client_stream.rs
+13 -2
View File
@@ -121,6 +121,7 @@ impl ActiveClientStream {
let stream_strategy_flag_copy = Arc::clone(&stream_strategy_flag);
let waker_copy = Arc::clone(waker);
let grace_period_millis = stream_details.grace_period_millis;
let provider_guard = stream_details.provider_connection_guard.clone();
let address = addr.to_string();
tokio::spawn(async move {
@@ -133,6 +134,9 @@ impl ActiveClientStream {
if active_connections > max_connections {
info!("User connections exhausted for active clients: {username}");
stream_strategy_flag_copy.store(USER_EXHAUSTED_STREAM, std::sync::atomic::Ordering::SeqCst);
if let Some(guard) = provider_guard.as_ref() {
guard.release();
}
if let Some(flag) = reconnect_flag {
info!("Stopped reconnecting, user connections exhausted: {username}");
flag.notify();
@@ -147,6 +151,9 @@ impl ActiveClientStream {
info!("Provider connections exhausted for active clients: {provider_name}");
provider_manager.release_connection(&address);
stream_strategy_flag_copy.store(PROVIDER_EXHAUSTED_STREAM, std::sync::atomic::Ordering::SeqCst);
if let Some(guard) = provider_guard.as_ref() {
guard.release();
}
if let Some(flag) = reconnect_flag {
info!("Stopped reconnecting, provider connections exhausted {provider_name}");
flag.notify();
@@ -196,8 +203,12 @@ impl Stream for ActiveClientStream {
}
let buffer_opt = match flag {
USER_EXHAUSTED_STREAM => self.custom_video.0.as_mut(),
PROVIDER_EXHAUSTED_STREAM => self.custom_video.1.as_mut(),
USER_EXHAUSTED_STREAM => {
self.custom_video.0.as_mut()
},
PROVIDER_EXHAUSTED_STREAM => {
self.custom_video.1.as_mut()
},
_ => None,
};