Protocol loop should continue on error (#387)

near · Nov 28, 2023 · 3cb44f3 · 3cb44f3
1 parent 0357770
commit 3cb44f3
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/node/src/protocol/mod.rs b/node/src/protocol/mod.rs
@@ -189,13 +189,28 @@ impl MpcSignProtocol {
                 }
             }
 
-            let mut state = {
-                let guard = self.state.write().await;
+            let state = {
+                let guard = self.state.read().await;
                 guard.clone()
             };
-            state = state.progress(&self.ctx).await?;
-            state = state.advance(&self.ctx, contract_state).await?;
-            state.handle(&self.ctx, &mut queue).await?;
+            let state = match state.progress(&self.ctx).await {
+                Ok(state) => state,
+                Err(err) => {
+                    tracing::info!("protocol unable to progress: {err:?}");
+                    continue;
+                }
+            };
+            let mut state = match state.advance(&self.ctx, contract_state).await {
+                Ok(state) => state,
+                Err(err) => {
+                    tracing::info!("protocol unable to advance: {err:?}");
+                    continue;
+                }
+            };
+            if let Err(err) = state.handle(&self.ctx, &mut queue).await {
+                tracing::info!("protocol unable to handle messages: {err:?}");
+                continue;
+            }
 
             let mut guard = self.state.write().await;
             *guard = state;