Add timeout to get_meta_info (#346)

Co-authored-by: simone <simone.raponi@equixely.com>
2024-04-03 22:22:06 +08:00
parent c9de3e169c
commit ff99c38a07
1 changed files with 4 additions and 2 deletions
@@ -248,9 +248,11 @@ class StreamExecutor:
    def set_var(self, name, value):
        self.variables[name] = value

-    def get_meta_info(self, name):
+    def get_meta_info(self, name, timeout=None):
        if name in self.variable_event:
-            self.variable_event[name].wait()
+            got = self.variable_event[name].wait(timeout)
+            if not got:
+                raise TimeoutError(f"Timeout while waiting for event '{name}'")
        ret = self.meta_info.get(name, None)
        return ret