IntenseWebs/veilid
3
0
Fork 0
mirror of https://gitlab.com/veilid/veilid.git synced 2025-02-20 11:48:35 -06:00
Code Issues Packages Projects Releases Wiki Activity

wasm refactor for flows

Browse Source
This commit is contained in:
Christien Rioux 2023-11-04 14:24:40 -04:00
parent 0640342556
commit 38cd29f9a1
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
veilid-core/src/network_manager/wasm/mod.rs
View File

@ -246,11 +246,11 @@ impl Network {
} }
#[cfg_attr(feature="verbose-tracing", instrument(level="trace", err, skip(self, data), fields(data.len = data.len())))] #[cfg_attr(feature="verbose-tracing", instrument(level="trace", err, skip(self, data), fields(data.len = data.len())))]
pub async fn send_data_to_existing_connection( pub async fn send_data_to_existing_flow(
&self, &self,
flow: Flow, flow: Flow,
data: Vec<u8>, data: Vec<u8>,
) -> EyreResult<Option<Vec<u8>>> { ) -> EyreResult<SendDataToExistingFlowResult> {
let data_len = data.len(); let data_len = data.len();
match flow.protocol_type() { match flow.protocol_type() {
ProtocolType::UDP => { ProtocolType::UDP => {
@ -276,18 +276,18 @@ impl Network {
); );
// Data was consumed // Data was consumed
return Ok(None); return Ok(SendDataToExistingFlowResult::Sent(conn.unique_flow()));
} }
ConnectionHandleSendResult::NotSent(data) => { ConnectionHandleSendResult::NotSent(data) => {
// Couldn't send // Couldn't send
// Pass the data back out so we don't own it any more // Pass the data back out so we don't own it any more
return Ok(Some(data)); return Ok(SendDataToExistingFlowResult::NotSent(data));
} }
} }
} }
// Connection didn't exist // Connection didn't exist
// Pass the data back out so we don't own it any more // Pass the data back out so we don't own it any more
Ok(Some(data)) Ok(SendDataToExistingFlowResult::NotSent(data))
} }
#[cfg_attr(feature="verbose-tracing", instrument(level="trace", err, skip(self, data), fields(data.len = data.len())))] #[cfg_attr(feature="verbose-tracing", instrument(level="trace", err, skip(self, data), fields(data.len = data.len())))]
@ -295,7 +295,7 @@ impl Network {
&self, &self,
dial_info: DialInfo, dial_info: DialInfo,
data: Vec<u8>, data: Vec<u8>,
) -> EyreResult<NetworkResult<Flow>> { ) -> EyreResult<NetworkResult<UniqueFlow>> {
self.record_dial_info_failure(dial_info.clone(), async move { self.record_dial_info_failure(dial_info.clone(), async move {
let data_len = data.len(); let data_len = data.len();
if dial_info.protocol_type() == ProtocolType::UDP { if dial_info.protocol_type() == ProtocolType::UDP {
@ -318,13 +318,13 @@ impl Network {
"failed to send", "failed to send",
))); )));
} }
let flow = conn.flow(); let unique_flow = conn.unique_flow();
// Network accounting // Network accounting
self.network_manager() self.network_manager()
.stats_packet_sent(dial_info.ip_addr(), ByteCount::new(data_len as u64)); .stats_packet_sent(dial_info.ip_addr(), ByteCount::new(data_len as u64));
Ok(NetworkResult::value(flow)) Ok(NetworkResult::value(unique_flow))
}) })
.await .await
} }