Unverified Commit ff99c38a authored by SimoneRaponi's avatar SimoneRaponi Committed by GitHub
Browse files

Add timeout to get_meta_info (#346)


Co-authored-by: default avatarsimone <simone.raponi@equixely.com>
parent c9de3e16
...@@ -248,9 +248,11 @@ class StreamExecutor: ...@@ -248,9 +248,11 @@ class StreamExecutor:
def set_var(self, name, value): def set_var(self, name, value):
self.variables[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: 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) ret = self.meta_info.get(name, None)
return ret return ret
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment