67 lines
2.0 KiB
Python
Executable File
67 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Bench Graph client responsible for sending benchmarking data in JSON format to
|
|
the Bench Graph server.
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
from argparse import ArgumentParser
|
|
from datetime import datetime
|
|
|
|
import requests
|
|
|
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
GITHUB_REPOSITORY = os.getenv("GITHUB_REPOSITORY", "")
|
|
GITHUB_SHA = os.getenv("GITHUB_SHA", "")
|
|
GITHUB_REF = os.getenv("GITHUB_REF", "")
|
|
|
|
BENCH_GRAPH_SERVER_ENDPOINT = os.getenv(
|
|
"BENCH_GRAPH_SERVER_ENDPOINT",
|
|
"http://bench-graph-api:9001")
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def parse_args():
|
|
argp = ArgumentParser(description=__doc__)
|
|
argp.add_argument("--benchmark-name", type=str, required=True)
|
|
argp.add_argument("--benchmark-results-path", type=str, required=True)
|
|
argp.add_argument("--github-run-id", type=int, required=True)
|
|
argp.add_argument("--github-run-number", type=int, required=True)
|
|
argp.add_argument("--head-branch-name", type=str, required=True)
|
|
return argp.parse_args()
|
|
|
|
|
|
def post_measurement(args):
|
|
with open(args.benchmark_results_path, "r") as f:
|
|
data = json.load(f)
|
|
timestamp = datetime.now().timestamp()
|
|
req = requests.post(
|
|
f"{BENCH_GRAPH_SERVER_ENDPOINT}/measurements",
|
|
json={
|
|
"name": args.benchmark_name,
|
|
"timestamp": timestamp,
|
|
"git_repo": GITHUB_REPOSITORY,
|
|
"git_ref": GITHUB_REF,
|
|
"git_sha": GITHUB_SHA,
|
|
"github_run_id": args.github_run_id,
|
|
"github_run_number": args.github_run_number,
|
|
"results": data,
|
|
"git_branch": args.head_branch_name},
|
|
timeout=1)
|
|
assert req.status_code == 200, \
|
|
f"Uploading {args.benchmark_name} data failed."
|
|
log.info(f"{args.benchmark_name} data sent to "
|
|
f"{BENCH_GRAPH_SERVER_ENDPOINT}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_args()
|
|
logging.basicConfig(level=logging.INFO)
|
|
post_measurement(args)
|