a.tolmachev
46 lines
1.3 KiB
Rust
46 lines
1.3 KiB
Rust
use base64::{Engine as _, engine::general_purpose::STANDARD};
|
|
use tokio::net::TcpListener;
|
|
use tonic::{Request, Response, Status, transport::Server};
|
|
|
|
pub mod echo {
|
|
tonic::include_proto!("echo");
|
|
pub const FILE_DESCRIPTOR_SET: &[u8] = tonic::include_file_descriptor_set!("echo_descriptor");
|
|
}
|
|
|
|
#[derive(Default)]
|
|
pub struct EchoServiceImpl;
|
|
|
|
#[tonic::async_trait]
|
|
impl echo::echo_service_server::EchoService for EchoServiceImpl {
|
|
async fn unary_echo(
|
|
&self,
|
|
request: Request<echo::EchoRequest>,
|
|
) -> Result<Response<echo::EchoResponse>, Status> {
|
|
Ok(Response::new(echo::EchoResponse {
|
|
message: request.into_inner().message,
|
|
}))
|
|
}
|
|
}
|
|
|
|
pub async fn spawn_unary_echo_server() -> String {
|
|
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
|
let address = listener.local_addr().unwrap();
|
|
let incoming = tonic::transport::server::TcpIncoming::from(listener);
|
|
|
|
tokio::spawn(async move {
|
|
Server::builder()
|
|
.add_service(echo::echo_service_server::EchoServiceServer::new(
|
|
EchoServiceImpl,
|
|
))
|
|
.serve_with_incoming(incoming)
|
|
.await
|
|
.unwrap();
|
|
});
|
|
|
|
format!("http://{}", address)
|
|
}
|
|
|
|
pub fn descriptor_set_b64() -> String {
|
|
STANDARD.encode(echo::FILE_DESCRIPTOR_SET)
|
|
}
|