[python] - Implement RaftNode.store

lablup · Jan 2, 2024 · b637c86 · b637c86
1 parent e97ae48
commit b637c86
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/binding/python/raftify.pyi b/binding/python/raftify.pyi
@@ -92,6 +92,8 @@ class RaftNode:
         """ """
     async def get_cluster_size(self) -> None:
         """ """
+    async def store(self) -> AbstractStateMachine:
+        """ """
 
 class ClusterJoinTicket:
     """ """
@@ -199,6 +201,7 @@ class Config:
 
     raft_config: RaftConfig
     log_dir: str
+    save_compacted_logs: bool
     compacted_log_dir: str
     compacted_log_size_threshold: int
     snapshot_interval: float
@@ -210,6 +213,7 @@ class Config:
 
 class RaftServiceClient:
     """ """
+
     @staticmethod
     async def build(self, addr: str) -> "RaftServiceClient":
         """ """

diff --git a/binding/python/src/bindings/peers.rs b/binding/python/src/bindings/peers.rs
@@ -40,7 +40,7 @@ impl PyPeers {
             .inner
             .iter()
             .map(|(id, peer)| (id, peer.addr.to_string()))
-            .collect::<Vec<(u64, String)>>();
+            .collect::<Vec<_>>();
 
         Ok(new_py_list::<(u64, String), _>(py, peer_items)?.to_object(py))
     }

diff --git a/binding/python/src/bindings/raft_node.rs b/binding/python/src/bindings/raft_node.rs
@@ -140,4 +140,8 @@ impl PyRaftNode {
     pub async fn get_cluster_size(&mut self) -> PyResult<usize> {
         Ok(self.inner.get_cluster_size().await)
     }
+
+    pub async fn store(&self) -> PyResult<PyFSM> {
+        Ok(self.inner.store().await)
+    }
 }