nordic-dev.net/embassy
2
0
Fork 0
mirror of https://github.com/embassy-rs/embassy.git synced 2024-11-25 16:23:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

stm32: i2c: Clean up conditional code a bit

Browse Source 
By moving conditional code inside the functions, we can
reduce duplication and in one case we can even eliminate one...
This commit is contained in:
Priit Laes 2023-12-19 11:48:58 +02:00
parent 08e9a4d84a
commit fc724dd707
1 changed files with 15 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
embassy-stm32/src/i2c/mod.rs
View File

@ -148,38 +148,31 @@ struct Timeout {
#[allow(dead_code)] #[allow(dead_code)]
impl Timeout { impl Timeout {
#[cfg(not(feature = "time"))]
#[inline] #[inline]
fn check(self) -> Result<(), Error> { fn check(self) -> Result<(), Error> {
#[cfg(feature = "time")]
if Instant::now() > self.deadline {
return Err(Error::Timeout);
}
Ok(()) Ok(())
} }
#[cfg(feature = "time")]
#[inline] #[inline]
fn check(self) -> Result<(), Error> { fn with<R>(self, fut: impl Future<Output = Result<R, Error>>) -> impl Future<Output = Result<R, Error>> {
if Instant::now() > self.deadline { #[cfg(feature = "time")]
Err(Error::Timeout) {
} else { use futures::FutureExt;
Ok(())
embassy_futures::select::select(embassy_time::Timer::at(self.deadline), fut).map(|r| match r {
embassy_futures::select::Either::First(_) => Err(Error::Timeout),
embassy_futures::select::Either::Second(r) => r,
})
} }
}
#[cfg(not(feature = "time"))] #[cfg(not(feature = "time"))]
#[inline]
fn with<R>(self, fut: impl Future<Output = Result<R, Error>>) -> impl Future<Output = Result<R, Error>> {
fut fut
} }
#[cfg(feature = "time")]
#[inline]
fn with<R>(self, fut: impl Future<Output = Result<R, Error>>) -> impl Future<Output = Result<R, Error>> {
use futures::FutureExt;
embassy_futures::select::select(embassy_time::Timer::at(self.deadline), fut).map(|r| match r {
embassy_futures::select::Either::First(_) => Err(Error::Timeout),
embassy_futures::select::Either::Second(r) => r,
})
}
} }
pub(crate) mod sealed { pub(crate) mod sealed {