Finish implementation of new Rust Tokio Target
This commit is contained in:
75
examples/Rust/Server/src/main.rs
Normal file
75
examples/Rust/Server/src/main.rs
Normal file
@ -0,0 +1,75 @@
|
||||
use async_trait::async_trait;
|
||||
use example::base_lib::JRPCResult;
|
||||
use example::server::{SimpleTestService, SimpleTestServiceHandler};
|
||||
use example::{JRPCServer, Result, Test2};
|
||||
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
|
||||
use tokio::net::TcpStream;
|
||||
use tokio::sync::mpsc::channel;
|
||||
|
||||
#[derive(Clone)]
|
||||
struct TestServiceImplementation {}
|
||||
|
||||
#[async_trait]
|
||||
impl SimpleTestService for TestServiceImplementation {
|
||||
async fn GetTest(&self, name: String, age: i64) -> Result<Test2> {
|
||||
return Ok(Test2 { name, age });
|
||||
}
|
||||
|
||||
async fn TestNot(&self) -> Result<()> {
|
||||
return Ok(());
|
||||
}
|
||||
}
|
||||
|
||||
fn handle_connection(stream: TcpStream, server: &JRPCServer) {
|
||||
let (response_sender, mut response_receiver) = channel::<JRPCResult>(1);
|
||||
|
||||
let session = server.get_session(response_sender);
|
||||
|
||||
let (r, mut w) = stream.into_split();
|
||||
tokio::spawn(async move {
|
||||
loop {
|
||||
let res = response_receiver.recv().await;
|
||||
match res {
|
||||
None => break,
|
||||
Some(res) => w
|
||||
.write_all((serde_json::to_string(&res).unwrap() + "\n").as_bytes())
|
||||
.await
|
||||
.unwrap(),
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
tokio::spawn(async move {
|
||||
let mut reader = BufReader::new(r);
|
||||
loop {
|
||||
let mut line = String::new();
|
||||
let res = reader.read_line(&mut line).await;
|
||||
match res {
|
||||
Ok(size) => {
|
||||
if size > 0 {
|
||||
session.handle_request(serde_json::from_str(&line).unwrap())
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
Err(_) => break,
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
pub async fn main() {
|
||||
let mut srv = JRPCServer::new();
|
||||
srv.add_service(SimpleTestServiceHandler::new(Box::from(
|
||||
TestServiceImplementation {},
|
||||
)));
|
||||
|
||||
let listener = tokio::net::TcpListener::bind("0.0.0.0:4321").await.unwrap();
|
||||
|
||||
loop {
|
||||
let (stream, address) = listener.accept().await.unwrap();
|
||||
println!("Got connection from {}", address);
|
||||
handle_connection(stream, &srv);
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user