Browse Source

timeout option added in etcd_wrapper.watch_prefix()

master
_ahmedbilal 3 years ago
parent
commit
615a070918
  1. 36
      etcd_wrapper.py

36
etcd_wrapper.py

@ -1,6 +1,11 @@
import etcd3
import json
import signal
from dataclasses import dataclass
from collections import namedtuple
PseudoEtcdMeta = namedtuple("PseudoEtcdMeta", ["key"])
@dataclass(init=False)
@ -42,11 +47,26 @@ class Etcd3Wrapper(object):
for entry in r:
yield EtcdEntry(*entry[::-1], value_in_json=value_in_json)
def watch_prefix(self, *args, value_in_json=False, **kwargs):
r, _ = self.client.watch_prefix(*args, **kwargs)
for event in r:
# if e.value is None don't propagate its value
if event.value is None:
continue
event = EtcdEntry(event, event.value, value_in_json=value_in_json)
yield event
def watch_prefix(self, *args, timeout=0, value_in_json=False, **kwargs):
if timeout > 0:
signal.signal(signal.SIGALRM, raise_timeout)
while True:
try:
if timeout > 0:
signal.alarm(timeout)
r, _ = self.client.watch_prefix(*args, **kwargs)
for event in r:
# if e.value is None don't propagate its value
if event.value is None:
continue
event = EtcdEntry(event, event.value, value_in_json=value_in_json)
yield event
except TimeoutError:
_value = {"status": "TIMEOUT"}
yield EtcdEntry(PseudoEtcdMeta(key=b"TIMEOUT"), value=str.encode(json.dumps(_value)))
def raise_timeout(signum, frame):
_ = signum, frame
raise TimeoutError()

Loading…
Cancel
Save