Differential D7715 Diff 26074 services/tunnelbroker/src/grpc/mod.rs

Changeset View

Standalone View

services/tunnelbroker/src/grpc/mod.rs

mod proto {		mod proto {
tonic::include_proto!("tunnelbroker");		tonic::include_proto!("tunnelbroker");
}		}

use proto::tunnelbroker_service_server::{		use proto::tunnelbroker_service_server::{
TunnelbrokerService, TunnelbrokerServiceServer,		TunnelbrokerService, TunnelbrokerServiceServer,
};		};
use proto::Empty;		use proto::Empty;
use tonic::transport::Server;		use tonic::transport::Server;
use tonic::Status;		use tonic::Status;
use tracing::debug;		use tracing::debug;

use crate::{constants, ACTIVE_CONNECTIONS};		use crate::{constants, ACTIVE_CONNECTIONS, CONFIG};

#[derive(Debug, Default)]		#[derive(Debug, Default)]
struct TunnelbrokerGRPC {}		struct TunnelbrokerGRPC {}

#[tonic::async_trait]		#[tonic::async_trait]
impl TunnelbrokerService for TunnelbrokerGRPC {		impl TunnelbrokerService for TunnelbrokerGRPC {
async fn send_message_to_device(		async fn send_message_to_device(
&self,		&self,
Show All 9 Lines	) -> Result<tonic::Response<proto::Empty>, tonic::Status> {
tx.send(message.payload).expect("Unable to send message");		tx.send(message.payload).expect("Unable to send message");

let response = tonic::Response::new(Empty {});		let response = tonic::Response::new(Empty {});
Ok(response)		Ok(response)
}		}
}		}

pub async fn run_server() -> Result<(), tonic::transport::Error> {		pub async fn run_server() -> Result<(), tonic::transport::Error> {
let addr = format!("[::1]:{}", constants::GRPC_SERVER_PORT)		let addr = format!("[::1]:{}", CONFIG.grpc_port)
.parse()		.parse()
.expect("Unable to parse gRPC address");		.expect("Unable to parse gRPC address");

tracing::info!("Websocket server listening on {}", &addr);		tracing::info!("Websocket server listening on {}", &addr);
Server::builder()		Server::builder()
.http2_keepalive_interval(Some(constants::GRPC_KEEP_ALIVE_PING_INTERVAL))		.http2_keepalive_interval(Some(constants::GRPC_KEEP_ALIVE_PING_INTERVAL))
.http2_keepalive_timeout(Some(constants::GRPC_KEEP_ALIVE_PING_TIMEOUT))		.http2_keepalive_timeout(Some(constants::GRPC_KEEP_ALIVE_PING_TIMEOUT))
.add_service(TunnelbrokerServiceServer::new(TunnelbrokerGRPC::default()))		.add_service(TunnelbrokerServiceServer::new(TunnelbrokerGRPC::default()))
.serve(addr)		.serve(addr)
.await		.await
}		}