add UserName type

webrtc-rs · Jan 7, 2024 · 4e17ce7 · 4e17ce7
1 parent 7d9c784
commit 4e17ce7
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/src/server/session/endpoint/candidate.rs b/src/server/session/endpoint/candidate.rs
@@ -1,5 +1,5 @@
 use crate::server::certificate::RTCDtlsFingerprint;
-use crate::shared::types::{EndpointId, SessionId};
+use crate::shared::types::{EndpointId, SessionId, UserName};
 use base64::{prelude::BASE64_STANDARD, Engine};
 use ring::rand::{SecureRandom, SystemRandom};
 use sdp::util::ConnectionRole;
@@ -121,7 +121,7 @@ impl Candidate {
         self.endpoint_id
     }
 
-    pub(crate) fn username(&self) -> String {
+    pub(crate) fn username(&self) -> UserName {
         format!(
             "{}:{}",
             self.local_conn_cred.ice_ufrag, self.peer_conn_cred.ice_ufrag

diff --git a/src/server/session/mod.rs b/src/server/session/mod.rs
@@ -7,15 +7,15 @@ pub mod endpoint;
 
 use crate::server::certificate::RTCDtlsFingerprint;
 use crate::server::session::endpoint::candidate::{Candidate, ConnectionCredentials};
-use crate::shared::types::{EndpointId, SessionId};
+use crate::shared::types::{EndpointId, SessionId, UserName};
 use endpoint::Endpoint;
 
 #[derive(Default, Debug)]
 pub struct Session {
     session_id: SessionId,
     fingerprint: RTCDtlsFingerprint,
     endpoints: RefCell<HashMap<EndpointId, Rc<Endpoint>>>,
-    candidates: RefCell<HashMap<String, Rc<Candidate>>>,
+    candidates: RefCell<HashMap<UserName, Rc<Candidate>>>,
 }
 
 impl Session {
@@ -69,7 +69,7 @@ impl Session {
         candidates.remove(&username).is_some()
     }
 
-    pub(crate) fn find_candidate(&self, username: &str) -> Option<Rc<Candidate>> {
+    pub(crate) fn find_candidate(&self, username: &UserName) -> Option<Rc<Candidate>> {
         let candidates = self.candidates.borrow();
         candidates.get(username).cloned()
     }

diff --git a/src/shared/types.rs b/src/shared/types.rs
@@ -1,8 +1,9 @@
 use retty::transport::TransportContext;
 use std::net::SocketAddr;
 
-pub type EndpointId = u64;
 pub type SessionId = u64;
+pub type EndpointId = u64;
+pub type UserName = String;
 
 #[derive(Debug, Copy, Clone, Eq, PartialEq, Hash, Ord, PartialOrd)]
 pub struct FourTuple {