xref: /webrtc/util/src/vnet/chunk_queue.rs (revision ffe74184)

#[cfg(test)]
mod chunk_queue_test;

use super::chunk::*;

use std::collections::VecDeque;
use tokio::sync::RwLock;

#[derive(Default)]
pub(crate) struct ChunkQueue {
    chunks: RwLock<VecDeque<Box<dyn Chunk + Send + Sync>>>,
    max_size: usize, // 0 or negative value: unlimited
}

impl ChunkQueue {
    pub(crate) fn new(max_size: usize) -> Self {
        ChunkQueue {
            chunks: RwLock::new(VecDeque::new()),
            max_size,
        }
    }

    pub(crate) async fn push(&self, c: Box<dyn Chunk + Send + Sync>) -> bool {
        let mut chunks = self.chunks.write().await;

        if self.max_size > 0 && chunks.len() >= self.max_size {
            false // dropped
        } else {
            chunks.push_back(c);
            true
        }
    }

    pub(crate) async fn pop(&self) -> Option<Box<dyn Chunk + Send + Sync>> {
        let mut chunks = self.chunks.write().await;
        chunks.pop_front()
    }

    pub(crate) async fn peek(&self) -> Option<Box<dyn Chunk + Send + Sync>> {
        let chunks = self.chunks.read().await;
        chunks.front().map(|chunk| chunk.clone_to())
    }
}