Jędrzej Stuczyński
56a384ea09
* [wasm-client] keeping ownership over 'ReceivedBufferRequestSender' channel when spawning 'ResponsePusher' * Bump version of Typescript SDK to RC.10 * GitHub Actions workflow to publish SDK to NPM * Bump package version manually --------- Co-authored-by: Mark Sinclair <mmsinclair@users.noreply.github.com>
74 lines
2.5 KiB
Rust
74 lines
2.5 KiB
Rust
// Copyright 2022 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use futures::channel::mpsc;
|
|
use futures::StreamExt;
|
|
use js_sys::Uint8Array;
|
|
use wasm_bindgen::JsValue;
|
|
use wasm_bindgen_futures::spawn_local;
|
|
use wasm_client_core::client::base_client::ClientOutput;
|
|
use wasm_client_core::client::received_buffer::{
|
|
ReceivedBufferMessage, ReceivedBufferRequestSender, ReconstructedMessagesReceiver,
|
|
};
|
|
use wasm_utils::console_error;
|
|
|
|
pub(crate) struct ResponsePusher {
|
|
reconstructed_receiver: ReconstructedMessagesReceiver,
|
|
|
|
// we need to keep that channel alive as not to trigger the shutdown
|
|
_received_buffer_request_sender: ReceivedBufferRequestSender,
|
|
|
|
on_message: js_sys::Function,
|
|
}
|
|
|
|
impl ResponsePusher {
|
|
pub(crate) fn new(
|
|
client_output: ClientOutput,
|
|
on_message: js_sys::Function,
|
|
// on_mix_fetch_message: js_sys::Function,
|
|
) -> Self {
|
|
// register our output
|
|
let (reconstructed_sender, reconstructed_receiver) = mpsc::unbounded();
|
|
|
|
// tell the buffer to start sending stuff to us
|
|
client_output
|
|
.received_buffer_request_sender
|
|
.unbounded_send(ReceivedBufferMessage::ReceiverAnnounce(
|
|
reconstructed_sender,
|
|
))
|
|
.expect("the buffer request failed!");
|
|
|
|
ResponsePusher {
|
|
reconstructed_receiver,
|
|
_received_buffer_request_sender: client_output.received_buffer_request_sender,
|
|
on_message,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn start(mut self) {
|
|
spawn_local(async move {
|
|
// for some reason if this channel is not explicitly moved into the block,
|
|
// it gets dropped when spawning the promise
|
|
let _request_sender = self._received_buffer_request_sender;
|
|
|
|
let this = JsValue::null();
|
|
|
|
while let Some(reconstructed) = self.reconstructed_receiver.next().await {
|
|
for reconstructed_msg in reconstructed {
|
|
let (msg, tag) = reconstructed_msg.into_inner();
|
|
|
|
let msg_slice: &[u8] = &msg;
|
|
let array = Uint8Array::from(msg_slice);
|
|
let arg1 = JsValue::from(array);
|
|
let arg2 = JsValue::from(tag);
|
|
self.on_message
|
|
.call2(&this, &arg1, &arg2)
|
|
.expect("on binary message failed!");
|
|
}
|
|
}
|
|
|
|
console_error!("we stopped receiving reconstructed messages!")
|
|
})
|
|
}
|
|
}
|