perf(langserver): try to use the last initialized namespace for completions to be a bit more responsive

Author: d-biehl

packages/language_server/src/robotcode/language_server/robotframework/parts/completion.py

@@ -150,7 +150,7 @@ def collect(
         position: Position,
         context: Optional[CompletionContext],
     ) -> Union[List[CompletionItem], CompletionList, None]:
-        namespace = self.parent.documents_cache.get_namespace(document)
+        namespace = self.parent.documents_cache.get_initialized_namespace(document)
         model = self.parent.documents_cache.get_model(document, False)
 
         config = self.get_config(document)
@@ -172,7 +172,7 @@ def resolve(self, sender: Any, completion_item: CompletionItem) -> CompletionIte
             if document_uri is not None:
                 document = self.parent.documents.get(document_uri)
                 if document is not None:
-                    namespace = self.parent.documents_cache.get_namespace(document)
+                    namespace = self.parent.documents_cache.get_initialized_namespace(document)
                     model = self.parent.documents_cache.get_model(document, False)
                     if namespace is not None:
                         config = self.get_config(document)

packages/robot/src/robotcode/robot/diagnostics/document_cache_helper.py

@@ -40,6 +40,10 @@ class UnknownFileTypeError(Exception):
     pass
 
 
+class _CacheEntry:
+    pass
+
+
 class DocumentsCacheHelper:
     _logger = LoggingDescriptor()
 
@@ -50,6 +54,8 @@ def __init__(
         file_watcher_manager: FileWatcherManagerBase,
         robot_profile: Optional[RobotBaseProfile],
     ) -> None:
+        self.INITIALIZED_NAMESPACE = _CacheEntry()
+
         self.workspace = workspace
         self.documents_manager = documents_manager
         self.file_watcher_manager = file_watcher_manager
@@ -425,6 +431,20 @@ def __invalidate_namespace(self, sender: Namespace) -> None:
 
             self.namespace_invalidated(self, sender, callback_filter=language_id_filter(document))
 
+    def __namespace_initialized(self, sender: Namespace) -> None:
+        if sender.document is not None:
+            self._logger.critical(
+                lambda: f"Save initialized Namespace: {sender.document.uri if sender.document else None}"
+            )
+            sender.document.set_data(self.INITIALIZED_NAMESPACE, sender)
+
+    def get_initialized_namespace(self, document: TextDocument) -> Namespace:
+        result: Optional[Namespace] = document.get_data(self.INITIALIZED_NAMESPACE)
+        if result is None:
+            self._logger.critical(lambda: f"There is not initialized Namespace: {document.uri if document else None}")
+            result = self.get_namespace(document)
+        return result
+
     def __get_namespace_for_document_type(
         self, document: TextDocument, document_type: Optional[DocumentType]
     ) -> Namespace:
@@ -451,6 +471,7 @@ def __get_namespace_for_document_type(
             workspace_languages,
         )
         result.has_invalidated.add(self.__invalidate_namespace)
+        result.has_initialized.add(self.__namespace_initialized)
 
         return result