rsfu 0.1.1

a sfu library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use super::buffer::AtomicBuffer;
use super::rtcpreader::RTCPReader;
use std::collections::HashMap;
use std::sync::Arc;
use tokio::sync::Mutex;
#[derive(Default)]
pub struct Factory {
    #[allow(dead_code)]
    video_pool_size: usize,
    #[allow(dead_code)]
    audio_pool_size: usize,
    pub rtp_buffers: HashMap<u32, Arc<AtomicBuffer>>,
    pub rtcp_readers: HashMap<u32, Arc<Mutex<RTCPReader>>>,
}
#[derive(Default)]
pub struct AtomicFactory {
    factory: Arc<Mutex<Factory>>,
}

impl AtomicFactory {
    pub fn new(video_pool_size: usize, audio_pool_size: usize) -> Self {
        Self {
            factory: Arc::new(Mutex::new(Factory {
                video_pool_size,
                audio_pool_size,
                ..Default::default()
            })),
        }
    }

    pub async fn get_or_new_rtcp_buffer(&self, ssrc: u32) -> Arc<Mutex<RTCPReader>> {
        let factory = &mut self.factory.lock().await;

        if let Some(reader) = factory.rtcp_readers.get_mut(&ssrc) {
            return reader.clone();
        }

        let reader = Arc::new(Mutex::new(RTCPReader::new(ssrc)));
        factory.rtcp_readers.insert(ssrc, reader.clone());

        reader
    }

    pub async fn get_or_new_buffer(&self, ssrc: u32) -> Arc<AtomicBuffer> {
        let factory = &mut self.factory.lock().await;

        if let Some(reader) = factory.rtp_buffers.get_mut(&ssrc) {
            return reader.clone();
        }

        let reader = Arc::new(AtomicBuffer::new(ssrc));
        factory.rtp_buffers.insert(ssrc, reader.clone());

        reader
    }
}