mirror of
https://github.com/isj4/vagus.git
synced 2025-02-02 06:08:42 -05:00
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
class Instance(object):
|
|
def __init__(self,source,instance_identifier,keepalive_lifetime,end_of_life,extra_info):
|
|
self.source = source
|
|
self.instance_identifier = instance_identifier
|
|
self.keepalive_lifetime = keepalive_lifetime
|
|
self.end_of_life = end_of_life
|
|
self.extra_info = extra_info
|
|
def __str__(self):
|
|
return str(self.instance_identifier)
|
|
|
|
class InstanceDict(dict):
|
|
def update(self,source,instance_identifier,keepalive_lifetime,end_of_life,extra_info):
|
|
i = Instance(source,instance_identifier,keepalive_lifetime,end_of_life,extra_info)
|
|
if instance_identifier in self:
|
|
self[instance_identifier] = i
|
|
return False
|
|
else:
|
|
self[instance_identifier] = i
|
|
return True
|
|
|
|
def timeout_expired_instances(self,now):
|
|
removed_instance_ids = []
|
|
for k in list(self.keys()):
|
|
if self[k].end_of_life <= now:
|
|
removed_instance_ids.append(self[k].instance_identifier)
|
|
del self[k]
|
|
any_removed = True
|
|
return removed_instance_ids
|
|
|
|
def get_lowest_keepalive_lifetime(self):
|
|
m = None
|
|
for v in self.values():
|
|
if m==None or v.keepalive_lifetime<m:
|
|
m = v.keepalive_lifetime
|
|
return m
|
|
|
|
|
|
if __name__ == "__main__":
|
|
d = InstanceDict()
|
|
assert 1 not in d
|
|
assert d.update("me",1,1234,5678,"foo")
|
|
assert 1 in d
|
|
assert d[1].source == "me"
|
|
assert d[1].keepalive_lifetime == 1234
|
|
assert d[1].end_of_life == 5678
|
|
assert d[1].extra_info == "foo"
|
|
assert not d.update("me",1,12345,67890,"foo")
|
|
assert d.update("me",2,12345,67890,"boo")
|
|
|
|
|
|
d = InstanceDict()
|
|
assert not d.timeout_expired_instances(10)
|
|
d.update("me",1,17,17,"foo")
|
|
d.update("me",2,42,42,"boo")
|
|
d.update("me",3,117,117,"goo")
|
|
assert not d.timeout_expired_instances(10)
|
|
assert d.timeout_expired_instances(20)
|
|
assert not d.timeout_expired_instances(20)
|
|
assert d.timeout_expired_instances(200)
|
|
assert len(d)==0
|