Differential D8748 Diff 30122 services/identity/src/tunnelbroker.rs

Changeset View

Standalone View

				mod proto {
				tonic::include_proto!("tunnelbroker");
				}
				use crate::config::CONFIG;
				use proto::tunnelbroker_service_client::TunnelbrokerServiceClient;
				use proto::{Empty, MessageToDevice};
				use tonic::transport::Channel;
				use tonic::Response;
				use tracing::error;
				use tunnelbroker_messages as messages;

				use crate::error::Error;

				pub async fn create_tunnelbroker_client(
				) -> Result<TunnelbrokerServiceClient<Channel>, Error> {
				TunnelbrokerServiceClient::connect(CONFIG.tunnelbroker_endpoint.to_string())
				.await
				.map_err(\|e\| {
				error!("Unable able to connect to tunnelbroker: {:?}", e);
				e.into()
				})
				}

				pub async fn send_refresh_keys_request(
				device_id: &str,
				) -> Result<Response<Empty>, Error> {
				use crate::constants::ONETIME_KEY_REFRESH_NUMBER;

				let mut tunnelbroker_client = create_tunnelbroker_client().await?;

				let refresh_request = messages::RefreshKeyRequest {
				device_id: device_id.to_string(),
				number_of_keys: ONETIME_KEY_REFRESH_NUMBER,
				};

				let payload = serde_json::to_string(&refresh_request).unwrap();
				let request = MessageToDevice {
				device_id: device_id.to_string(),
				payload,
				};

				let grpc_message = tonic::Request::new(request);
				Ok(
				tunnelbroker_client
				.send_message_to_device(grpc_message)
				.await?,
				)
				}