Fix failure in creating base keys

This commit is contained in:
Joshua Boniface 2021-07-18 20:58:14 -04:00
parent 602093029c
commit cffc96d156
1 changed files with 9 additions and 2 deletions

View File

@ -164,7 +164,11 @@ class ZookeeperLogger(Thread):
self.queue = Queue() self.queue = Queue()
self.zkhandler = None self.zkhandler = None
self.start_zkhandler() self.start_zkhandler()
self.zkhandler.write([(('logs', self.node), '')]) # Ensure the root keys for this are instantiated
self.zkhandler.write([
('base.logs', ''),
(('logs', self.node), '')
])
self.running = False self.running = False
Thread.__init__(self, args=(), kwargs=None) Thread.__init__(self, args=(), kwargs=None)
@ -181,7 +185,10 @@ class ZookeeperLogger(Thread):
def run(self): def run(self):
self.running = True self.running = True
# Get the logs that are currently in Zookeeper and populate our deque # Get the logs that are currently in Zookeeper and populate our deque
logs = deque(self.zkhandler.read(('logs.messages', self.node)).split('\n'), self.max_lines) raw_logs = self.zkhandler.read(('logs.messages', self.node))
if raw_logs is None:
raw_logs = ''
logs = deque(raw_logs.split('\n'), self.max_lines)
while self.running: while self.running:
# Get a new message # Get a new message
try: try: