feat: complete transfer and eviction execution
This commit is contained in:
@@ -0,0 +1,65 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Drive a covered cache eviction through the bot-free HTTP adapter."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import sys
|
||||
import time
|
||||
import urllib.request
|
||||
|
||||
|
||||
CONTROL = "http://127.0.0.1:18081/test/v1"
|
||||
INFO_HASH = "330be0cb7c2201135a2de63b28e77993745ff688"
|
||||
|
||||
|
||||
def get_json(url: str):
|
||||
with urllib.request.urlopen(url, timeout=35) as response:
|
||||
return json.load(response)
|
||||
|
||||
|
||||
def post_json(url: str, value: object):
|
||||
request = urllib.request.Request(
|
||||
url,
|
||||
data=json.dumps(value, separators=(",", ":")).encode(),
|
||||
method="POST",
|
||||
headers={"Content-Type": "application/json"},
|
||||
)
|
||||
with urllib.request.urlopen(request, timeout=35) as response:
|
||||
return json.load(response)
|
||||
|
||||
|
||||
def main() -> int:
|
||||
preview = post_json(f"{CONTROL}/jobs/preview", {
|
||||
"operation": "evict_cache",
|
||||
"cache_client_id": "cache-1",
|
||||
"resource_id": {"info_hash_v1_hex": INFO_HASH},
|
||||
})
|
||||
definition = preview["definition"]
|
||||
job_id = definition["job_id"]
|
||||
post_json(f"{CONTROL}/jobs", {
|
||||
"preview_revision": preview["preview_revision"],
|
||||
"definition": definition,
|
||||
})
|
||||
deadline = time.monotonic() + 180
|
||||
while time.monotonic() < deadline:
|
||||
job = get_json(f"{CONTROL}/jobs/{job_id}")
|
||||
if job["state"] == "JOB_STATE_SUCCEEDED" and job["committed"]:
|
||||
print(json.dumps({
|
||||
"job_id": job_id,
|
||||
"state": job["state"],
|
||||
"committed": job["committed"],
|
||||
}, sort_keys=True))
|
||||
return 0
|
||||
if job["state"] in {"JOB_STATE_FAILED", "JOB_STATE_CANCELLED"}:
|
||||
raise RuntimeError(f"eviction did not succeed: {job}")
|
||||
time.sleep(0.5)
|
||||
raise RuntimeError("timed out waiting for cache eviction")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
raise SystemExit(main())
|
||||
except Exception as exc:
|
||||
print(f"eviction happy-path failed: {exc}", file=sys.stderr)
|
||||
raise
|
||||
Reference in New Issue
Block a user