55 lines
1.4 KiB
Rust
55 lines
1.4 KiB
Rust
pub mod pb {
|
|
tonic::include_proto!("/grpc.examples.unaryecho");
|
|
}
|
|
|
|
use pb::{EchoRequest, EchoResponse};
|
|
use tonic::{
|
|
transport::{
|
|
server::{TcpConnectInfo, TlsConnectInfo},
|
|
Identity, Server, ServerTlsConfig,
|
|
},
|
|
Request, Response, Status,
|
|
};
|
|
|
|
type EchoResult<T> = Result<Response<T>, Status>;
|
|
|
|
#[derive(Default)]
|
|
pub struct EchoServer;
|
|
|
|
#[tonic::async_trait]
|
|
impl pb::echo_server::Echo for EchoServer {
|
|
async fn unary_echo(&self, request: Request<EchoRequest>) -> EchoResult<EchoResponse> {
|
|
let conn_info = request
|
|
.extensions()
|
|
.get::<TlsConnectInfo<TcpConnectInfo>>()
|
|
.unwrap();
|
|
println!(
|
|
"Got a request from {:?} with info {:?}",
|
|
request.remote_addr(),
|
|
conn_info
|
|
);
|
|
|
|
let message = request.into_inner().message;
|
|
Ok(Response::new(EchoResponse { message }))
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let cert = tokio::fs::read("examples/data/tls/server.pem").await?;
|
|
let key = tokio::fs::read("examples/data/tls/server.key").await?;
|
|
|
|
let identity = Identity::from_pem(cert, key);
|
|
|
|
let addr = "[::1]:50051".parse().unwrap();
|
|
let server = EchoServer::default();
|
|
|
|
Server::builder()
|
|
.tls_config(ServerTlsConfig::new().identity(identity))?
|
|
.add_service(pb::echo_server::EchoServer::new(server))
|
|
.serve(addr)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|