Differential D7693 Diff 26315 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 tonic::transport::Server;			use tonic::transport::Server;
				use tonic::Status;
				use tracing::debug;

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

	#[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,
	_request: tonic::Request<proto::MessageToDevice>,			request: tonic::Request<proto::MessageToDevice>,
	) -> Result<tonic::Response<proto::Empty>, tonic::Status> {			) -> Result<tonic::Response<proto::Empty>, tonic::Status> {
	unimplemented!()			let message = request.into_inner();

				debug!("Received message for {}", &message.device_id);
				// TODO: Persist messages for inactive connections
				let tx = ACTIVE_CONNECTIONS
				.get(&message.device_id)
				.ok_or(Status::unavailable("Device does not exist"))?;
				tx.send(message.payload).expect("Unable to send message");

				let response = tonic::Response::new(Empty {});
				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]:{}", constants::GRPC_SERVER_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
	}			}