Page MenuHomePhorge
Files F32223315

D6459.1765208249.diff
No OneTemporary
Actions

Size
2 KB
Referenced Files
None
Subscribers
None

D6459.1765208249.diff
View Options

diff --git a/keyserver/addons/rust-node-addon/src/lib.rs b/keyserver/addons/rust-node-addon/src/lib.rs
--- a/keyserver/addons/rust-node-addon/src/lib.rs
+++ b/keyserver/addons/rust-node-addon/src/lib.rs
@@ -1,4 +1,5 @@
pub mod identity_client;
+pub mod tunnelbroker_client;
#[macro_use]
extern crate napi_derive;
diff --git a/keyserver/addons/rust-node-addon/src/tunnelbroker_client.rs b/keyserver/addons/rust-node-addon/src/tunnelbroker_client.rs
new file mode 100644
--- /dev/null
+++ b/keyserver/addons/rust-node-addon/src/tunnelbroker_client.rs
@@ -0,0 +1,61 @@
+use napi::threadsafe_function::{
+ ThreadsafeFunction, ThreadsafeFunctionCallMode,
+};
+use napi_derive::napi;
+use tokio::sync::mpsc;
+use tokio_stream::{wrappers::ReceiverStream, StreamExt};
+use tonic::Request;
+use tracing::error;
+
+use tunnelbroker::protobuff::message_to_client::Data::MessagesToDeliver;
+use tunnelbroker_client as tunnelbroker;
+
+#[napi]
+pub struct TunnelbrokerClient {
+ tx: mpsc::Sender<tunnelbroker::protobuff::MessageToTunnelbroker>,
+}
+
+#[napi]
+impl TunnelbrokerClient {
+ #[napi(constructor)]
+ pub fn new(
+ address: String,
+ on_receive_callback: ThreadsafeFunction<String>,
+ ) -> Self {
+ let mut client = tunnelbroker::initialize_client(address);
+ let (tx, rx) = mpsc::channel(1);
+ let stream = ReceiverStream::new(rx);
+
+ // Spawning asynchronous Tokio task for handling incoming messages from the client
+ // and calling the callback function with the received payload
+ tunnelbroker::RUNTIME.spawn({
+ async move {
+ let response = client
+ .messages_stream(Request::new(stream))
+ .await
+ .expect("Failed to receive messages stream from Tunnelbroker");
+ let mut resp_stream = response.into_inner();
+ while let Some(received) = resp_stream.next().await {
+ if let Some(message_data) =
+ received.expect("Error on getting messages data").data
+ {
+ match message_data {
+ MessagesToDeliver(messages_to_send) => {
+ for message in messages_to_send.messages {
+ on_receive_callback.call(
+ Ok(message.payload),
+ ThreadsafeFunctionCallMode::NonBlocking,
+ );
+ }
+ }
+ _ => {
+ error!("Received an unexpected message type");
+ }
+ }
+ }
+ }
+ }
+ });
+ TunnelbrokerClient { tx }
+ }
+}

File Metadata

Mime Type
text/plain
Expires
Mon, Dec 8, 3:37 PM (8 h, 45 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
5848895
Default Alt Text
D6459.1765208249.diff (2 KB)

Event Timeline