Add timeout to get_meta_info (#346)

Co-authored-by: simone <simone.raponi@equixely.com>
This commit is contained in:
SimoneRaponi
2024-04-03 16:22:06 +02:00
committed by GitHub
parent c9de3e169c
commit ff99c38a07

View File

@@ -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