mirror of
https://github.com/dbarzin/pandora-box.git
synced 2025-07-19 05:19:40 +02:00
add wait for workers to be ready
This commit is contained in:
parent
192414a369
commit
512589f636
2 changed files with 22 additions and 0 deletions
|
@ -875,10 +875,29 @@ def get_lock(process_name):
|
|||
|
||||
# --------------------------------------
|
||||
|
||||
def get_enabled_workers():
|
||||
config_dir = Path("~/pandora/pandora/workers")
|
||||
yml_files = list(config_dir.glob("*.yml"))
|
||||
return [file.stem for file in yml_files]
|
||||
|
||||
def wait_for_workers():
|
||||
target_count = get_enabled_workers() - 1
|
||||
while True:
|
||||
workers = self.pandora.get_enabled_workers()
|
||||
log(f"Workers ready : {len(workers)}/ {target_count}")
|
||||
if len(workers) >= target_count:
|
||||
break
|
||||
time.sleep(2)
|
||||
|
||||
# --------------------------------------
|
||||
|
||||
def main(_):
|
||||
"""Main entry point"""
|
||||
print("main")
|
||||
try:
|
||||
# Wait for workers to start
|
||||
|
||||
# Enter the mail loop
|
||||
state = "START"
|
||||
while state != "STOP":
|
||||
state = loop(state)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue