GoldenPogacsa/producer
GoldenPogacsa/producer
Archived
7
0
Fork 0
Code Issues Pull Requests Releases Activity

Added communicator instance in ConsumerLocator

Browse Source
This commit is contained in:
Scharnitzky Donát 2020-03-29 17:19:05 +02:00
parent bc99060d11
commit 53ab2193b1
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
consumerlocator.py
View File

@ -19,18 +19,19 @@ class ConsumerLocator:
Manages the list of consumers. Manages the list of consumers.
""" """
def __init__(self): def __init__(self, comm: communicator.Communicator):
""" """
Initialize class. Initialize class.
""" """
self.consumerList = [{"Host": "KnownHost", "State": True, "LastOk": datetime.datetime.now()}] self.consumerList = [{"Host": "KnownHost", "State": True, "LastOk": datetime.datetime.now()}]
self.currentConsumer = self.consumerList[0]["Host"] self.currentConsumer = self.consumerList[0]["Host"]
self.communicator = comm
def learnconsumerlist(self): def learnconsumerlist(self):
"""" """"
Learns the list of consumers. Learns the list of consumers.
""" """
recievedConsumerList = communicator.discoveravailableconsumers() recievedConsumerList = self.communicator.discoveravailableconsumers()
for consumer in recievedConsumerList: for consumer in recievedConsumerList:
self.consumerList.append({"Host": consumer, "State": True, "LastOk": datetime.datetime.now()}) self.consumerList.append({"Host": consumer, "State": True, "LastOk": datetime.datetime.now()})
self.updateConsumerList() self.updateConsumerList()
@ -41,7 +42,7 @@ class ConsumerLocator:
""" """
removeList = [] removeList = []
for consumer in self.consumerList: for consumer in self.consumerList:
if not communicator.checkconsumer(consumer["Host"]): if not self.communicator.checkconsumer(consumer["Host"]):
consumer["State"] = False consumer["State"] = False
if datetime.datetime.now() - consumer["LastOk"] > datetime.timedelta(hours=1): if datetime.datetime.now() - consumer["LastOk"] > datetime.timedelta(hours=1):
removeList.append(consumer) removeList.append(consumer)
@ -85,7 +86,7 @@ class ConsumerLocator:
Check the consumers health. Check the consumers health.
:return: True if OK, False if fail :return: True if OK, False if fail
""" """
if communicator.checkconsumer(self.currentConsumer["Host"]): if self.communicator.checkconsumer(self.currentConsumer["Host"]):
return True return True
else: else:
return False return False