perf: prefetch package index for completion (#983)

Myriad-Dreamin · Dec 11, 2024 · a86f7a4 · a86f7a4
1 parent 4f5c9c7
commit a86f7a4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/crates/tinymist-world/src/package.rs b/crates/tinymist-world/src/package.rs
@@ -218,6 +218,11 @@ impl PackageStorage {
         }
     }
 
+    /// Get the cached package index without network access.
+    pub fn cached_index(&self) -> Option<&[(PackageSpec, Option<EcoString>)]> {
+        self.index.get().map(Vec::as_slice)
+    }
+
     /// Download the package index. The result of this is cached for efficiency.
     pub fn download_index(&self) -> &[(PackageSpec, Option<EcoString>)] {
         self.index.get_or_init(|| {

diff --git a/crates/tinymist/src/server.rs b/crates/tinymist/src/server.rs
@@ -1069,6 +1069,16 @@ impl LanguageState {
             }
             fut_stat.stat.snap();
 
+            if matches!(query, Completion(..)) {
+                // Prefetch the package index for completion.
+                if snap.world.registry.cached_index().is_none() {
+                    let registry = snap.world.registry.clone();
+                    tokio::spawn(async move {
+                        let _ = registry.download_index();
+                    });
+                }
+            }
+
             match query {
                 SemanticTokensFull(req) => snap.run_semantic(req, R::SemanticTokensFull),
                 SemanticTokensDelta(req) => snap.run_semantic(req, R::SemanticTokensDelta),