From 721cbeccd3bad115610550164937f67ecc665a91 Mon Sep 17 00:00:00 2001
From: OMGeeky <aalaalgmx@gmail.com>
Date: Sat, 13 Apr 2024 13:09:10 +0200
Subject: [PATCH] Cleanup

---
 gdriver-backend/src/service.rs |  7 +++----
 gdriver-client/src/service.rs  | 17 ++++++++++++++---
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/gdriver-backend/src/service.rs b/gdriver-backend/src/service.rs
index 7b1348e..a400c5c 100644
--- a/gdriver-backend/src/service.rs
+++ b/gdriver-backend/src/service.rs
@@ -148,9 +148,8 @@ async fn long_running_task(drive: &Arc<Mutex<Drive>>) {
     print_sample_tracking_state(drive).await;
 }
 async fn print_sample_tracking_state(drive: &Arc<Mutex<Drive>>) {
-    let lock = drive.lock();
-    let drive = lock.await;
-    let state = drive.get_file_tracking_state(&ROOT_ID);
+    let drive_lock = drive.lock().await;
+    let state = drive_lock.get_file_tracking_state(&ROOT_ID);
     dbg!(state);
 }
 pub async fn start() -> Result<()> {
@@ -170,7 +169,7 @@ pub async fn start() -> Result<()> {
         // Ignore accept errors.
         .filter_map(|r| future::ready(r.ok()))
         .map(server::BaseChannel::with_defaults)
-        // // Limit channels to 1 per IP.
+        // Limit channels to 1 per IP.
         .max_channels_per_key(1, |t| t.transport().peer_addr().unwrap().ip())
         // serve is generated by the service attribute. It takes as input any type implementing
         // the generated World trait.
diff --git a/gdriver-client/src/service.rs b/gdriver-client/src/service.rs
index 528b489..5fb2081 100644
--- a/gdriver-client/src/service.rs
+++ b/gdriver-client/src/service.rs
@@ -1,3 +1,4 @@
+
 use std::time;
 
 use gdriver_common::ipc::gdriver_service::{BackendActionRequest, GDriverServiceClient};
@@ -9,7 +10,11 @@ pub async fn start() -> Result<()> {
     let config = &CONFIGURATION;
     println!("Config: {:?}", **config);
     let client: GDriverServiceClient = create_client(config.ip, config.port).await?;
+    run_long_stuff_test(&client).await;
+    Ok(())
+}
 
+async fn ping(client: &GDriverServiceClient) -> Result<()>{
     let hello = client
         .do_something2(tarpc::context::current(), BackendActionRequest::Ping)
         .await;
@@ -17,9 +22,14 @@ pub async fn start() -> Result<()> {
         Ok(hello) => println!("Yay: {:?}", hello),
         Err(e) => {
             println!(":( {:?}", (e));
-            dbg!(e);
+            dbg!(&e);
+            return Err(Box::new(e));
         }
     }
+    Ok(())
+}
+#[allow(unused)]
+async fn run_long_stuff_test(client: &GDriverServiceClient) {
     let start = time::SystemTime::now();
     let hello = client
         .do_something2(tarpc::context::current(), BackendActionRequest::RunLong)
@@ -45,17 +55,18 @@ pub async fn start() -> Result<()> {
         Ok(hello) => println!("Start Long returned after {} seconds: {:?}", seconds, hello),
         Err(e) => println!(":( {:?}", (e)),
     }
-    Ok(())
 }
+
 pub async fn create_client(ip: IpAddr, port: u16) -> Result<GDriverServiceClient> {
     let server_addr = (ip, port);
     let transport = tarpc::serde_transport::tcp::connect(&server_addr, Json::default)
         .await
         .map_err(|e| {
-            println!("Could not connect");
+            println!("Could not connect to backend. Please make sure it is started before this app.");
             e
         })?;
     let service = GDriverServiceClient::new(client::Config::default(), transport);
     let client = service.spawn();
+    let _ = ping(&client).await;
     Ok(client)
 }