26 lines
908 B
Python
26 lines
908 B
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Restart both AO3 workers manually."""
|
||
|
|
import subprocess, time, os
|
||
|
|
|
||
|
|
BASE_DIR = "/home/ubuntu/ao3-mirror"
|
||
|
|
SHM_DIR = "/dev/shm/ao3"
|
||
|
|
VENV = "/home/ubuntu/.hermes/hermes-agent/venv/bin/python3"
|
||
|
|
|
||
|
|
os.makedirs(SHM_DIR, exist_ok=True)
|
||
|
|
|
||
|
|
for i, port in [(0, 8081), (1, 8082)]:
|
||
|
|
log = f"{BASE_DIR}/worker-{i}.log"
|
||
|
|
proc = subprocess.Popen(
|
||
|
|
["taskset", "-c", str(i), VENV, "-m", "uvicorn", "app:app",
|
||
|
|
"--host", "127.0.0.1", "--port", str(port), "--workers", "1",
|
||
|
|
"--loop", "uvloop", "--http", "httptools",
|
||
|
|
"--log-level", "warning", "--timeout-keep-alive", "30"],
|
||
|
|
cwd=BASE_DIR, stdout=open(log, "a"), stderr=open(log, "a"),
|
||
|
|
)
|
||
|
|
pid_file = f"{SHM_DIR}/worker-{i}.pid"
|
||
|
|
with open(pid_file, "w") as f:
|
||
|
|
f.write(str(proc.pid))
|
||
|
|
print(f"Worker {i} port {port} started PID={proc.pid}")
|
||
|
|
time.sleep(2)
|
||
|
|
print("Both workers started")
|