def connection_listener(self, state): self.log.warning(f"Zookeeper connection transitioned to: {state}") if state == KazooState.SUSPENDED: self.log.warning("Zookeeper connection suspended, waiting to see if it recovers.") if not self.waiting_for_reconnect: self.waiting_for_reconnect = True reconnection_checker = PaastaThread(target=self.reconnection_listener) reconnection_checker.daemon = True reconnection_checker.start() elif state == KazooState.LOST: self.log.error("Leadership lost, quitting!") self._terminate()
def setUp(self): self.thread = PaastaThread()