Use thread to get the context tree

Author: mmokko

robocorp-code/src/robocorp_code/inspector/java/robocorp_java/_context_tree.py

@@ -0,0 +1,23 @@
+import queue
+import threading
+
+from JABWrapper.context_tree import ContextTree
+from JABWrapper.jab_wrapper import JavaAccessBridgeWrapper
+
+
+class ContextTreeThread(threading.Thread):
+    """
+    Building the ContextTree in a big java application may take several minutes.
+    """
+
+    def __init__(self, jab_wrapper: JavaAccessBridgeWrapper):
+        super().__init__()
+        self._queue = queue.Queue()
+        self._jab_wrapper = jab_wrapper
+
+    def run(self) -> None:
+        context_tree = ContextTree(self._jab_wrapper)
+        self._queue.put(context_tree)
+
+    def get_context_tree(self) -> ContextTree:
+        return self._queue.get()

robocorp-code/src/robocorp_code/inspector/java/robocorp_java/_event_pump.py

@@ -17,7 +17,7 @@
 REMOVE_FROM_QUEUE = 0x0001
 
 
-class _EventPumpThread(threading.Thread):
+class EventPumpThread(threading.Thread):
     def __init__(
         self,
     ) -> None:

robocorp-code/src/robocorp_code/inspector/java/robocorp_java/_inspector.py

@@ -7,9 +7,9 @@
 class ElementInspector:
     def _start_event_pump(func, *args, **kwargs):
         def wrapper(self: "ElementInspector", *args, **kwargs):
-            from ._event_pump import _EventPumpThread
+            from ._event_pump import EventPumpThread
 
-            event_pump_thread = _EventPumpThread()
+            event_pump_thread = EventPumpThread()
             event_pump_thread.start()
             jab_wrapper = event_pump_thread.get_wrapper()
             ret = func(self, jab_wrapper, *args, **kwargs)
@@ -30,7 +30,14 @@ def collect_tree(
         locator: Optional[str] = None,
     ) -> Union[ContextNode, List[ContextNode]]:
         jab_wrapper.switch_window_by_title(window)
-        context_tree = ContextTree(jab_wrapper)
+
+        from ._context_tree import ContextTreeThread
+
+        context_tree_thread = ContextTreeThread(jab_wrapper)
+        context_tree_thread.start()
+        context_tree = context_tree_thread.get_context_tree()
+        context_tree_thread.join()
+
         if locator:
             from ._locators import find_elements_from_tree