replicator: Add set_primary_handshake_retries

haaawk · haaawk · commit 48fe492ba5b7 · 2024-08-19T10:48:23.000+02:00
Signed-off-by: Piotr Jastrzebski &lt;piotr@chiselstrike.com&gt;
diff --git a/libsql-replication/src/replicator.rs b/libsql-replication/src/replicator.rs
@@ -195,6 +195,11 @@ where
         self.state = ReplicatorState::NeedHandshake;
     }
 
+    /// configure number of handshake retries.
+    pub fn set_primary_handshake_retries(&mut self, retries: usize) {
+        self.max_handshake_retries = retries;
+    }
+
     pub fn client_mut(&mut self) -> &mut C {
         &mut self.client
     }

Original file line number	Diff line number	Diff line change
`@@ -195,6 +195,11 @@ where`
`195`	`195`	`self.state = ReplicatorState::NeedHandshake;`
`196`	`196`	`}`
`197`	`197`
	`198`	`+ /// configure number of handshake retries.`
	`199`	`+ pub fn set_primary_handshake_retries(&mut self, retries: usize) {`
	`200`	`+ self.max_handshake_retries = retries;`
	`201`	`+ }`
	`202`	`+`
`198`	`203`	`pub fn client_mut(&mut self) -> &mut C {`
`199`	`204`	`&mut self.client`
`200`	`205`	`}`