Skip to content

Commit

Permalink
Refactor: delete async-trait, fix typo, use tokio to replace async_std
Browse files Browse the repository at this point in the history
  • Loading branch information
getong committed Jan 28, 2024
1 parent 06b431b commit 0651bdf
Show file tree
Hide file tree
Showing 8 changed files with 18 additions and 16 deletions.
1 change: 0 additions & 1 deletion examples/raft-kv-memstore-singlethreaded/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@ repository = "https://github.com/datafuselabs/openraft"
[dependencies]
openraft = { path = "../../openraft", features = ["serde", "storage-v2", "singlethreaded"] }

async-trait = "0.1.36"
clap = { version = "4.1.11", features = ["derive", "env"] }
reqwest = { version = "0.11.9", features = ["json"] }
serde = { version = "1.0.114", features = ["derive"] }
Expand Down
1 change: 0 additions & 1 deletion examples/raft-kv-memstore/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@ path = "src/bin/main.rs"
openraft = { path = "../../openraft", features = ["serde", "storage-v2"] }

actix-web = "4.0.0-rc.2"
async-trait = "0.1.36"
clap = { version = "4.1.11", features = ["derive", "env"] }
reqwest = { version = "0.11.9", features = ["json"] }
serde = { version = "1.0.114", features = ["derive"] }
Expand Down
10 changes: 7 additions & 3 deletions examples/raft-kv-rocksdb/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,7 @@ path = "src/bin/main.rs"
[dependencies]
openraft = { path = "../../openraft", features = ["serde", "storage-v2"] }

async-std = { version = "1.12.0", features = ["attributes", "tokio1"] }
async-trait = "0.1.36"
tokio = { version = "1.35.1", features = ["full"] }
byteorder = "1.4.3"
clap = { version = "4.1.11", features = ["derive", "env"] }
reqwest = { version = "0.11.9", features = ["json"] }
Expand All @@ -34,7 +33,12 @@ serde_json = "1.0.57"
tide = { version = "0.16" }
# for toy-rpc, use `serde_json` instead of the default `serde_bincode`:
# bincode which enabled by default by toy-rpc, does not support `#[serde(flatten)]`: https://docs.rs/bincode/2.0.0-alpha.1/bincode/serde/index.html#known-issues
toy-rpc = { version = "0.8.6", default-features = false, features = [ "serde_json", "ws_async_std", "server", "client", "async_std_runtime", ] }
toy-rpc = { version = "0.8.6", features = [
"ws_tokio",
"server",
"client",
"tokio_runtime",
] }
tracing = "0.1.40"
tracing-subscriber = { version = "0.3.0", features = ["env-filter"] }

Expand Down
4 changes: 2 additions & 2 deletions examples/raft-kv-rocksdb/src/app.rs
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
use std::collections::BTreeMap;
use std::sync::Arc;

use async_std::sync::RwLock;
use openraft::Config;
use tokio::sync::RwLock;

use crate::ExampleRaft;
use crate::NodeId;
Expand All @@ -12,7 +12,7 @@ use crate::NodeId;
pub struct App {
pub id: NodeId,
pub api_addr: String,
pub rcp_addr: String,
pub rpc_addr: String,
pub raft: ExampleRaft,
pub key_values: Arc<RwLock<BTreeMap<String, String>>>,
pub config: Arc<Config>,
Expand Down
2 changes: 1 addition & 1 deletion examples/raft-kv-rocksdb/src/bin/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ pub struct Opt {
pub rpc_addr: String,
}

#[async_std::main]
#[tokio::main]
async fn main() -> std::io::Result<()> {
// Setup the logger
tracing_subscriber::fmt()
Expand Down
12 changes: 6 additions & 6 deletions examples/raft-kv-rocksdb/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@ use std::io::Cursor;
use std::path::Path;
use std::sync::Arc;

use async_std::net::TcpListener;
use async_std::task;
use openraft::Config;
use openraft::TokioRuntime;
use tokio::net::TcpListener;
use tokio::task;

use crate::app::App;
use crate::network::api;
Expand Down Expand Up @@ -79,7 +79,7 @@ pub async fn start_example_raft_node<P>(
node_id: NodeId,
dir: P,
http_addr: String,
rcp_addr: String,
rpc_addr: String,
) -> std::io::Result<()>
where
P: AsRef<Path>,
Expand Down Expand Up @@ -107,7 +107,7 @@ where
let app = Arc::new(App {
id: node_id,
api_addr: http_addr.clone(),
rcp_addr: rcp_addr.clone(),
rpc_addr: rpc_addr.clone(),
raft,
key_values: kvs,
config,
Expand All @@ -117,7 +117,7 @@ where

let server = toy_rpc::Server::builder().register(echo_service).build();

let listener = TcpListener::bind(rcp_addr).await.unwrap();
let listener = TcpListener::bind(rpc_addr).await.unwrap();
let handle = task::spawn(async move {
server.accept_websocket(listener).await.unwrap();
});
Expand All @@ -130,6 +130,6 @@ where
api::rest(&mut app);

app.listen(http_addr).await?;
handle.await;
_ = handle.await;
Ok(())
}
2 changes: 1 addition & 1 deletion examples/raft-kv-rocksdb/src/network/management.rs
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ async fn init(req: Request<Arc<App>>) -> tide::Result {
let mut nodes = BTreeMap::new();
let node = Node {
api_addr: req.state().api_addr.clone(),
rpc_addr: req.state().rcp_addr.clone(),
rpc_addr: req.state().rpc_addr.clone(),
};

nodes.insert(req.state().id, node);
Expand Down
2 changes: 1 addition & 1 deletion examples/raft-kv-rocksdb/src/store.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@ use std::ops::RangeBounds;
use std::path::Path;
use std::sync::Arc;

use async_std::sync::RwLock;
use byteorder::BigEndian;
use byteorder::ReadBytesExt;
use byteorder::WriteBytesExt;
Expand Down Expand Up @@ -35,6 +34,7 @@ use rocksdb::Options;
use rocksdb::DB;
use serde::Deserialize;
use serde::Serialize;
use tokio::sync::RwLock;

use crate::typ;
use crate::Node;
Expand Down

0 comments on commit 0651bdf

Please sign in to comment.