set injector to NeedHandshake state on unexpected errror

tursodatabase · Nov 23, 2023 · 42b1a56 · 42b1a56
1 parent 7aa3ddb
commit 42b1a56
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/libsql-replication/src/replicator.rs b/libsql-replication/src/replicator.rs
@@ -237,7 +237,12 @@ impl<C: ReplicatorClient> Replicator<C> {
                 ReplicatorState::NeedHandshake
             }
             Err(Error::NeedSnapshot) => ReplicatorState::NeedSnapshot,
-            Err(e) => return Err(e),
+            Err(e) => {
+                // an error here could be due to a disconnection, it's safe to rollback to a
+                // NeedHandshake state again, to avoid entering a busy loop.
+                self.state = ReplicatorState::NeedHandshake;
+                return Err(e);
+            }
         };
 
         Ok(())