Kieran/zap-stream-core
1
0
Fork 0
mirror of https://github.com/v0l/zap-stream-core.git synced 2025-06-20 13:40:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: add claude suggestions for performance

Browse Source
This commit is contained in:
Kieran
2025-06-06 15:39:45 +01:00
parent fbe0c58771
commit f5fe85e5c5
3 changed files with 74 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
crates/core/src/ingress/srt.rs
View File

@ -3,13 +3,15 @@ use crate::overseer::Overseer;
use anyhow::Result;
use futures_util::stream::FusedStream;
use futures_util::StreamExt;
use log::info;
use log::{info, warn};
use srt_tokio::{SrtListener, SrtSocket};
use std::io::Read;
use std::net::SocketAddr;
use std::sync::Arc;
use tokio::runtime::Handle;
const MAX_SRT_BUFFER_SIZE: usize = 10 * 1024 * 1024; // 10MB limit
pub async fn listen(out_dir: String, addr: String, overseer: Arc<dyn Overseer>) -> Result<()> {
let binder: SocketAddr = addr.parse()?;
let (_binding, mut packets) = SrtListener::builder().bind(binder).await?;
@ -48,6 +50,19 @@ struct SrtReader {
pub buf: Vec<u8>,
}
impl SrtReader {
/// Add data to buffer with size limit to prevent unbounded growth
fn add_to_buffer(&mut self, data: &[u8]) {
if self.buf.len() + data.len() > MAX_SRT_BUFFER_SIZE {
let bytes_to_drop = (self.buf.len() + data.len()) - MAX_SRT_BUFFER_SIZE;
warn!("SRT buffer full ({} bytes), dropping {} oldest bytes",
self.buf.len(), bytes_to_drop);
self.buf.drain(..bytes_to_drop);
}
self.buf.extend(data);
}
}
impl Read for SrtReader {
fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
let (mut rx, _) = self.socket.split_mut();
@ -56,7 +71,7 @@ impl Read for SrtReader {
return Ok(0);
}
if let Some((_, data)) = self.handle.block_on(rx.next()) {
self.buf.extend(data.iter().as_slice());
self.add_to_buffer(data.iter().as_slice());
}
}
let drain = self.buf.drain(..buf.len());