From 4cd25a84423e6ac9762ddfbde441555c5a718138 Mon Sep 17 00:00:00 2001 From: Patryk Hegenberg Date: Sun, 28 Apr 2024 13:53:39 +0200 Subject: [PATCH] feature(relay): Add new Transfer struct to AppState To share the transfers between different routes and connections it is necessary to hold all transfers in the AppState. --- src/relay/appstate.rs | 8 +++++--- src/relay/client.rs | 5 ++--- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/src/relay/appstate.rs b/src/relay/appstate.rs index ac76b01..ee2d560 100644 --- a/src/relay/appstate.rs +++ b/src/relay/appstate.rs @@ -2,6 +2,7 @@ use std::{collections::HashMap, sync::Arc}; use tokio::sync::RwLock; use crate::relay::room::Room; +use crate::relay::transfer::Transfer; /// A struct that holds all of the rooms that the server knows about. /// @@ -11,6 +12,7 @@ use crate::relay::room::Room; #[derive(Debug)] pub struct AppState { pub rooms: HashMap, + pub transfers: Vec, } impl AppState { @@ -44,15 +46,15 @@ impl AppState { Arc::new(RwLock::new(AppState { // Initialize the list of rooms to be empty. rooms: HashMap::new(), + transfers: Vec::new(), })) } - } #[cfg(test)] mod tests { use super::*; - use std::sync::{Arc}; + use std::sync::Arc; #[test] fn test_new() { @@ -60,4 +62,4 @@ mod tests { assert!(Arc::ptr_eq(&app_state, &app_state.clone())); } -} \ No newline at end of file +} diff --git a/src/relay/client.rs b/src/relay/client.rs index 52c20f8..60371b7 100644 --- a/src/relay/client.rs +++ b/src/relay/client.rs @@ -491,6 +491,5 @@ impl Client { // TODO: Add tests #[cfg(test)] mod tests { - use super::*; - -} \ No newline at end of file + // use super::*; +}