make transceivers in Endpoint private

webrtc-rs · Jan 25, 2024 · e78e805 · e78e805
1 parent 0304a1e
commit e78e805
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/src/server/endpoint/mod.rs b/src/server/endpoint/mod.rs
@@ -14,8 +14,7 @@ pub(crate) struct Endpoint {
     session: Weak<Session>,
     endpoint_id: EndpointId,
     transports: RefCell<HashMap<FourTuple, Transport>>,
-
-    pub(crate) transceivers: RefCell<HashMap<Mid, RTCRtpTransceiver>>,
+    transceivers: RefCell<HashMap<Mid, RTCRtpTransceiver>>,
 }
 
 impl Endpoint {
@@ -41,6 +40,11 @@ impl Endpoint {
         transports.insert(*transport.four_tuple(), transport);
     }
 
+    pub(crate) fn remove_transport(&self, four_tuple: &FourTuple) {
+        let mut transports = self.transports.borrow_mut();
+        transports.remove(four_tuple);
+    }
+
     pub(crate) fn has_transport(&self, four_tuple: &FourTuple) -> bool {
         let transports = self.transports.borrow();
         transports.contains_key(four_tuple)
@@ -49,4 +53,8 @@ impl Endpoint {
     pub(crate) fn transports(&self) -> &RefCell<HashMap<FourTuple, Transport>> {
         &self.transports
     }
+
+    pub(crate) fn transceivers(&self) -> &RefCell<HashMap<Mid, RTCRtpTransceiver>> {
+        &self.transceivers
+    }
 }
diff --git a/src/server/session/mod.rs b/src/server/session/mod.rs
@@ -98,7 +98,7 @@ impl Session {
             .as_ref()
             .ok_or(Error::Other("Unparsed remote description".to_string()))?;
 
-        let mut local_transceivers = endpoint.transceivers.borrow_mut();
+        let mut local_transceivers = endpoint.transceivers().borrow_mut();
         let we_offer = remote_description.sdp_type == RTCSdpType::Answer;
 
         if !we_offer {
@@ -250,7 +250,7 @@ impl Session {
 
         let media_sections = {
             let mut local_transceivers = if let Some(endpoint) = endpoint.as_ref() {
-                endpoint.transceivers.borrow_mut()
+                endpoint.transceivers().borrow_mut()
             } else {
                 empty_transceivers.borrow_mut()
             };
@@ -334,7 +334,7 @@ impl Session {
             };
 
         let local_transceiver = if let Some(endpoint) = endpoint.as_ref() {
-            endpoint.transceivers.borrow()
+            endpoint.transceivers().borrow()
         } else {
             empty_transceivers.borrow()
         };