Use ok_or_elese for lazy heap alloc.

privacy-scaling-explorations · May 21, 2024 · 94d01f4 · 94d01f4
1 parent 1d0b2ab
commit 94d01f4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/crates/mpz-ole/src/rot/receiver.rs b/crates/mpz-ole/src/rot/receiver.rs
@@ -78,10 +78,12 @@ where
     F: Field + Serialize + Deserialize,
 {
     async fn receive(&mut self, ctx: &mut Ctx, b_k: Vec<F>) -> Result<Vec<F>, OLEError> {
-        let (receiver_adjust, adjust) = self.core.adjust(b_k).ok_or(OLEError::new(
-            OLEErrorKind::InsufficientOLEs,
-            "Not enough OLEs available".into(),
-        ))?;
+        let (receiver_adjust, adjust) = self.core.adjust(b_k).ok_or_else(|| {
+            OLEError::new(
+                OLEErrorKind::InsufficientOLEs,
+                "Not enough OLEs available".into(),
+            )
+        })?;
 
         let channel = ctx.io_mut();
         channel.send(adjust).await?;

diff --git a/crates/mpz-ole/src/rot/sender.rs b/crates/mpz-ole/src/rot/sender.rs
@@ -82,10 +82,12 @@ where
     F: Field + Serialize + Deserialize,
 {
     async fn send(&mut self, ctx: &mut Ctx, a_k: Vec<F>) -> Result<Vec<F>, OLEError> {
-        let (sender_adjust, adjust) = self.core.adjust(a_k).ok_or(OLEError::new(
-            OLEErrorKind::InsufficientOLEs,
-            "Not enough OLEs available".into(),
-        ))?;
+        let (sender_adjust, adjust) = self.core.adjust(a_k).ok_or_else(|| {
+            OLEError::new(
+                OLEErrorKind::InsufficientOLEs,
+                "Not enough OLEs available".into(),
+            )
+        })?;
 
         let channel = ctx.io_mut();
         channel.send(adjust).await?;