lucas/rustler-core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 🚑 crash when not rustler available for a market

Browse Source
This commit is contained in:
Lucas Colombo 2024-06-07 11:26:02 -03:00
parent 321ddd6bcb
commit ecfaf6f1a1
Signed by: lucas
GPG Key ID: EF34786CFEFFAE35
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/rustlers/rustlerjar.rs
View File

@ -76,18 +76,18 @@ impl RustlerJar {
/// get the Rustler for the given market /// get the Rustler for the given market
pub fn get(&self, market: &market::Model) -> Option<&Arc<Mutex<Box<dyn Rustler>>>> { pub fn get(&self, market: &market::Model) -> Option<&Arc<Mutex<Box<dyn Rustler>>>> {
let key = self.get_key(market); let key = self.get_key(market)?;
self.rustlers.get(key) self.rustlers.get(key)
} }
/// get the mutable Rustler for the given market as a mutable reference /// get the mutable Rustler for the given market as a mutable reference
pub fn get_mut(&mut self, market: &market::Model) -> Option<&mut Arc<Mutex<Box<dyn Rustler>>>> { pub fn get_mut(&mut self, market: &market::Model) -> Option<&mut Arc<Mutex<Box<dyn Rustler>>>> {
let key = self.get_key(market).to_owned(); let key = self.get_key(market)?.to_owned();
self.rustlers.get_mut(&key) self.rustlers.get_mut(&key)
} }
/// get the key from the mappings for the given market /// get the key from the mappings for the given market
fn get_key(&self, market: &market::Model) -> &str { fn get_key(&self, market: &market::Model) -> Option<&String> {
self.mappings.get(&market.short_name).unwrap() self.mappings.get(&market.short_name)
} }
} }