Inital pass at route guide

This commit is contained in:
Lucio Franco
2019-08-16 14:46:18 -04:00
parent 28a0b9e286
commit 1630547f97
10 changed files with 331 additions and 114 deletions
+6 -3
View File
@@ -1,7 +1,7 @@
#![feature(async_await)]
use std::time::Duration;
use tokio::{timer::Delay, net::TcpListener};
use tokio::{net::TcpListener, timer::Delay};
use tonic::{Request, Response, Status};
use tower_h2::Server;
@@ -16,7 +16,10 @@ pub struct MyGreeter {
#[tonic::server(service = "helloworld.Greeter", proto = "hello_world")]
impl MyGreeter {
pub async fn say_hello(&self, request: Request<hello_world::HelloRequest>) -> Result<Response<hello_world::HelloReply>, Status> {
pub async fn say_hello(
&self,
request: Request<hello_world::HelloRequest>,
) -> Result<Response<hello_world::HelloReply>, Status> {
println!("Got a request: {:?}", request);
let string = &self.data;
@@ -27,7 +30,7 @@ impl MyGreeter {
println!("My data: {:?}", string);
Delay::new(when).await;
let reply = hello_world::HelloReply {
message: "Zomg, it works!".into(),
};
+68
View File
@@ -0,0 +1,68 @@
#![feature(async_await)]
use futures::Stream;
use tokio::net::TcpListener;
use tonic::{Request, Response, Status};
use tower_h2::Server;
pub mod routeguide {
include!(concat!(env!("OUT_DIR"), "/routeguide.rs"));
}
use routeguide::*;
type BoxStream<T> = Pin<Box<dyn Stream<Item = Result<T, Status>> + Send + 'static>>;
#[derive(Default, Clone)]
pub struct RouteGuide {
data: String,
}
#[tonic::server(service = "routeguide.RouteGuide", proto = "routeguide")]
impl RouteGuide {
pub async fn get_feature(&self, _req: Request<Point>) -> Result<Response<Feature>, Status> {
unimplemented!()
}
pub async fn list_features(
&self,
_req: Request<Rectangle>,
) -> Result<Response<BoxStream<Feature>>, Status> {
unimplemented!()
}
pub async fn record_route(
&self,
_req: Request<impl Stream<Item = Result<Point, Status>>>,
) -> Result<Response<RouteSummary>, Status> {
unimplemented!()
}
// pub async fn route_chat(
// &self,
// _req: Request<impl Stream<Item = Result<RouteNote, Status>>>,
// ) -> Result<Response<impl Stream<Item = Result<RouteNote, Status>>>, Status> {
// unimplemented!()
// }
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let addr = "[::1]:50051".parse().unwrap();
let mut bind = TcpListener::bind(&addr)?;
let route_guide = RouteGuide::default();
let mut server = Server::new(RouteGuideServer::new(route_guide), Default::default());
while let Ok((sock, _addr)) = bind.accept().await {
if let Err(e) = sock.set_nodelay(true) {
return Err(e.into());
}
if let Err(e) = server.serve(sock).await {
println!("H2 ERROR: {}", e);
}
}
Ok(())
}