chore: cache relationship

Okabe-Rintarou-0 · Oct 1, 2024 · 3768dfd · 3768dfd
1 parent 7884840
commit 3768dfd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src-tauri/src/app/basic.rs b/src-tauri/src/app/basic.rs
@@ -28,7 +28,10 @@ use crate::{
     utils::{self, TempFile},
 };
 
-use super::{constants::COURSES_CACHE_KEY, App};
+use super::{
+    constants::{COURSES_CACHE_KEY, RELATIONSHIP_CACHE_KEY},
+    App,
+};
 
 const MY_CANVAS_FILES_FOLDER_NAME: &str = "我的Canvas文件";
 
@@ -851,8 +854,13 @@ impl App {
     }
 
     pub async fn collect_relationship(&self) -> Result<RelationshipTopo> {
+        let topo = self.cache.get(RELATIONSHIP_CACHE_KEY)?;
+        if let Some(topo) = topo {
+            return Ok(topo);
+        }
         let token = self.config.read().await.token.clone();
         let topo = self.client.clone().collect_relationship(&token).await?;
+        self.cache.set(RELATIONSHIP_CACHE_KEY, topo.clone())?;
         Ok(topo)
     }
 

diff --git a/src-tauri/src/app/constants.rs b/src-tauri/src/app/constants.rs
@@ -1 +1,2 @@
 pub const COURSES_CACHE_KEY: &str = "courses_cache_key";
+pub const RELATIONSHIP_CACHE_KEY: &str = "relationship_cache_key";